[skip ci] Add comments for exported things in policy.go (#11876)

Signed-off-by: shaoyue.chen <shaoyue.chen@zilliz.com>
pull/11910/head
shaoyue 2021-11-16 13:21:16 +08:00 committed by GitHub
parent 887f9e010f
commit 6322eb31e7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions

View File

@ -315,6 +315,7 @@ func AvgAssignUnregisteredChannels(store ROChannelStore, nodeID int64) ChannelOp
return opSet
}
// ConsistentHashDeregisterPolicy return a DeregisterPolicy that uses consistent hash
func ConsistentHashDeregisterPolicy(hashring *consistent.Consistent) DeregisterPolicy {
return func(store ROChannelStore, nodeID int64) ChannelOpSet {
hashring.Set(formatNodeIDsWithFilter(store.GetNodes(), nodeID))
@ -365,12 +366,15 @@ func ConsistentHashDeregisterPolicy(hashring *consistent.Consistent) DeregisterP
}
}
// ChannelReassignPolicy is a policy for reassigning channels
type ChannelReassignPolicy func(store ROChannelStore, reassigns []*NodeChannelInfo) ChannelOpSet
// EmptyReassignPolicy is a dummy reassign policy
func EmptyReassignPolicy(store ROChannelStore, reassigns []*NodeChannelInfo) ChannelOpSet {
return nil
}
// AverageReassignPolicy is a reassign policy that evenly assign channels
func AverageReassignPolicy(store ROChannelStore, reassigns []*NodeChannelInfo) ChannelOpSet {
channels := store.GetNodesChannels()
filterMap := make(map[int64]struct{})