proxy

type AllocRet struct {
	BidStart proto.BlobID `json:"bid_start"`
	BidEnd   proto.BlobID `json:"bid_end"`
	Vid      proto.Vid    `json:"vid"`
}

type AllocVolsArgs struct {
	Fsize    uint64            `json:"fsize"`
	CodeMode codemode.CodeMode `json:"code_mode"`
	BidCount uint64            `json:"bid_count"`
	Excludes []proto.Vid       `json:"excludes"`
	Discards []proto.Vid       `json:"discards"`
}

type Allocator interface {
	VolumeAlloc(ctx context.Context, host string, args *AllocVolsArgs) (ret []AllocRet, err error)
	ListVolumes(ctx context.Context, host string, args *ListVolsArgs) (ret VolumeList, err error)
}

type BlobDelete struct {
	Bid proto.BlobID `json:"bid"`
	Vid proto.Vid    `json:"vid"`
}

type CacheDiskArgs struct {
	DiskID proto.DiskID `json:"disk_id"`
	Flush  bool         `json:"flush,omitempty"`
}

type CacheVolumeArgs struct {
	Vid     proto.Vid `json:"vid"`
	Version uint32    `json:"version,omitempty"`
	Flush   bool      `json:"flush,omitempty"`
}

type Cacher interface {
	GetCacheVolume(ctx context.Context, host string, args *CacheVolumeArgs) (*VersionVolume, error)
	GetCacheDisk(ctx context.Context, host string, args *CacheDiskArgs) (*blobnode.DiskInfo, error)
	// Erase cache in proxy memory and diskv.
	// Volume key is "volume-{vid}", and disk key is "disk-{disk_id}".
	// Notice: Erase all if key is "ALL"!
	Erase(ctx context.Context, host string, key string) error
}

type Client interface {
	MsgSender
	Allocator
	Cacher
}

type Config struct {
	rpc.Config
}

type DeleteArgs struct {
	ClusterID proto.ClusterID `json:"cluster_id"`
	Blobs     []BlobDelete    `json:"blobs"`
}

type DiscardVolsArgs struct {
	CodeMode codemode.CodeMode `json:"code_mode"`
	Discards []proto.Vid       `json:"discards"`
}

type LbConfig struct {
	Config

	HostRetry          int   `json:"host_retry"`
	HostSyncIntervalMs int64 `json:"host_sync_interval_ms"`
}

type LbMsgSender interface {
	SendDeleteMsg(ctx context.Context, args *DeleteArgs) error
	SendShardRepairMsg(ctx context.Context, args *ShardRepairArgs) error
}

type ListVolsArgs struct {
	CodeMode codemode.CodeMode `json:"code_mode"`
}

type MsgSender interface {
	SendDeleteMsg(ctx context.Context, host string, args *DeleteArgs) error
	SendShardRepairMsg(ctx context.Context, host string, args *ShardRepairArgs) error
}

type ShardRepairArgs struct {
	ClusterID proto.ClusterID `json:"cluster_id"`
	Bid       proto.BlobID    `json:"bid"`
	Vid       proto.Vid       `json:"vid"`
	BadIdxes  []uint8         `json:"bad_idxes"`
	Reason    string          `json:"reason"`
}

type VersionVolume struct {
	clustermgr.VolumeInfo
	Version uint32 `json:"version,omitempty"`
}

type VolumeList struct {
	Vids    []proto.Vid                  `json:"vids"`
	Volumes []clustermgr.AllocVolumeInfo `json:"volumes"`
}