raftstore

type Epoch = metapb.ResourceEpoch

type FailureCallback func(requestID []byte, err error)

type LogReader struct {
	sync.Mutex
	// contains filtered or unexported fields
}

type Replica = metapb.Replica

type RetryController interface {
	// Retry used to control retry if retryable error encountered. returns false means stop retry.
	Retry(requestID []byte) (rpc.Request, bool)
}

type Router interface {
	// Start the router
	Start() error
	// Stop stops the router
	Stop()
	// SelectShard returns a shard and leader store that the key is in the range [shard.Start, shard.End).
	// If returns leader address is "", means the current shard has no leader
	SelectShard(group uint64, key []byte) (Shard, string)
	// Every do with all shards
	Every(group uint64, mustLeader bool, fn func(shard Shard, store meta.Store) bool)
	// ForeachShards foreach shards
	ForeachShards(group uint64, fn func(shard Shard) bool)
	// GetShard returns the shard by shard id
	GetShard(id uint64) Shard
	// UpdateLeader update shard leader
	UpdateLeader(shardID uint64, leaderReplciaID uint64)

	// LeaderStore return leader replica store
	LeaderReplicaStore(shardID uint64) meta.Store
	// RandomReplicaStore return random replica store
	RandomReplicaStore(shardID uint64) meta.Store

	// GetShardStats returns the runtime stats info of the shard
	GetShardStats(id uint64) metapb.ResourceStats
	// GetStoreStats returns the runtime stats info of the store
	GetStoreStats(id uint64) metapb.ContainerStats
}

type Shard = meta.Shard

type ShardsPool interface {
	// Alloc alloc a shard from shards pool, returns error if no idle shards left. The `purpose` is used to avoid
	// duplicate allocation.
	Alloc(group uint64, purpose []byte) (meta.AllocatedShard, error)
}

type ShardsProxy interface {
	Start() error
	Stop() error
	Dispatch(req rpc.Request) error
	DispatchTo(req rpc.Request, shard Shard, store string) error
	SetCallback(SuccessCallback, FailureCallback)
	SetRetryController(retryController RetryController)
	OnResponse(rpc.ResponseBatch)
	Router() Router
}

type Store interface {
	// Start the raft store
	Start()
	// Stop the raft store
	Stop()
	// GetConfig returns the config of the store
	GetConfig() *config.Config
	// Meta returns store meta
	Meta() meta.Store
	// GetRouter returns a router
	GetRouter() Router
	// GetShardsProxy get shards proxy to dispatch requests
	GetShardsProxy() ShardsProxy
	// OnRequest receive a request, and call cb while the request is completed
	OnRequest(rpc.Request) error
	// OnRequestWithCB receive a request, and call cb while the request is completed
	OnRequestWithCB(req rpc.Request, cb func(resp rpc.ResponseBatch)) error
	// DataStorage returns a DataStorage of the shard group
	DataStorageByGroup(uint64) storage.DataStorage
	// MaybeLeader returns the shard replica maybe leader
	MaybeLeader(uint64) bool
	// AllocID returns a uint64 id, panic if has a error
	MustAllocID() uint64
	// Prophet return current prophet instance
	Prophet() prophet.Prophet

	// CreateResourcePool create resource pools, the resource pool will create shards,
	// and try to maintain the number of shards in the pool not less than the `capacity`
	// parameter. This is an idempotent operation.
	CreateResourcePool(...metapb.ResourcePool) (ShardsPool, error)
	// GetResourcePool returns `ShardsPool`, nil if `CreateResourcePool` not completed
	GetResourcePool() ShardsPool
}

type SuccessCallback func(resp rpc.Response)

type TestClusterOption func(*testClusterOptions)

type TestDataBuilder struct {
}

type TestKVClient interface {
	// Set set key-value to the backend kv storage
	Set(key, value string, timeout time.Duration) error
	// Get returns the value of the specific key from backend kv storage
	Get(key string, timeout time.Duration) (string, error)
	// UpdateLabel update the shard label
	UpdateLabel(shard, group uint64, key, value string, timeout time.Duration) error
	// Close close the test client
	Close()
}

type TestRaftCluster interface {
	// EveryStore do fn at every store, it can be used to init some store register
	EveryStore(fn func(i int, store Store))
	// GetStore returns the node store
	GetStore(node int) Store
	// GetStoreByID returns the store
	GetStoreByID(id uint64) Store
	// Start start each node sequentially
	Start()
	// Stop stop each node sequentially
	Stop()
	// StartWithConcurrent after starting the first node, other nodes start concurrently
	StartWithConcurrent(bool)
	// Restart restart the cluster
	Restart()
	// RestartWithFunc restart the cluster, `beforeStartFunc` is called before starting
	RestartWithFunc(beforeStartFunc func())
	// StartNode start the node
	StartNode(node int)
	// StopNode stop the node
	StopNode(node int)
	// RestartNode restart the node
	RestartNode(node int)
	// StartNetworkPartition node will in network partition, must call after node started
	StartNetworkPartition(partitions [][]int)
	// StopNetworkPartition stop network partition
	StopNetworkPartition()
	// GetPRCount returns the number of replicas on the node
	GetPRCount(node int) int
	// GetShardByIndex returns the shard by `shardIndex`, `shardIndex` is the order in which
	// the shard is created on the node
	GetShardByIndex(node int, shardIndex int) Shard
	// GetShardByID returns the shard from the node by shard id
	GetShardByID(node int, shardID uint64) Shard
	// CheckShardCount check whether the number of shards on each node is correct
	CheckShardCount(countPerNode int)
	// CheckShardRange check whether the range field of the shard on each node is correct,
	// `shardIndex` is the order in which the shard is created on the node
	CheckShardRange(shardIndex int, start, end []byte)
	// WaitRemovedByShardID check whether the specific shard removed from every node until timeout
	WaitRemovedByShardID(shardID uint64, timeout time.Duration)
	// WaitRemovedByShardIDAt check whether the specific shard removed from specific node until timeout
	WaitRemovedByShardIDAt(shardID uint64, nodes []int, timeout time.Duration)
	// WaitLeadersByCount check that the number of leaders of the cluster reaches at least the specified value
	// until the timeout
	WaitLeadersByCount(count int, timeout time.Duration)
	// WaitLeadersByCountsAndShardGroupAndLabel check that the number of leaders of the cluster reaches at least the specified value
	// until the timeout
	WaitLeadersByCountsAndShardGroupAndLabel(counts []int, group uint64, key, value string, timeout time.Duration)
	// WaitShardByCount check that the number of shard of the cluster reaches at least the specified value
	// until the timeout
	WaitShardByCount(count int, timeout time.Duration)
	// WaitShardByLabel check that the shard has the specified label until the timeout
	WaitShardByLabel(id uint64, label, value string, timeout time.Duration)
	// WaitVoterReplicaByCount check that the number of voter shard of the cluster reaches at least the specified value
	// until the timeout
	WaitVoterReplicaByCountPerNode(count int, timeout time.Duration)
	// WaitVoterReplicaByCounts check that the number of voter shard of the cluster reaches at least the specified value
	// until the timeout
	WaitVoterReplicaByCounts(counts []int, timeout time.Duration)
	// WaitVoterReplicaByCountsAndShardGroup check that the number of voter shard of the cluster reaches at least the specified value
	// until the timeout
	WaitVoterReplicaByCountsAndShardGroup(counts []int, shardGroup uint64, timeout time.Duration)
	// WaitVoterReplicaByCountsAndShardGroupAndLabel check that the number of voter shard of the cluster reaches at least the specified value
	// until the timeout
	WaitVoterReplicaByCountsAndShardGroupAndLabel(counts []int, shardGroup uint64, label, value string, timeout time.Duration)
	// WaitShardByCountPerNode check that the number of shard of each node reaches at least the specified value
	// until the timeout
	WaitShardByCountPerNode(count int, timeout time.Duration)
	// WaitAllReplicasChangeToVoter check that the role of shard of each node change to voter until the timeout
	WaitAllReplicasChangeToVoter(shard uint64, timeout time.Duration)
	// WaitShardByCountOnNode check that the number of shard of the specified node reaches at least the specified value
	// until the timeout
	WaitShardByCountOnNode(node, count int, timeout time.Duration)
	// WaitShardSplitByCount check whether the count of shard split reaches a specific value until timeout
	WaitShardSplitByCount(id uint64, count int, timeout time.Duration)
	// WaitShardByCounts check whether the number of shards reaches a specific value until timeout
	WaitShardByCounts(counts []int, timeout time.Duration)
	// WaitShardStateChangedTo check whether the state of shard changes to the specific value until timeout
	WaitShardStateChangedTo(shardID uint64, to metapb.ResourceState, timeout time.Duration)
	// GetShardLeaderStore return the leader node of the shard
	GetShardLeaderStore(shardID uint64) Store
	// GetProphet returns the prophet instance
	GetProphet() prophet.Prophet
	// CreateTestKVClient create and returns a kv client
	CreateTestKVClient(node int) TestKVClient
}