mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
cb7f2fa6fd
Related to #39095 https://go.dev/doc/modules/version-numbers Update pkg version according to golang dep version convention --------- Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
31 lines
1002 B
Go
31 lines
1002 B
Go
package util
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/milvus-io/milvus/pkg/v2/util/paramtable"
|
|
"github.com/milvus-io/milvus/pkg/v2/util/typeutil"
|
|
)
|
|
|
|
// GetAllTopicsFromConfiguration gets all topics from configuration.
|
|
// It's a utility function to fetch all topics from configuration.
|
|
func GetAllTopicsFromConfiguration() typeutil.Set[string] {
|
|
var channels typeutil.Set[string]
|
|
if paramtable.Get().CommonCfg.PreCreatedTopicEnabled.GetAsBool() {
|
|
channels = typeutil.NewSet[string](paramtable.Get().CommonCfg.TopicNames.GetAsStrings()...)
|
|
} else {
|
|
channels = genChannelNames(paramtable.Get().CommonCfg.RootCoordDml.GetValue(), paramtable.Get().RootCoordCfg.DmlChannelNum.GetAsInt())
|
|
}
|
|
return channels
|
|
}
|
|
|
|
// genChannelNames generates channel names with prefix and number.
|
|
func genChannelNames(prefix string, num int) typeutil.Set[string] {
|
|
results := typeutil.NewSet[string]()
|
|
for idx := 0; idx < num; idx++ {
|
|
result := fmt.Sprintf("%s_%d", prefix, idx)
|
|
results.Insert(result)
|
|
}
|
|
return results
|
|
}
|