milvus/internal/datacoord/channel_manager_factory.go

// Licensed to the LF AI & Data foundation under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package datacoord

// ChannelPolicyFactory is the abstract factory that creates policies for channel manager.
type ChannelPolicyFactory interface {
	// NewBalancePolicy creates a new channel balance policy.
	NewBalancePolicy() BalanceChannelPolicy

	NewAssignPolicy() AssignPolicy
}

// ChannelPolicyFactoryV1 equal to policy batch
type ChannelPolicyFactoryV1 struct{}

// NewChannelPolicyFactoryV1 helper function creates a Channel policy factory v1 from kv.
func NewChannelPolicyFactoryV1() *ChannelPolicyFactoryV1 {
	return &ChannelPolicyFactoryV1{}
}

func (f *ChannelPolicyFactoryV1) NewBalancePolicy() BalanceChannelPolicy {
	return AvgBalanceChannelPolicy
}

func (f *ChannelPolicyFactoryV1) NewAssignPolicy() AssignPolicy {
	return AvgAssignByCountPolicy
}

func NewEmptyChannelPolicyFactory() *EmptyChannelPolicyFactory {
	return &EmptyChannelPolicyFactory{}
}

type EmptyChannelPolicyFactory struct{}

func (f *EmptyChannelPolicyFactory) NewBalancePolicy() BalanceChannelPolicy {
	return EmptyBalancePolicy
}

func (f *EmptyChannelPolicyFactory) NewAssignPolicy() AssignPolicy {
	return EmptyAssignPolicy
}