package storage

import (
	"errors"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strings"
	"sync"
	"sync/atomic"
	"syscall"

	"github.com/seaweedfs/seaweedfs/weed/pb"
	"github.com/seaweedfs/seaweedfs/weed/storage/volume_info"
	"github.com/seaweedfs/seaweedfs/weed/util"

	"google.golang.org/grpc"

	"github.com/seaweedfs/seaweedfs/weed/glog"
	"github.com/seaweedfs/seaweedfs/weed/pb/master_pb"
	"github.com/seaweedfs/seaweedfs/weed/pb/volume_server_pb"
	"github.com/seaweedfs/seaweedfs/weed/stats"
	"github.com/seaweedfs/seaweedfs/weed/storage/erasure_coding"
	"github.com/seaweedfs/seaweedfs/weed/storage/needle"
	"github.com/seaweedfs/seaweedfs/weed/storage/super_block"
	. "github.com/seaweedfs/seaweedfs/weed/storage/types"
)

const (
	MAX_TTL_VOLUME_REMOVAL_DELAY = 10 // 10 minutes
	HEARTBEAT_CHAN_SIZE          = 1024
)

type ReadOption struct {
	// request
	ReadDeleted     bool
	AttemptMetaOnly bool
	MustMetaOnly    bool
	// response
	IsMetaOnly     bool // read status
	VolumeRevision uint16
	IsOutOfRange   bool // whether read over MaxPossibleVolumeSize

	// If HasSlowRead is set to true:
	//  * read requests and write requests compete for the lock.
	//  * large file read P99 latency on busy sites will go up, due to the need to get locks multiple times.
	//  * write requests will see lower latency.
	// If HasSlowRead is set to false:
	//  * read requests should complete asap, not blocking other requests.
	//  * write requests may see high latency when downloading large files.
	HasSlowRead bool

	// increasing ReadBufferSize can reduce the number of get locks times and shorten read P99 latency.
	// but will increase memory usage a bit. Use with hasSlowRead normally.
	ReadBufferSize int
}

/*
 * A VolumeServer contains one Store
 */
type Store struct {
	MasterAddress       pb.ServerAddress
	grpcDialOption      grpc.DialOption
	volumeSizeLimit     uint64      // read from the master
	preallocate         atomic.Bool // read from the master
	Ip                  string
	Port                int
	GrpcPort            int
	PublicUrl           string
	Id                  string // volume server id, independent of ip:port for stable identification
	Locations           []*DiskLocation
	dataCenter          string // optional information, overwriting master setting if exists
	rack                string // optional information, overwriting master setting if exists
	connected           bool
	NeedleMapKind       NeedleMapKind
	State               *State
	StateUpdateChan     chan *volume_server_pb.VolumeServerState
	NewVolumesChan      chan *master_pb.VolumeShortInformationMessage
	DeletedVolumesChan  chan *master_pb.VolumeShortInformationMessage
	NewEcShardsChan     chan *master_pb.VolumeEcShardInformationMessage
	DeletedEcShardsChan chan *master_pb.VolumeEcShardInformationMessage
	isStopping          atomic.Bool
	volumeReport        volumeReportState
	// One heartbeat at a time: the report state is marked in place as the scan
	// runs, so two overlapping scans would each forget what the other marked
	// and name every volume it holds as departed.
	collectHeartbeatLock sync.Mutex
	// Collections the last heartbeat set per-collection gauges for. Those gauges
	// are only ever set for collections still held here, so one whose last
	// volume leaves - moved away by volume.balance, say - would keep reporting
	// the heartbeat that saw it. Written only from the heartbeat goroutine.
	reportedCollections   map[string]struct{}
	reportedEcCollections map[string]struct{}
}

func (s *Store) String() (str string) {
	str = fmt.Sprintf("Id:%s, Ip:%s, Port:%d, GrpcPort:%d PublicUrl:%s, dataCenter:%s, rack:%s, connected:%v, volumeSizeLimit:%d", s.Id, s.Ip, s.Port, s.GrpcPort, s.PublicUrl, s.dataCenter, s.rack, s.connected, s.GetVolumeSizeLimit())
	return
}

func NewStore(
	grpcDialOption grpc.DialOption,
	ip string, port int, grpcPort int, publicUrl string, id string,
	dirnames []string, maxVolumeCounts []int32, minFreeSpaces []util.MinFreeSpace,
	idxFolder string,
	needleMapKind NeedleMapKind,
	diskTypes []DiskType,
	diskTags [][]string,
	ldbTimeout int64,
	diskProbeConfig stats.DiskIOProbeConfig,
) (s *Store) {
	s = &Store{
		grpcDialOption: grpcDialOption,
		Port:           port,
		Ip:             ip,
		GrpcPort:       grpcPort,
		PublicUrl:      publicUrl,
		Id:             id,
		NeedleMapKind:  needleMapKind,
		Locations:      make([]*DiskLocation, 0),

		StateUpdateChan:     make(chan *volume_server_pb.VolumeServerState, HEARTBEAT_CHAN_SIZE),
		NewVolumesChan:      make(chan *master_pb.VolumeShortInformationMessage, HEARTBEAT_CHAN_SIZE),
		DeletedVolumesChan:  make(chan *master_pb.VolumeShortInformationMessage, HEARTBEAT_CHAN_SIZE),
		NewEcShardsChan:     make(chan *master_pb.VolumeEcShardInformationMessage, HEARTBEAT_CHAN_SIZE),
		DeletedEcShardsChan: make(chan *master_pb.VolumeEcShardInformationMessage, HEARTBEAT_CHAN_SIZE),
	}

	var wg sync.WaitGroup
	for i := 0; i < len(dirnames); i++ {
		var tags []string
		if i < len(diskTags) {
			tags = diskTags[i]
		}
		location := NewDiskLocation(dirnames[i], int32(maxVolumeCounts[i]), minFreeSpaces[i], idxFolder, diskTypes[i], tags, diskProbeConfig)
		s.Locations = append(s.Locations, location)
		stats.VolumeServerMaxVolumeCounter.Add(float64(maxVolumeCounts[i]))

		diskId := uint32(i) // Track disk ID

		location.ecShardNotifyHandler = func(collection string, vid needle.VolumeId, shardId erasure_coding.ShardId, ecVolume *erasure_coding.EcVolume) {
			si := erasure_coding.NewShardsInfo()
			si.Set(erasure_coding.NewShardInfo(shardId, erasure_coding.ShardSize(ecVolume.ShardSize())))

			// Use non-blocking send during startup to avoid deadlock
			// The channel reader only starts after connecting to master, but we're loading during startup
			select {
			case s.NewEcShardsChan <- &master_pb.VolumeEcShardInformationMessage{
				Id:          uint32(vid),
				Collection:  collection,
				EcIndexBits: si.Bitmap(),
				ShardSizes:  si.SizesInt64(),
				DiskType:    string(location.DiskType),
				ExpireAtSec: ecVolume.ExpireAtSec,
				DiskId:      diskId,
				EncodeTsNs:  ecVolume.EncodeTsNs,
			}:
			default:
				// Channel full during startup - this is OK, heartbeat will report EC shards later
				glog.V(2).Infof("NewEcShardsChan full during startup for shard %d.%d, will be reported in heartbeat", vid, shardId)
			}
		}

		wg.Add(1)
		go func(id uint32, diskLoc *DiskLocation) {
			defer wg.Done()
			diskLoc.loadExistingVolumesWithId(needleMapKind, ldbTimeout, id)
		}(diskId, location)
	}
	wg.Wait()

	// First, scrub partial EC artefacts left on one disk by an interrupted
	// encode while the source .dat still lives on a sibling disk of the
	// same store. The per-disk loader cannot see the sibling .dat and so
	// loads the partial shards as if they were a distributed-EC layout,
	// which makes the volume server heartbeat both a regular replica and
	// an EC shard set for the same vid (issue #9478). Running before the
	// cross-disk reconcile keeps that pass from later re-loading shards
	// we just cleaned up.
	s.pruneIncompleteEcWithSiblingDat()

	// Physically mirror EC sidecars onto every shard-bearing disk so
	// each disk mounts self-contained. Must run before the cross-disk
	// reconciler so the orphan pass can prefer the local IdxDirectory.
	s.mirrorEcMetadataToShardDisks()

	// Cross-disk fallback for orphan shards — ec.balance can land
	// shards on one disk while leaving the index on another. Still
	// needed after the mirror pass for volumes whose mirror failed
	// (read-only target, out of space, partial copy).
	s.reconcileEcShardsAcrossDisks()

	// Resolve state.pb's directory via the first disk location so it inherits
	// the same `~` expansion and empty-idxFolder fallback used for .idx files,
	// and is never written as a relative path against the process CWD (#9173).
	stateDir := idxFolder
	if len(s.Locations) > 0 {
		stateDir = s.Locations[0].IdxDirectory
	} else if stateDir != "" {
		stateDir = util.ResolvePath(stateDir)
	}
	var err error
	s.State, err = NewState(stateDir)
	if err != nil {
		glog.Fatalf("failed to resolve state for volume %s: %v", id, err)
	}

	return
}

func (s *Store) LoadState() error {
	err := s.State.Load()
	if s.State.Proto() != nil && err == nil {
		select {
		case s.StateUpdateChan <- s.State.Proto():
		default:
			glog.V(2).Infof("StateUpdateChan full during LoadState, state will be reported in heartbeat")
		}
	}
	return err
}

func (s *Store) SaveState() error {
	if s.State.Proto() == nil {
		glog.Warningf("tried to save empty state for store %s", s.Id)
		return nil
	}

	err := s.State.Save()
	if s.State.Proto() != nil && err == nil {
		select {
		case s.StateUpdateChan <- s.State.Proto():
		default:
			glog.V(2).Infof("StateUpdateChan full during SaveState, state will be reported in heartbeat")
		}
	}
	return err
}

func (s *Store) AddVolume(volumeId needle.VolumeId, collection string, needleMapKind NeedleMapKind, replicaPlacement string, ttlString string, preallocate int64, ver needle.Version, MemoryMapMaxSizeMb uint32, diskType DiskType, ldbTimeout int64) error {
	rt, e := super_block.NewReplicaPlacementFromString(replicaPlacement)
	if e != nil {
		return e
	}
	ttl, e := needle.ReadTTL(ttlString)
	if e != nil {
		return e
	}
	e = s.addVolume(volumeId, collection, needleMapKind, rt, ttl, preallocate, ver, MemoryMapMaxSizeMb, diskType, ldbTimeout)
	return e
}

func (s *Store) DeleteCollection(collection string) (e error) {
	for _, location := range s.Locations {
		deleted, err := location.DeleteCollectionFromDiskLocation(collection)
		// Name every volume destroyed. Waiting for the next heartbeat to say so
		// by omission only works while heartbeats carry the whole list, and a
		// volume grown and destroyed between two of them was never reported at
		// all, so nothing else would ever tell the master its slot came free.
		for _, v := range deleted {
			s.DeletedVolumesChan <- &master_pb.VolumeShortInformationMessage{
				Id:               uint32(v.Id),
				Collection:       v.Collection,
				ReplicaPlacement: uint32(v.ReplicaPlacement.Byte()),
				Version:          uint32(v.Version()),
				Ttl:              v.Ttl.ToUint32(),
				DiskType:         string(location.DiskType),
				DiskId:           v.diskId,
			}
		}
		if err != nil {
			return err
		}
		stats.DeleteCollectionMetrics(collection)
	}
	return
}

func (s *Store) findVolume(vid needle.VolumeId) *Volume {
	for _, location := range s.Locations {
		if v, found := location.FindVolume(vid); found {
			return v
		}
	}
	return nil
}
func (s *Store) FindFreeLocation(filterFn func(location *DiskLocation) bool) (ret *DiskLocation) {
	max := int32(0)
	for _, location := range s.Locations {
		if filterFn != nil && !filterFn(location) {
			continue
		}
		if location.isDiskSpaceLow.Load() {
			continue
		}
		currentFreeCount := location.MaxVolumeCount - int32(location.VolumesLen())
		currentFreeCount *= erasure_coding.DataShardsCount
		currentFreeCount -= int32(location.EcShardCount())
		currentFreeCount /= erasure_coding.DataShardsCount
		if currentFreeCount > max {
			max = currentFreeCount
			ret = location
		}
	}
	return ret
}
func (s *Store) addVolume(vid needle.VolumeId, collection string, needleMapKind NeedleMapKind, replicaPlacement *super_block.ReplicaPlacement, ttl *needle.TTL, preallocate int64, ver needle.Version, memoryMapMaxSizeMb uint32, diskType DiskType, ldbTimeout int64) error {
	if s.findVolume(vid) != nil {
		return fmt.Errorf("Volume Id %d already exists!", vid)
	}

	// Find location with lowest local volume count (load balancing)
	var location *DiskLocation
	var diskId uint32
	var minVolCount int
	for i, loc := range s.Locations {
		if loc.DiskType == diskType && s.hasFreeDiskLocation(loc) {
			volCount := loc.LocalVolumesLen()
			if location == nil || volCount < minVolCount {
				location = loc
				diskId = uint32(i)
				minVolCount = volCount
			}
		}
	}

	if location != nil {
		glog.V(0).Infof("In dir %s (disk ID %d) adds volume:%v collection:%s replicaPlacement:%v ttl:%v",
			location.Directory, diskId, vid, collection, replicaPlacement, ttl)
		if volume, err := NewVolume(location.Directory, location.IdxDirectory, collection, vid, needleMapKind, replicaPlacement, ttl, preallocate, ver, memoryMapMaxSizeMb, ldbTimeout); err == nil {
			volume.diskId = diskId // Set the disk ID
			location.SetVolume(vid, volume)
			glog.V(0).Infof("add volume %d on disk ID %d", vid, diskId)
			s.NewVolumesChan <- &master_pb.VolumeShortInformationMessage{
				Id:               uint32(vid),
				Collection:       collection,
				ReplicaPlacement: uint32(replicaPlacement.Byte()),
				Version:          uint32(volume.Version()),
				Ttl:              ttl.ToUint32(),
				DiskType:         string(diskType),
				DiskId:           diskId,
			}
			return nil
		} else {
			return err
		}
	}
	return fmt.Errorf("No more free space left")
}

// hasFreeDiskLocation checks if a disk location has free space
func (s *Store) hasFreeDiskLocation(location *DiskLocation) bool {
	// Check if disk space is low first
	if location.isDiskSpaceLow.Load() {
		return false
	}

	// Check if disk is available
	if location.isDiskUnavailable.Load() {
		return false
	}

	// If MaxVolumeCount is 0, it means unlimited volumes are allowed
	if location.MaxVolumeCount == 0 {
		return true
	}

	// Check if current volume count is below the maximum
	return int64(location.VolumesLen()) < int64(location.MaxVolumeCount)
}

func (s *Store) VolumeInfos() (allStats []*VolumeInfo) {
	for _, location := range s.Locations {
		stats := collectStatsForOneLocation(location)
		allStats = append(allStats, stats...)
	}
	sortVolumeInfos(allStats)
	return allStats
}

func collectStatsForOneLocation(location *DiskLocation) (stats []*VolumeInfo) {
	location.volumesLock.RLock()
	defer location.volumesLock.RUnlock()

	for k, v := range location.volumes {
		s := collectStatForOneVolume(k, v)
		stats = append(stats, s)
	}
	return stats
}

func collectStatForOneVolume(vid needle.VolumeId, v *Volume) (s *VolumeInfo) {

	s = &VolumeInfo{
		Id:               vid,
		Collection:       v.Collection,
		ReplicaPlacement: v.ReplicaPlacement,
		Version:          v.Version(),
		ReadOnly:         v.IsReadOnly(),
		Ttl:              v.Ttl,
		CompactRevision:  uint32(v.CompactionRevision),
		DiskType:         v.DiskType().String(),
		DiskId:           v.diskId,
	}
	s.RemoteStorageName, _ = v.RemoteStorageNameKey()

	v.dataFileAccessLock.RLock()
	defer v.dataFileAccessLock.RUnlock()

	if v.nm == nil {
		return
	}

	s.FileCount = countAsUint32(uint64(v.nm.FileCount()))
	s.DeleteCount = countAsUint32(uint64(v.nm.DeletedCount()))
	s.DeletedByteCount = v.nm.DeletedSize()
	s.Size = v.nm.ContentSize()
	if v.DataBackend != nil {
		if _, modTime, e := v.DataBackend.GetStat(); e == nil {
			s.ModifiedAtSecond = modTime.Unix()
		}
	}
	return
}

func (s *Store) SetDataCenter(dataCenter string) {
	s.dataCenter = dataCenter
}
func (s *Store) SetRack(rack string) {
	s.rack = rack
}
func (s *Store) GetDataCenter() string {
	return s.dataCenter
}
func (s *Store) GetRack() string {
	return s.rack
}

func (s *Store) CollectHeartbeat() *master_pb.Heartbeat {
	s.collectHeartbeatLock.Lock()
	defer s.collectHeartbeatLock.Unlock()

	var volumeMessages []*master_pb.VolumeInformationMessage
	// Covers every volume held, whether or not this heartbeat names it, so the
	// master can tell whether applying what it was sent leaves it current.
	// Volumes skipped below -- quarantined, phantom, expired -- are in neither.
	var volumeDigest uint64
	sendFullList, reportGeneration, reportPass := s.volumeReport.begin()
	maxVolumeCounts := make(map[string]uint32)
	// Per-disk effective max for DiskTag, captured alongside the per-type sum.
	diskMaxByID := make(map[int]int32)
	diskTotalBytes := make(map[string]uint64)
	diskFreeBytes := make(map[string]uint64)
	var maxFileKey NeedleId
	collectionVolumeSize := make(map[string]int64)
	collectionVolumeDeletedBytes := make(map[string]int64)
	collectionVolumeReadOnlyCount := make(map[string]map[string]int)
	// Filled once per volume and kept only by the heartbeat that carries it, so
	// a server with nothing to say fills the same message all the way through.
	scratchMessage := &master_pb.VolumeInformationMessage{}
	for diskID, location := range s.Locations {
		if location.isDiskUnavailable.Load() {
			continue
		}
		var deleteVids []needle.VolumeId
		effectiveMaxCount := location.MaxVolumeCount
		if location.isDiskSpaceLow.Load() {
			usedSlots := int32(location.LocalVolumesLen())
			usedSlots += int32(erasure_coding.VolumeSlots(int64(location.EcShardCount())))
			effectiveMaxCount = usedSlots
		}
		if effectiveMaxCount < 0 {
			effectiveMaxCount = 0
		}
		maxVolumeCounts[string(location.DiskType)] += uint32(effectiveMaxCount)
		diskMaxByID[diskID] = effectiveMaxCount
		// Sum physical capacity per disk type. This assumes one location per
		// filesystem; if several -dir on one mount share a disk type, its total and
		// free are both counted once per location, so the used ratio the balance
		// gate relies on stays correct, but absolute capacity is over-reported.
		// Reporting per physical disk (mirroring max_volume_count_by_disk) is the
		// exact fix.
		diskTotalBytes[string(location.DiskType)] += location.diskTotalBytes.Load()
		diskFreeBytes[string(location.DiskType)] += location.diskFreeBytes.Load()
		location.volumesLock.RLock()
		for _, v := range location.volumes {
			curMaxFileKey, volumeMessage := v.ToVolumeInformationMessage(scratchMessage)
			if volumeMessage == nil {
				continue
			}
			if maxFileKey < curMaxFileKey {
				maxFileKey = curMaxFileKey
			}

			ioErr, ioCount, quarantined := v.getIoErrorState()
			if quarantined || (ioErr != nil && ioCount >= IoErrorTolerance) {
				// Sustained EIO: stop announcing this replica so the master
				// re-replicates from healthy peers, and mark it read-only
				// so further writes fail fast instead of producing more
				// EIOs. Never physically delete the data — the disk may
				// be transiently bad and this could be the last good
				// copy. The quarantine is sticky: a stray successful read
				// clears the streak counter but must not silently put a
				// known-bad replica back into rotation; recovery is via
				// MarkVolumeWritable.
				if !quarantined {
					glog.Warningf("volume %d quarantined after %d consecutive IO errors: %v",
						v.Id, ioCount, ioErr)
					v.markIoQuarantined()
				}
				v.noWriteLock.Lock()
				v.noWriteOrDelete = true
				v.noWriteLock.Unlock()
				// Skip per-volume size and read-only bookkeeping: a
				// quarantined replica should not be summed into the
				// collection's reported total nor counted in the
				// read-only stats.
				continue
			}

			shouldDeleteVolume := false
			if !v.expired(volumeMessage.Size, s.GetVolumeSizeLimit()) {
				reportHash := reportHashOf(volumeMessage)
				volumeDigest ^= reportHash
				if s.volumeReport.record(volumeMessage, reportHash, reportPass) || sendFullList {
					volumeMessages = append(volumeMessages, volumeMessage)
					scratchMessage = &master_pb.VolumeInformationMessage{}
				}
			} else {
				if v.expiredLongEnough(MAX_TTL_VOLUME_REMOVAL_DELAY) {
					deleteVids = append(deleteVids, v.Id)
					shouldDeleteVolume = true
				} else {
					glog.V(0).Infof("volume %d is expired", v.Id)
				}
			}

			// The totals are rebuilt from scratch every heartbeat, so a volume
			// on its way out is simply not added. Subtracting it took the
			// surviving volumes' sizes down with it, and an entry here is also
			// what says the collection is still on this server.
			if !shouldDeleteVolume {
				collectionVolumeSize[v.Collection] += int64(volumeMessage.Size)
				collectionVolumeDeletedBytes[v.Collection] += int64(volumeMessage.DeletedByteCount)

				counts, exist := collectionVolumeReadOnlyCount[v.Collection]
				if !exist {
					counts = map[string]int{
						stats.IsReadOnly:       0,
						stats.NoWriteOrDelete:  0,
						stats.NoWriteCanDelete: 0,
						stats.IsDiskSpaceLow:   0,
					}
					collectionVolumeReadOnlyCount[v.Collection] = counts
				}
				if readOnly, noWriteOrDelete, noWriteCanDelete, diskSpaceLow := v.ReadOnlyReasons(); readOnly {
					counts[stats.IsReadOnly] += 1
					if noWriteOrDelete {
						counts[stats.NoWriteOrDelete] += 1
					}
					if noWriteCanDelete {
						counts[stats.NoWriteCanDelete] += 1
					}
					if diskSpaceLow {
						counts[stats.IsDiskSpaceLow] += 1
					}
				}
			}
		}
		location.volumesLock.RUnlock()

		if len(deleteVids) > 0 {
			// delete expired volumes.
			location.volumesLock.Lock()
			for _, vid := range deleteVids {
				found, err := location.deleteVolumeById(vid, false, false)
				if err == nil {
					if found {
						glog.V(0).Infof("volume %d is deleted", vid)
					}
				} else {
					glog.Warningf("delete volume %d: %v", vid, err)
				}
			}
			location.volumesLock.Unlock()
		}
	}

	// delete expired ec volumes
	ecVolumeMessages, deletedEcVolumes := s.deleteExpiredEcVolumes()

	var uuidList []string
	for _, loc := range s.Locations {
		uuidList = append(uuidList, loc.DirectoryUuid)
	}
	var diskTags []*master_pb.DiskTag
	for diskID, loc := range s.Locations {
		diskTags = append(diskTags, &master_pb.DiskTag{
			DiskId:         uint32(diskID),
			Tags:           append([]string(nil), loc.Tags...),
			Type:           string(loc.DiskType),
			MaxVolumeCount: int64(diskMaxByID[diskID]),
		})
	}

	for col, size := range collectionVolumeSize {
		stats.VolumeServerDiskSizeGauge.WithLabelValues(col, "normal").Set(float64(size))
	}

	for col, deletedBytes := range collectionVolumeDeletedBytes {
		stats.VolumeServerDiskSizeGauge.WithLabelValues(col, "deleted_bytes").Set(float64(deletedBytes))
	}

	for col, types := range collectionVolumeReadOnlyCount {
		for t, count := range types {
			stats.VolumeServerReadOnlyVolumeGauge.WithLabelValues(col, t).Set(float64(count))
		}
	}

	// collectionVolumeReadOnlyCount has an entry for every collection that kept
	// a volume through this pass, including the ones counting zero read-only
	// volumes.
	for col := range s.reportedCollections {
		if _, stillHere := collectionVolumeReadOnlyCount[col]; !stillHere {
			stats.DeleteVolumeServerCollectionMetrics(col)
		}
	}
	s.reportedCollections = make(map[string]struct{}, len(collectionVolumeReadOnlyCount))
	for col := range collectionVolumeReadOnlyCount {
		s.reportedCollections[col] = struct{}{}
	}

	departedVolumes := s.volumeReport.commit(reportPass, reportGeneration, sendFullList)

	// has_no_volumes says the server holds nothing, so it may only be derived
	// from a full list. Deriving it from a changed-only heartbeat would make a
	// quiet one read as an empty server and drop every volume on it.
	heartbeatVolumes, changedVolumes := volumeMessages, []*master_pb.VolumeInformationMessage(nil)
	hasNoVolumes := len(volumeMessages) == 0
	if !sendFullList {
		heartbeatVolumes, changedVolumes = nil, volumeMessages
		hasNoVolumes = false
	}

	// The state rides along on every heartbeat, not only when it changes: the
	// first heartbeat to a master that just took the lead is how that master
	// learns this server is in maintenance mode.
	var state *volume_server_pb.VolumeServerState
	if s.State != nil {
		state = s.State.Proto()
	}

	return &master_pb.Heartbeat{
		Ip:              s.Ip,
		Port:            uint32(s.Port),
		GrpcPort:        uint32(s.GrpcPort),
		PublicUrl:       s.PublicUrl,
		Id:              s.Id,
		MaxVolumeCounts: maxVolumeCounts,
		DiskTotalBytes:  diskTotalBytes,
		DiskFreeBytes:   diskFreeBytes,
		MaxFileKey:      NeedleIdToUint64(maxFileKey),
		DataCenter:      s.dataCenter,
		Rack:            s.rack,
		Volumes:         heartbeatVolumes,
		ChangedVolumes:  changedVolumes,
		DeletedVolumes:  departedVolumes,
		VolumeDigest:    &volumeDigest,
		DeletedEcShards: deletedEcVolumes,
		HasNoVolumes:    hasNoVolumes,
		HasNoEcShards:   len(ecVolumeMessages) == 0,
		LocationUuids:   uuidList,
		DiskTags:        diskTags,
		State:           state,
	}

}

// reportHashOf digests a volume exactly as the master will digest what it
// stores for that volume, by running the master's own hash over the same
// conversion the master applies to the message.
func reportHashOf(m *master_pb.VolumeInformationMessage) uint64 {
	vi, err := NewVolumeInfo(m)
	if err != nil {
		glog.Warningf("volume %d: cannot digest heartbeat report: %v", m.Id, err)
		return 0
	}
	return vi.ReportHash()
}

// ResetVolumeReporting forgets what the master was told, so the next heartbeat
// carries the whole list. Called when a connection is established, since a
// reconnect may reach a master that knows nothing about this server.
func (s *Store) ResetVolumeReporting() {
	s.volumeReport.reset()
}

// AcceptVolumeChanges records that the master compares digests, so heartbeats
// may carry only what changed.
func (s *Store) AcceptVolumeChanges() {
	s.volumeReport.acceptDeltas()
}

// RequestFullVolumeList makes the next heartbeat carry the whole list.
func (s *Store) RequestFullVolumeList() {
	s.volumeReport.requestFullList()
}

func (s *Store) deleteExpiredEcVolumes() (ecShards, deleted []*master_pb.VolumeEcShardInformationMessage) {
	for diskId, location := range s.Locations {
		if location.isDiskUnavailable.Load() {
			continue
		}

		// Collect ecVolume to be deleted
		var toDeleteEvs []*erasure_coding.EcVolume
		location.ecVolumesLock.RLock()
		for _, ev := range location.ecVolumes {
			if ev.IsTimeToDestroy() {
				toDeleteEvs = append(toDeleteEvs, ev)
			} else {
				messages := ev.ToVolumeEcShardInformationMessage(uint32(diskId))
				ecShards = append(ecShards, messages...)
			}
		}
		location.ecVolumesLock.RUnlock()

		// Delete expired volumes
		for _, ev := range toDeleteEvs {
			messages := ev.ToVolumeEcShardInformationMessage(uint32(diskId))
			// deleteEcVolumeById has its own lock
			err := location.deleteEcVolumeById(ev.VolumeId)
			if err != nil {
				ecShards = append(ecShards, messages...)
				glog.Errorf("delete EcVolume err %d: %v", ev.VolumeId, err)
				continue
			}
			// No need for additional lock here since we only need the messages
			// from volumes that were already collected
			deleted = append(deleted, messages...)
		}
	}
	return
}

func (s *Store) SetStopping() {
	s.isStopping.Store(true)
	for _, location := range s.Locations {
		location.SetStopping()
	}
}

func (s *Store) IsStopping() bool {
	return s.isStopping.Load()
}

func (s *Store) LoadNewVolumes() {
	for _, location := range s.Locations {
		location.loadExistingVolumes(s.NeedleMapKind, 0)
	}
}

func (s *Store) Close() {
	for _, location := range s.Locations {
		location.Close()
	}
}

func (s *Store) WriteVolumeNeedle(i needle.VolumeId, n *needle.Needle, checkCookie bool, fsync bool) (isUnchanged bool, err error) {
	if v := s.findVolume(i); v != nil {
		if v.IsReadOnly() {
			err = fmt.Errorf("volume %d is read only", i)
			return
		}
		_, _, isUnchanged, err = v.writeNeedle2(n, checkCookie, fsync, s.isStopping.Load())
		return
	}
	glog.V(0).Infoln("volume", i, "not found!")
	err = fmt.Errorf("volume %d not found on %s:%d", i, s.Ip, s.Port)
	return
}

func (s *Store) DeleteVolumeNeedle(i needle.VolumeId, n *needle.Needle) (Size, error) {
	if v := s.findVolume(i); v != nil {
		if v.noWriteOrDelete {
			return 0, fmt.Errorf("volume %d is read only", i)
		}
		return v.deleteNeedle2(n)
	}
	return 0, fmt.Errorf("volume %d not found on %s:%d", i, s.Ip, s.Port)
}

func (s *Store) ReadVolumeNeedle(i needle.VolumeId, n *needle.Needle, readOption *ReadOption, onReadSizeFn func(size Size)) (int, error) {
	if v := s.findVolume(i); v != nil {
		return v.readNeedle(n, readOption, onReadSizeFn)
	}
	return 0, fmt.Errorf("volume %d not found", i)
}

func (s *Store) ReadVolumeNeedleMetaAt(i needle.VolumeId, n *needle.Needle, offset int64, size int32) error {
	if v := s.findVolume(i); v != nil {
		return v.readNeedleMetaAt(n, offset, size)
	}
	return fmt.Errorf("volume %d not found", i)
}

func (s *Store) ReadVolumeNeedleDataInto(i needle.VolumeId, n *needle.Needle, readOption *ReadOption, writer io.Writer, offset int64, size int64) error {
	if v := s.findVolume(i); v != nil {
		return v.readNeedleDataInto(n, readOption, writer, offset, size)
	}
	return fmt.Errorf("volume %d not found", i)
}
func (s *Store) GetVolume(i needle.VolumeId) *Volume {
	return s.findVolume(i)
}

func (s *Store) HasVolume(i needle.VolumeId) bool {
	v := s.findVolume(i)
	return v != nil
}

func (s *Store) MarkVolumeReadonly(i needle.VolumeId, canDelete bool, persist bool) error {
	v := s.findVolume(i)
	if v == nil {
		return fmt.Errorf("volume %d not found", i)
	}
	if canDelete && !v.HasRemoteFile() {
		// deletes append tombstones to .idx, which a readonly boot opened
		// O_RDONLY; remote volumes already delete through a RDWR idx
		if err := v.reopenIdxForWrite(); err != nil {
			return fmt.Errorf("volume %d reopen idx for write: %v", i, err)
		}
	}
	v.noWriteLock.Lock()
	v.noWriteOrDelete = !canDelete
	if canDelete {
		v.noWriteCanDelete = true
	} else if !v.HasRemoteFile() {
		// downgrading a canDelete mark; remote volumes keep their derived flag
		v.noWriteCanDelete = false
	}
	if persist {
		v.PersistReadOnly(true, canDelete)
	}
	v.noWriteLock.Unlock()
	return nil
}

func (s *Store) MarkVolumeWritable(i needle.VolumeId) error {
	v := s.findVolume(i)
	if v == nil {
		return fmt.Errorf("volume %d not found", i)
	}
	// If the volume booted with .vif ReadOnly=true, .idx is opened O_RDONLY
	// and v.nm is a SortedFileNeedleMap that rejects Put. Swap to writable
	// form before flipping the flag so the next write doesn't race past a
	// stale read-only handle.
	if err := v.reopenIdxForWrite(); err != nil {
		return fmt.Errorf("volume %d reopen idx for write: %v", i, err)
	}
	v.noWriteLock.Lock()
	v.noWriteOrDelete = false
	// Remote-tiered volumes must stay noWriteCanDelete regardless of marks.
	if !v.HasRemoteFile() {
		v.noWriteCanDelete = false
	}
	v.PersistReadOnly(false, false)
	v.noWriteLock.Unlock()
	// Clear the EIO streak and the sticky quarantine flag so the next
	// CollectHeartbeat can announce the volume again. If the disk is
	// still bad, the next failed op will re-arm the streak.
	v.resetIoErrorState()
	return nil
}

func (s *Store) MountVolume(i needle.VolumeId) error {
	for diskId, location := range s.Locations {
		if found := location.LoadVolume(uint32(diskId), i, s.NeedleMapKind); found == true {
			glog.V(0).Infof("mount volume %d", i)
			v := s.findVolume(i)
			v.diskId = uint32(diskId) // Set disk ID when mounting
			s.NewVolumesChan <- &master_pb.VolumeShortInformationMessage{
				Id:               uint32(v.Id),
				Collection:       v.Collection,
				ReplicaPlacement: uint32(v.ReplicaPlacement.Byte()),
				Version:          uint32(v.Version()),
				Ttl:              v.Ttl.ToUint32(),
				DiskType:         string(v.location.DiskType),
				DiskId:           uint32(diskId),
			}
			return nil
		}
	}

	return fmt.Errorf("volume %d not found on disk", i)
}

func (s *Store) UnmountVolume(i needle.VolumeId) error {
	// A volume id can be mounted on more than one disk of this server (e.g. a stale
	// twin re-attached after a disk repair, since NewStore has no cross-disk
	// duplicate guard). Unmount every copy, not just the first match, so a stale
	// twin cannot survive and re-register as the volume's content. A no-op unmount
	// (no copy present) is not an error, matching the prior behavior.
	var errs []error
	for _, location := range s.Locations {
		v, found := location.FindVolume(i)
		if !found {
			continue
		}
		message := master_pb.VolumeShortInformationMessage{
			Id:               uint32(v.Id),
			Collection:       v.Collection,
			ReplicaPlacement: uint32(v.ReplicaPlacement.Byte()),
			Version:          uint32(v.Version()),
			Ttl:              v.Ttl.ToUint32(),
			DiskType:         string(location.DiskType),
			DiskId:           v.diskId,
		}
		if err := location.UnloadVolume(i); err != nil {
			if err == ErrVolumeNotFound {
				continue
			}
			// Keep going so the other copies are still unmounted; surface the
			// failure so a copy left mounted is not reported as success.
			glog.Errorf("UnmountVolume %d on %s: %v", i, location.Directory, err)
			errs = append(errs, err)
			continue
		}
		glog.V(0).Infof("UnmountVolume %d disk_id:%d", i, v.diskId)
		s.DeletedVolumesChan <- &message
	}
	return errors.Join(errs...)
}

// ConsolidateVolumeIndex returns a volume's index to the configured -dir.idx
// directory when it is currently co-located with the data. A decode/reconstruct
// leaves the rebuilt .idx next to the .dat so the on-demand mount can find the
// volume while the old EC .ecx still coexists in the index directory; once the
// shards are gone this puts the index back on its own tier. It is a no-op when
// no separate index directory is configured or the index is already there.
//
// The relocation happens in place under the volume lock (see RelocateIndexTo),
// so the volume never leaves the mounted set and a concurrent read blocks
// briefly rather than failing.
func (s *Store) ConsolidateVolumeIndex(i needle.VolumeId) error {
	for _, location := range s.Locations {
		if v, found := location.FindVolume(i); found {
			if location.IdxDirectory == location.Directory {
				return nil
			}
			return v.RelocateIndexTo(location.IdxDirectory)
		}
	}
	return fmt.Errorf("volume %d not found on disk", i)
}

// RenameOrCopyFile moves src to dst, falling back to a copy when the two sit on
// different filesystems (os.Rename returns EXDEV across the data and -dir.idx
// disks, the separate media the flag exists to use).
func RenameOrCopyFile(src, dst string) error {
	if err := os.Rename(src, dst); err == nil {
		return nil
	} else if !errors.Is(err, syscall.EXDEV) {
		return err
	}
	in, err := os.Open(src)
	if err != nil {
		return err
	}
	defer in.Close()
	out, err := os.OpenFile(dst, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
	if err != nil {
		return err
	}
	if _, err := io.Copy(out, in); err != nil {
		out.Close()
		os.Remove(dst)
		return err
	}
	if err := out.Sync(); err != nil {
		out.Close()
		os.Remove(dst)
		return err
	}
	if err := out.Close(); err != nil {
		os.Remove(dst)
		return err
	}
	// Roll the copy back if the source cannot be removed, so a failure never
	// leaves two divergent copies (the loader would keep using the data-dir one
	// while the idx-dir orphan goes stale).
	if err := os.Remove(src); err != nil {
		os.Remove(dst)
		return err
	}
	return nil
}

func (s *Store) DeleteVolume(i needle.VolumeId, onlyEmpty bool, keepRemoteData bool) error {
	// Delete every copy of the volume id across disks, not just the first match, so
	// a stale twin (e.g. a re-attached disk; NewStore has no cross-disk duplicate
	// guard) cannot survive a delete and re-register as the volume's content.
	deletedAny := false
	var errs []error
	for _, location := range s.Locations {
		v, found := location.FindVolume(i)
		if !found {
			continue
		}
		message := master_pb.VolumeShortInformationMessage{
			Id:               uint32(v.Id),
			Collection:       v.Collection,
			ReplicaPlacement: uint32(v.ReplicaPlacement.Byte()),
			Version:          uint32(v.Version()),
			Ttl:              v.Ttl.ToUint32(),
			DiskType:         string(location.DiskType),
			DiskId:           v.diskId,
		}
		err := location.DeleteVolume(i, onlyEmpty, keepRemoteData)
		if err == nil {
			glog.V(0).Infof("DeleteVolume %d disk_id:%d", i, v.diskId)
			s.DeletedVolumesChan <- &message
			deletedAny = true
		} else if err == ErrVolumeNotFound {
			continue
		} else if err == ErrVolumeNotEmpty {
			// onlyEmpty: a non-empty copy aborts the delete rather than leaving a
			// partial result across disks.
			return fmt.Errorf("DeleteVolume %d: %w", i, err)
		} else {
			// A real failure on one disk must not be masked by another copy's
			// success: a stale copy left on the failing disk would re-register.
			glog.Errorf("DeleteVolume %d: %v", i, err)
			errs = append(errs, err)
		}
	}
	if len(errs) > 0 {
		return fmt.Errorf("DeleteVolume %d failed on some disks: %w", i, errors.Join(errs...))
	}
	if !deletedAny {
		return fmt.Errorf("delete volume %d not found on disk: %w", i, ErrVolumeNotFound)
	}
	return nil
}

func (s *Store) ConfigureVolume(i needle.VolumeId, replication string) error {

	for _, location := range s.Locations {
		fileInfo, found := location.LocateVolume(i)
		if !found {
			continue
		}
		// load, modify, save
		baseFileName := strings.TrimSuffix(fileInfo.Name(), filepath.Ext(fileInfo.Name()))
		vifFile := filepath.Join(location.Directory, baseFileName+".vif")
		volumeInfo, _, _, err := volume_info.MaybeLoadVolumeInfo(vifFile)
		if err != nil {
			return fmt.Errorf("volume %d failed to load vif: %v", i, err)
		}
		volumeInfo.Replication = replication
		err = volume_info.SaveVolumeInfo(vifFile, volumeInfo)
		if err != nil {
			return fmt.Errorf("volume %d failed to save vif: %v", i, err)
		}
		return nil
	}

	return fmt.Errorf("volume %d not found on disk", i)
}

func (s *Store) SetVolumeSizeLimit(x uint64) {
	atomic.StoreUint64(&s.volumeSizeLimit, x)
}

func (s *Store) GetVolumeSizeLimit() uint64 {
	return atomic.LoadUint64(&s.volumeSizeLimit)
}

func (s *Store) SetPreallocate(x bool) {
	s.preallocate.Store(x)
}

func (s *Store) GetPreallocate() bool {
	return s.preallocate.Load()
}

func (s *Store) MaybeAdjustVolumeMax() (hasChanges bool) {
	volumeSizeLimit := s.GetVolumeSizeLimit()
	if volumeSizeLimit == 0 {
		return
	}
	var newMaxVolumeCount int32
	for _, diskLocation := range s.Locations {
		if diskLocation.OriginalMaxVolumeCount == 0 {
			currentMaxVolumeCount := atomic.LoadInt32(&diskLocation.MaxVolumeCount)
			diskStatus := stats.NewDiskStatus(diskLocation.Directory)
			var unusedSpace uint64 = 0
			unclaimedSpaces := int64(diskStatus.Free)
			if !s.GetPreallocate() {
				unusedSpace = diskLocation.UnUsedSpace(volumeSizeLimit)
				unclaimedSpaces -= int64(unusedSpace)
			}
			volCount := diskLocation.VolumesLen()
			ecShardCount := diskLocation.EcShardCount()
			maxVolumeCount := int32(volCount) + int32((ecShardCount+erasure_coding.DataShardsCount-1)/erasure_coding.DataShardsCount)
			// One slot per full volume that fits in the unclaimed space.
			// A "- 1" here used to zero the count when the disk had room for
			// exactly one volume (free between 1x and 2x the limit), stranding
			// auto-sized disks at maxVolumeCount 0 with no writable volume.
			if unclaimedSpaces > 0 {
				maxVolumeCount += int32(uint64(unclaimedSpaces) / volumeSizeLimit)
			}
			// An auto-sized disk with free space always hosts at least one volume.
			if maxVolumeCount < 1 {
				maxVolumeCount = 1
			}
			newMaxVolumeCount = newMaxVolumeCount + maxVolumeCount
			atomic.StoreInt32(&diskLocation.MaxVolumeCount, maxVolumeCount)
			glog.V(4).Infof("disk %s max %d unclaimedSpace:%dMB, unused:%dMB volumeSizeLimit:%dMB",
				diskLocation.Directory, maxVolumeCount, unclaimedSpaces/1024/1024, unusedSpace/1024/1024, volumeSizeLimit/1024/1024)
			hasChanges = hasChanges || currentMaxVolumeCount != atomic.LoadInt32(&diskLocation.MaxVolumeCount)
		} else {
			newMaxVolumeCount = newMaxVolumeCount + diskLocation.OriginalMaxVolumeCount
		}
	}
	stats.VolumeServerMaxVolumeCounter.Set(float64(newMaxVolumeCount))
	return
}
