diskmgr

type AllocPolicy struct {
	Idc      string
	Vuids    []proto.Vuid
	Excludes []proto.DiskID
}

type DiskMgr struct {
	DiskMgrConfig
	// contains filtered or unexported fields
}

type DiskMgrAPI interface {
	// AllocDiskID return a unused disk id
	AllocDiskID(ctx context.Context) (proto.DiskID, error)
	// GetDiskInfo return disk info, it return ErrDiskNotFound if disk not found
	GetDiskInfo(ctx context.Context, id proto.DiskID) (*blobnode.DiskInfo, error)
	// CheckDiskInfoDuplicated return true if disk info already exit, like host and path duplicated
	CheckDiskInfoDuplicated(ctx context.Context, info *blobnode.DiskInfo) bool
	// IsDiskWritable judge disk if writable, disk status unmoral or readonly or heartbeat timeout will return true
	IsDiskWritable(ctx context.Context, id proto.DiskID) (bool, error)
	// SetStatus change disk status, in some case, change status is not allow
	// like change repairing/repaired/dropped into normal
	SetStatus(ctx context.Context, id proto.DiskID, status proto.DiskStatus, isCommit bool) error
	// IsDroppingDisk return true if the specified disk is dropping
	IsDroppingDisk(ctx context.Context, id proto.DiskID) (bool, error)
	// ListDroppingDisk return all dropping disk info
	ListDroppingDisk(ctx context.Context) ([]*blobnode.DiskInfo, error)
	// AllocChunk return available chunks in data center
	AllocChunks(ctx context.Context, policy *AllocPolicy) ([]proto.DiskID, error)
	// ListDiskInfo
	ListDiskInfo(ctx context.Context, opt *clustermgr.ListOptionArgs) (*clustermgr.ListDiskRet, error)
	// Stat return disk statistic info of a cluster
	Stat(ctx context.Context) *clustermgr.SpaceStatInfo
	// SwitchReadonly can switch disk's readonly or writable
	SwitchReadonly(diskID proto.DiskID, readonly bool) error
	// GetHeartbeatChangeDisks return any heartbeat change disks
	GetHeartbeatChangeDisks() []HeartbeatEvent
}

type DiskMgrConfig struct {
	RefreshIntervalS         int             `json:"refresh_interval_s"`
	RackAware                bool            `json:"rack_aware"`
	HostAware                bool            `json:"host_aware"`
	HeartbeatExpireIntervalS int             `json:"heartbeat_expire_interval_s"`
	FlushIntervalS           int             `json:"flush_interval_s"`
	ApplyConcurrency         uint32          `json:"apply_concurrency"`
	BlobNodeConfig           blobnode.Config `json:"blob_node_config"`
	AllocTolerateBuffer      int64           `json:"alloc_tolerate_buffer"`
	EnsureIndex              bool            `json:"ensure_index"`

	IDC       []string            `json:"-"`
	CodeModes []codemode.CodeMode `json:"-"`
	ChunkSize int64               `json:"-"`
}

type HeartbeatEvent struct {
	DiskID  proto.DiskID
	IsAlive bool
}