mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 03:13:22 +08:00
f18aa85083
issue: #40638 - Add `ChannelID` for streaming replica in future. - Remove the pchannel count fair balance policy for streaming. - Add Score based vchannel fair balance policy for streaming. - Add pchannel stats manager to collect the stats of pchannel for balancer. - Add configuration and metrics for new balance policy --------- Signed-off-by: chyezh <chyezh@outlook.com>
39 lines
1.0 KiB
Go
39 lines
1.0 KiB
Go
package channel
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/milvus-io/milvus/pkg/v2/streaming/util/types"
|
|
)
|
|
|
|
// ChannelID is the unique id of a channel.
|
|
type ChannelID = types.ChannelID
|
|
|
|
// newPChannelView creates a new pchannel view.
|
|
func newPChannelView(metas map[ChannelID]*PChannelMeta) *PChannelView {
|
|
view := &PChannelView{
|
|
Channels: make(map[ChannelID]*PChannelMeta, len(metas)),
|
|
Stats: make(map[ChannelID]PChannelStatsView, len(metas)),
|
|
}
|
|
for _, meta := range metas {
|
|
id := meta.ChannelInfo().ChannelID()
|
|
if _, ok := view.Channels[id]; ok {
|
|
panic(fmt.Sprintf("duplicate rw channel: %s", id.String()))
|
|
}
|
|
view.Channels[id] = meta
|
|
view.Stats[id] = StaticPChannelStatsManager.MustGet().GetPChannelStats(id).View()
|
|
}
|
|
return view
|
|
}
|
|
|
|
// PChannelView is the view of current pchannels.
|
|
type PChannelView struct {
|
|
Channels map[ChannelID]*PChannelMeta
|
|
Stats map[ChannelID]PChannelStatsView
|
|
}
|
|
|
|
// PChannelStatsView is the view of the pchannel stats.
|
|
type PChannelStatsView struct {
|
|
VChannels map[string]int64
|
|
}
|