volumemgr

type AllocVolumeCtx struct {
	Vids               []proto.Vid `json:"vids"`
	Host               string      `json:"host"`
	ExpireTime         int64       `json:"expire_time"`
	PendingAllocVolKey interface{} `json:"pending_alloc_vol_key"`
}

type ChangeVolStatusCtx struct {
	Vid      proto.Vid           `json:"vid"`
	TaskID   string              `json:"task_id"`
	TaskType base.VolumeTaskType `json:"type"`
}

type CreateVolumeCtx struct {
	VuInfos []*clustermgr.VolumeUnitInfo `json:"vu_infos"`
	VolInfo clustermgr.VolumeInfoBase    `json:"vol_info"`
	Vid     proto.Vid                    `json:"vid"`
}

type DeleteTaskCtx struct {
	Vid      proto.Vid           `json:"vid"`
	TaskType base.VolumeTaskType `json:"type"`
	TaskId   string              `json:"task_id"`
}

type NotifyFunc func(ctx context.Context, vol *volume) error

type VolumeMgr struct {
	VolumeMgrConfig
	// contains filtered or unexported fields
}

type VolumeMgrAPI interface {
	// GetVolumeInfo get volume  details info
	GetVolumeInfo(ctx context.Context, vid proto.Vid) (ret *cm.VolumeInfo, err error)

	// ListVolumeInfo list volumes
	ListVolumeInfo(ctx context.Context, args *cm.ListVolumeArgs) (ret []*cm.VolumeInfo, err error)

	// ListVolumeV2 list specified status volumes
	ListVolumeInfoV2(ctx context.Context, status proto.VolumeStatus) (ret []*cm.VolumeInfo, err error)

	// VolumeAlloc alloc volumes
	AllocVolume(ctx context.Context, mode codemode.CodeMode, count int, host string) (ret *cm.AllocatedVolumeInfos, err error)

	// ListAllocatedVolume list allocated volumes
	ListAllocatedVolume(ctx context.Context, host string, mode codemode.CodeMode) (ret *cm.AllocatedVolumeInfos)

	// PreUpdateVolumeUnit update volume
	PreUpdateVolumeUnit(ctx context.Context, args *cm.UpdateVolumeArgs) (err error)

	// PreRetainVolume retain volume
	PreRetainVolume(ctx context.Context, tokens []string, host string) (ret *cm.RetainVolumes, err error)

	// DiskWritableChange call when disk broken or heartbeat timeout or switch readonly, it'll refresh volume's health
	DiskWritableChange(ctx context.Context, diskID proto.DiskID) (err error)

	// AllocVolumeUnit alloc a new chunk to volume unit, it will increase volumeUnit's nextEpoch in memory
	AllocVolumeUnit(ctx context.Context, vuid proto.Vuid) (*cm.AllocVolumeUnit, error)

	// ReleaseVolumeUnit release old volume unit's chunk
	ReleaseVolumeUnit(ctx context.Context, vuid proto.Vuid, diskID proto.DiskID, force bool) (err error)

	// ListVolumeUnitInfo head all volume unit info in the disk
	ListVolumeUnitInfo(ctx context.Context, args *cm.ListVolumeUnitArgs) ([]*cm.VolumeUnitInfo, error)
	LockVolume(ctx context.Context, vid proto.Vid) error
	UnlockVolume(ctx context.Context, vid proto.Vid) error

	// Stat return volume statistic info
	Stat(ctx context.Context) (stat cm.VolumeStatInfo)
}

type VolumeMgrConfig struct {
	LocalHost      string          `json:"local_host"`
	BlobNodeConfig blobnode.Config `json:"blob_node_config"`
	// volume database path,include volumeTbl vuidTbl tokenTbl indexTbl taskTbl
	VolumeDBPath   string                `json:"volume_db_path"`
	VolumeDBOption kvstore.RocksDBOption `json:"volume_db_option"`
	// retain successful volume add another  retainTimeS second
	RetainTimeS                 int `json:"retain_time_s"`
	RetainThreshold             int `json:"retain_threshold"`
	FlushIntervalS              int `json:"flush_interval_s"`
	CheckExpiredVolumeIntervalS int `json:"check_expired_volume_interval_s"`

	VolumeSliceMapNum            uint32 `json:"volume_slice_map_num"`
	ApplyConcurrency             uint32 `json:"apply_concurrency"`
	MinAllocableVolumeCount      int    `json:"min_allocable_volume_count"`
	AllocatableDiskLoadThreshold int    `json:"allocatable_disk_load_threshold"`
	AllocFactor                  int    `json:"alloc_factor"`
	// the volume free size must big than AllocatableSize can alloc
	AllocatableSize uint64 `json:"allocatable_size"`

	// the volume in Proxy which free size small than FreezeThreshold treat filled
	FreezeThreshold  uint64            `json:"-"`
	IDC              []string          `json:"-"`
	UnavailableIDC   string            `json:"-"`
	ChunkSize        uint64            `json:"-"`
	CodeModePolicies []codemode.Policy `json:"-"`
	Region           string            `json:"-"`
	ClusterID        proto.ClusterID   `json:"-"`
}