2021-12-17 10:23:15 +08:00
|
|
|
// Copyright (C) 2019-2020 Zilliz. All rights reserved.
|
|
|
|
//
|
|
|
|
// Licensed 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 paramtable
|
|
|
|
|
|
|
|
import (
|
|
|
|
"math"
|
2021-12-23 18:39:11 +08:00
|
|
|
"os"
|
|
|
|
"path"
|
2021-12-17 10:23:15 +08:00
|
|
|
"strconv"
|
2021-12-23 18:39:11 +08:00
|
|
|
"strings"
|
2021-12-17 10:23:15 +08:00
|
|
|
"sync"
|
2021-12-23 18:39:11 +08:00
|
|
|
"time"
|
2021-12-17 10:23:15 +08:00
|
|
|
|
2021-12-23 18:39:11 +08:00
|
|
|
"go.uber.org/zap"
|
2022-04-07 22:05:32 +08:00
|
|
|
|
|
|
|
"github.com/milvus-io/milvus/internal/log"
|
2021-12-17 10:23:15 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2021-12-29 21:29:45 +08:00
|
|
|
// DefaultRetentionDuration defines the default duration for retention which is 5 days in seconds.
|
|
|
|
DefaultRetentionDuration = 3600 * 24 * 5
|
2022-04-08 20:29:33 +08:00
|
|
|
|
|
|
|
// DefaultIndexSliceSize defines the default slice size of index file when serializing.
|
|
|
|
DefaultIndexSliceSize = 4
|
2021-12-17 10:23:15 +08:00
|
|
|
)
|
|
|
|
|
2022-02-09 14:41:45 +08:00
|
|
|
// ComponentParam is used to quickly and easily access all components' configurations.
|
2022-02-08 20:57:47 +08:00
|
|
|
type ComponentParam struct {
|
|
|
|
ServiceParam
|
2022-02-07 10:09:45 +08:00
|
|
|
once sync.Once
|
2022-01-10 19:03:35 +08:00
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
CommonCfg commonConfig
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
RootCoordCfg rootCoordConfig
|
|
|
|
ProxyCfg proxyConfig
|
|
|
|
QueryCoordCfg queryCoordConfig
|
|
|
|
QueryNodeCfg queryNodeConfig
|
|
|
|
DataCoordCfg dataCoordConfig
|
|
|
|
DataNodeCfg dataNodeConfig
|
|
|
|
IndexCoordCfg indexCoordConfig
|
|
|
|
IndexNodeCfg indexNodeConfig
|
|
|
|
}
|
|
|
|
|
|
|
|
// InitOnce initialize once
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *ComponentParam) InitOnce() {
|
2021-12-23 18:39:11 +08:00
|
|
|
p.once.Do(func() {
|
|
|
|
p.Init()
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2022-01-06 13:51:21 +08:00
|
|
|
// Init initialize the global param table
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *ComponentParam) Init() {
|
|
|
|
p.ServiceParam.Init()
|
2022-01-10 14:51:34 +08:00
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
p.CommonCfg.init(&p.BaseTable)
|
2022-01-10 17:29:34 +08:00
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
p.RootCoordCfg.init(&p.BaseTable)
|
|
|
|
p.ProxyCfg.init(&p.BaseTable)
|
|
|
|
p.QueryCoordCfg.init(&p.BaseTable)
|
|
|
|
p.QueryNodeCfg.init(&p.BaseTable)
|
|
|
|
p.DataCoordCfg.init(&p.BaseTable)
|
|
|
|
p.DataNodeCfg.init(&p.BaseTable)
|
|
|
|
p.IndexCoordCfg.init(&p.BaseTable)
|
|
|
|
p.IndexNodeCfg.init(&p.BaseTable)
|
2022-01-10 17:29:34 +08:00
|
|
|
}
|
|
|
|
|
2022-02-07 10:09:45 +08:00
|
|
|
// SetLogConfig set log config with given role
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *ComponentParam) SetLogConfig(role string) {
|
2022-02-07 10:09:45 +08:00
|
|
|
p.BaseTable.RoleName = role
|
|
|
|
p.BaseTable.SetLogConfig()
|
2022-01-10 17:29:34 +08:00
|
|
|
}
|
|
|
|
|
2022-03-24 10:15:25 +08:00
|
|
|
func (p *ComponentParam) RocksmqEnable() bool {
|
|
|
|
return p.RocksmqCfg.Path != ""
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *ComponentParam) PulsarEnable() bool {
|
|
|
|
return p.PulsarCfg.Address != ""
|
|
|
|
}
|
|
|
|
|
2022-01-10 14:51:34 +08:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
2021-12-23 18:39:11 +08:00
|
|
|
// --- common ---
|
2022-01-10 19:03:35 +08:00
|
|
|
type commonConfig struct {
|
2022-02-08 20:57:47 +08:00
|
|
|
Base *BaseTable
|
2022-01-10 19:03:35 +08:00
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
ClusterPrefix string
|
2021-12-23 18:39:11 +08:00
|
|
|
|
2022-02-09 14:41:45 +08:00
|
|
|
ProxySubName string
|
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
RootCoordTimeTick string
|
|
|
|
RootCoordStatistics string
|
|
|
|
RootCoordDml string
|
|
|
|
RootCoordDelta string
|
|
|
|
RootCoordSubName string
|
2021-12-23 18:39:11 +08:00
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
QueryCoordSearch string
|
|
|
|
QueryCoordSearchResult string
|
|
|
|
QueryCoordTimeTick string
|
|
|
|
QueryNodeStats string
|
2022-02-09 14:41:45 +08:00
|
|
|
QueryNodeSubName string
|
2021-12-23 18:39:11 +08:00
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
DataCoordStatistic string
|
|
|
|
DataCoordTimeTick string
|
|
|
|
DataCoordSegmentInfo string
|
|
|
|
DataCoordSubName string
|
2022-02-09 14:41:45 +08:00
|
|
|
DataNodeSubName string
|
2022-03-04 11:17:56 +08:00
|
|
|
|
|
|
|
DefaultPartitionName string
|
|
|
|
DefaultIndexName string
|
|
|
|
RetentionDuration int64
|
|
|
|
|
2022-04-08 20:29:33 +08:00
|
|
|
SimdType string
|
|
|
|
IndexSliceSize int64
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) init(base *BaseTable) {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.Base = base
|
2021-12-23 18:39:11 +08:00
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
// must init cluster prefix first
|
|
|
|
p.initClusterPrefix()
|
2021-12-23 18:39:11 +08:00
|
|
|
|
2022-02-09 14:41:45 +08:00
|
|
|
p.initProxySubName()
|
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
p.initRootCoordTimeTick()
|
|
|
|
p.initRootCoordStatistics()
|
|
|
|
p.initRootCoordDml()
|
|
|
|
p.initRootCoordDelta()
|
|
|
|
p.initRootCoordSubName()
|
|
|
|
|
|
|
|
p.initQueryCoordSearch()
|
|
|
|
p.initQueryCoordSearchResult()
|
|
|
|
p.initQueryCoordTimeTick()
|
|
|
|
p.initQueryNodeStats()
|
2022-02-09 14:41:45 +08:00
|
|
|
p.initQueryNodeSubName()
|
2022-02-02 00:35:43 +08:00
|
|
|
|
|
|
|
p.initDataCoordStatistic()
|
|
|
|
p.initDataCoordTimeTick()
|
|
|
|
p.initDataCoordSegmentInfo()
|
|
|
|
p.initDataCoordSubName()
|
2022-02-09 14:41:45 +08:00
|
|
|
p.initDataNodeSubName()
|
2022-03-04 11:17:56 +08:00
|
|
|
|
|
|
|
p.initDefaultPartitionName()
|
|
|
|
p.initDefaultIndexName()
|
|
|
|
p.initRetentionDuration()
|
|
|
|
|
|
|
|
p.initSimdType()
|
2022-04-08 20:29:33 +08:00
|
|
|
p.initIndexSliceSize()
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initClusterPrefix() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.cluster",
|
|
|
|
"msgChannel.chanNamePrefix.cluster",
|
|
|
|
}
|
|
|
|
str, err := p.Base.Load2(keys)
|
2021-12-23 18:39:11 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2022-02-02 00:35:43 +08:00
|
|
|
p.ClusterPrefix = str
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initChanNamePrefix(keys []string) string {
|
|
|
|
value, err := p.Base.Load2(keys)
|
2021-12-23 18:39:11 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2022-02-02 00:35:43 +08:00
|
|
|
s := []string{p.ClusterPrefix, value}
|
|
|
|
return strings.Join(s, "-")
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
2022-02-09 14:41:45 +08:00
|
|
|
// --- proxy ---
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initProxySubName() {
|
|
|
|
keys := []string{
|
|
|
|
"common.subNamePrefix.proxySubNamePrefix",
|
|
|
|
"msgChannel.subNamePrefix.proxySubNamePrefix",
|
|
|
|
}
|
|
|
|
p.ProxySubName = p.initChanNamePrefix(keys)
|
2022-02-09 14:41:45 +08:00
|
|
|
}
|
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
// --- rootcoord ---
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initRootCoordTimeTick() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.rootCoordTimeTick",
|
|
|
|
"msgChannel.chanNamePrefix.rootCoordTimeTick",
|
|
|
|
}
|
|
|
|
p.RootCoordTimeTick = p.initChanNamePrefix(keys)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initRootCoordStatistics() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.rootCoordStatistics",
|
|
|
|
"msgChannel.chanNamePrefix.rootCoordStatistics",
|
|
|
|
}
|
|
|
|
p.RootCoordStatistics = p.initChanNamePrefix(keys)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initRootCoordDml() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.rootCoordDml",
|
|
|
|
"msgChannel.chanNamePrefix.rootCoordDml",
|
|
|
|
}
|
|
|
|
p.RootCoordDml = p.initChanNamePrefix(keys)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initRootCoordDelta() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.rootCoordDelta",
|
|
|
|
"msgChannel.chanNamePrefix.rootCoordDelta",
|
|
|
|
}
|
|
|
|
p.RootCoordDelta = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initRootCoordSubName() {
|
|
|
|
keys := []string{
|
|
|
|
"common.subNamePrefix.rootCoordSubNamePrefix",
|
|
|
|
"msgChannel.subNamePrefix.rootCoordSubNamePrefix",
|
|
|
|
}
|
|
|
|
p.RootCoordSubName = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// --- querycoord ---
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initQueryCoordSearch() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.search",
|
|
|
|
"msgChannel.chanNamePrefix.search",
|
|
|
|
}
|
|
|
|
p.QueryCoordSearch = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initQueryCoordSearchResult() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.searchResult",
|
|
|
|
"msgChannel.chanNamePrefix.searchResult",
|
|
|
|
}
|
|
|
|
p.QueryCoordSearchResult = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initQueryCoordTimeTick() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.queryTimeTick",
|
|
|
|
"msgChannel.chanNamePrefix.queryTimeTick",
|
|
|
|
}
|
|
|
|
p.QueryCoordTimeTick = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// --- querynode ---
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initQueryNodeStats() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.queryNodeStats",
|
|
|
|
"msgChannel.chanNamePrefix.queryNodeStats",
|
|
|
|
}
|
|
|
|
p.QueryNodeStats = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initQueryNodeSubName() {
|
|
|
|
keys := []string{
|
|
|
|
"common.subNamePrefix.queryNodeSubNamePrefix",
|
|
|
|
"msgChannel.subNamePrefix.queryNodeSubNamePrefix",
|
|
|
|
}
|
|
|
|
p.QueryNodeSubName = p.initChanNamePrefix(keys)
|
2022-02-09 14:41:45 +08:00
|
|
|
}
|
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
// --- datacoord ---
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initDataCoordStatistic() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.dataCoordStatistic",
|
|
|
|
"msgChannel.chanNamePrefix.dataCoordStatistic",
|
|
|
|
}
|
|
|
|
p.DataCoordStatistic = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initDataCoordTimeTick() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.dataCoordTimeTick",
|
|
|
|
"msgChannel.chanNamePrefix.dataCoordTimeTick",
|
|
|
|
}
|
|
|
|
p.DataCoordTimeTick = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initDataCoordSegmentInfo() {
|
|
|
|
keys := []string{
|
|
|
|
"common.chanNamePrefix.dataCoordSegmentInfo",
|
|
|
|
"msgChannel.chanNamePrefix.dataCoordSegmentInfo",
|
|
|
|
}
|
|
|
|
p.DataCoordSegmentInfo = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initDataCoordSubName() {
|
|
|
|
keys := []string{
|
|
|
|
"common.subNamePrefix.dataCoordSubNamePrefix",
|
|
|
|
"msgChannel.subNamePrefix.dataCoordSubNamePrefix",
|
|
|
|
}
|
|
|
|
p.DataCoordSubName = p.initChanNamePrefix(keys)
|
2022-02-02 00:35:43 +08:00
|
|
|
}
|
|
|
|
|
2022-03-04 11:17:56 +08:00
|
|
|
func (p *commonConfig) initDataNodeSubName() {
|
|
|
|
keys := []string{
|
|
|
|
"common.subNamePrefix.dataNodeSubNamePrefix",
|
|
|
|
"msgChannel.subNamePrefix.dataNodeSubNamePrefix",
|
|
|
|
}
|
|
|
|
p.DataNodeSubName = p.initChanNamePrefix(keys)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *commonConfig) initDefaultPartitionName() {
|
|
|
|
p.DefaultPartitionName = p.Base.LoadWithDefault("common.defaultPartitionName", "_default")
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *commonConfig) initDefaultIndexName() {
|
|
|
|
p.DefaultIndexName = p.Base.LoadWithDefault("common.defaultIndexName", "_default_idx")
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *commonConfig) initRetentionDuration() {
|
|
|
|
p.RetentionDuration = p.Base.ParseInt64WithDefault("common.retentionDuration", DefaultRetentionDuration)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *commonConfig) initSimdType() {
|
|
|
|
keys := []string{
|
|
|
|
"common.simdType",
|
|
|
|
"knowhere.simdType",
|
|
|
|
}
|
|
|
|
p.SimdType = p.Base.LoadWithDefault2(keys, "auto")
|
2022-02-09 14:41:45 +08:00
|
|
|
}
|
|
|
|
|
2022-04-08 20:29:33 +08:00
|
|
|
func (p *commonConfig) initIndexSliceSize() {
|
|
|
|
p.IndexSliceSize = p.Base.ParseInt64WithDefault("common.indexSliceSize", DefaultIndexSliceSize)
|
|
|
|
}
|
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// --- rootcoord ---
|
|
|
|
type rootCoordConfig struct {
|
2022-02-08 20:57:47 +08:00
|
|
|
Base *BaseTable
|
2022-02-02 00:35:43 +08:00
|
|
|
|
|
|
|
Address string
|
|
|
|
Port int
|
|
|
|
|
|
|
|
DmlChannelNum int64
|
|
|
|
MaxPartitionNum int64
|
|
|
|
MinSegmentSizeToEnableIndex int64
|
2022-04-03 11:37:29 +08:00
|
|
|
ImportTaskExpiration float64
|
|
|
|
ImportTaskRetention float64
|
|
|
|
ImportIndexCheckInterval float64
|
|
|
|
ImportIndexWaitLimit float64
|
2022-02-02 00:35:43 +08:00
|
|
|
|
2022-03-25 11:03:25 +08:00
|
|
|
// --- ETCD Path ---
|
|
|
|
ImportTaskSubPath string
|
|
|
|
|
2022-02-02 00:35:43 +08:00
|
|
|
CreatedTime time.Time
|
|
|
|
UpdatedTime time.Time
|
|
|
|
}
|
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *rootCoordConfig) init(base *BaseTable) {
|
|
|
|
p.Base = base
|
|
|
|
p.DmlChannelNum = p.Base.ParseInt64WithDefault("rootCoord.dmlChannelNum", 256)
|
|
|
|
p.MaxPartitionNum = p.Base.ParseInt64WithDefault("rootCoord.maxPartitionNum", 4096)
|
|
|
|
p.MinSegmentSizeToEnableIndex = p.Base.ParseInt64WithDefault("rootCoord.minSegmentSizeToEnableIndex", 1024)
|
2022-04-03 11:37:29 +08:00
|
|
|
p.ImportTaskExpiration = p.Base.ParseFloatWithDefault("rootCoord.importTaskExpiration", 3600)
|
|
|
|
p.ImportTaskRetention = p.Base.ParseFloatWithDefault("rootCoord.importTaskRetention", 3600*24)
|
|
|
|
p.ImportIndexCheckInterval = p.Base.ParseFloatWithDefault("rootCoord.importIndexCheckInterval", 60*5)
|
|
|
|
p.ImportIndexWaitLimit = p.Base.ParseFloatWithDefault("rootCoord.importIndexWaitLimit", 60*20)
|
2022-03-25 11:03:25 +08:00
|
|
|
p.ImportTaskSubPath = "importtask"
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// --- proxy ---
|
|
|
|
type proxyConfig struct {
|
2022-02-08 20:57:47 +08:00
|
|
|
Base *BaseTable
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
// NetworkPort & IP are not used
|
2022-01-10 14:51:34 +08:00
|
|
|
NetworkPort int
|
|
|
|
IP string
|
2021-12-23 18:39:11 +08:00
|
|
|
NetworkAddress string
|
|
|
|
|
|
|
|
Alias string
|
|
|
|
|
|
|
|
ProxyID UniqueID
|
|
|
|
TimeTickInterval time.Duration
|
|
|
|
MsgStreamTimeTickBufSize int64
|
|
|
|
MaxNameLength int64
|
|
|
|
MaxFieldNum int64
|
|
|
|
MaxShardNum int32
|
|
|
|
MaxDimension int64
|
|
|
|
BufFlagExpireTime time.Duration
|
|
|
|
BufFlagCleanupInterval time.Duration
|
2022-03-10 10:35:59 +08:00
|
|
|
GinLogging bool
|
2021-12-23 18:39:11 +08:00
|
|
|
|
2022-01-06 13:09:49 +08:00
|
|
|
// required from QueryCoord
|
2021-12-23 18:39:11 +08:00
|
|
|
SearchResultChannelNames []string
|
|
|
|
RetrieveResultChannelNames []string
|
|
|
|
|
|
|
|
MaxTaskNum int64
|
|
|
|
|
|
|
|
CreatedTime time.Time
|
|
|
|
UpdatedTime time.Time
|
|
|
|
}
|
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *proxyConfig) init(base *BaseTable) {
|
|
|
|
p.Base = base
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
p.initTimeTickInterval()
|
|
|
|
|
|
|
|
p.initMsgStreamTimeTickBufSize()
|
|
|
|
p.initMaxNameLength()
|
|
|
|
p.initMaxFieldNum()
|
|
|
|
p.initMaxShardNum()
|
|
|
|
p.initMaxDimension()
|
|
|
|
|
|
|
|
p.initMaxTaskNum()
|
|
|
|
p.initBufFlagExpireTime()
|
|
|
|
p.initBufFlagCleanupInterval()
|
2022-03-10 10:35:59 +08:00
|
|
|
p.initGinLogging()
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// InitAlias initialize Alias member.
|
|
|
|
func (p *proxyConfig) InitAlias(alias string) {
|
|
|
|
p.Alias = alias
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *proxyConfig) initTimeTickInterval() {
|
2022-02-08 20:57:47 +08:00
|
|
|
interval := p.Base.ParseIntWithDefault("proxy.timeTickInterval", 200)
|
2021-12-23 18:39:11 +08:00
|
|
|
p.TimeTickInterval = time.Duration(interval) * time.Millisecond
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *proxyConfig) initMsgStreamTimeTickBufSize() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.MsgStreamTimeTickBufSize = p.Base.ParseInt64WithDefault("proxy.msgStream.timeTick.bufSize", 512)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *proxyConfig) initMaxNameLength() {
|
2022-02-08 20:57:47 +08:00
|
|
|
str := p.Base.LoadWithDefault("proxy.maxNameLength", "255")
|
2021-12-23 18:39:11 +08:00
|
|
|
maxNameLength, err := strconv.ParseInt(str, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.MaxNameLength = maxNameLength
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *proxyConfig) initMaxShardNum() {
|
2022-02-08 20:57:47 +08:00
|
|
|
str := p.Base.LoadWithDefault("proxy.maxShardNum", "256")
|
2021-12-23 18:39:11 +08:00
|
|
|
maxShardNum, err := strconv.ParseInt(str, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.MaxShardNum = int32(maxShardNum)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *proxyConfig) initMaxFieldNum() {
|
2022-02-08 20:57:47 +08:00
|
|
|
str := p.Base.LoadWithDefault("proxy.maxFieldNum", "64")
|
2021-12-23 18:39:11 +08:00
|
|
|
maxFieldNum, err := strconv.ParseInt(str, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.MaxFieldNum = maxFieldNum
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *proxyConfig) initMaxDimension() {
|
2022-02-08 20:57:47 +08:00
|
|
|
str := p.Base.LoadWithDefault("proxy.maxDimension", "32768")
|
2021-12-23 18:39:11 +08:00
|
|
|
maxDimension, err := strconv.ParseInt(str, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.MaxDimension = maxDimension
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *proxyConfig) initMaxTaskNum() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.MaxTaskNum = p.Base.ParseInt64WithDefault("proxy.maxTaskNum", 1024)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *proxyConfig) initBufFlagExpireTime() {
|
2022-02-08 20:57:47 +08:00
|
|
|
expireTime := p.Base.ParseInt64WithDefault("proxy.bufFlagExpireTime", 3600)
|
2021-12-23 18:39:11 +08:00
|
|
|
p.BufFlagExpireTime = time.Duration(expireTime) * time.Second
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *proxyConfig) initBufFlagCleanupInterval() {
|
2022-02-08 20:57:47 +08:00
|
|
|
interval := p.Base.ParseInt64WithDefault("proxy.bufFlagCleanupInterval", 600)
|
2021-12-23 18:39:11 +08:00
|
|
|
p.BufFlagCleanupInterval = time.Duration(interval) * time.Second
|
|
|
|
}
|
|
|
|
|
2022-03-10 10:35:59 +08:00
|
|
|
func (p *proxyConfig) initGinLogging() {
|
|
|
|
// Gin logging is on by default.
|
|
|
|
p.GinLogging = p.Base.ParseBool("proxy.ginLogging", true)
|
|
|
|
}
|
|
|
|
|
2021-12-23 18:39:11 +08:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// --- querycoord ---
|
|
|
|
type queryCoordConfig struct {
|
2022-02-08 20:57:47 +08:00
|
|
|
Base *BaseTable
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
NodeID uint64
|
|
|
|
|
|
|
|
Address string
|
|
|
|
Port int
|
|
|
|
QueryCoordID UniqueID
|
|
|
|
|
|
|
|
CreatedTime time.Time
|
|
|
|
UpdatedTime time.Time
|
|
|
|
|
|
|
|
//---- Handoff ---
|
|
|
|
AutoHandoff bool
|
|
|
|
|
|
|
|
//---- Balance ---
|
|
|
|
AutoBalance bool
|
|
|
|
OverloadedMemoryThresholdPercentage float64
|
|
|
|
BalanceIntervalSeconds int64
|
|
|
|
MemoryUsageMaxDifferencePercentage float64
|
|
|
|
}
|
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *queryCoordConfig) init(base *BaseTable) {
|
|
|
|
p.Base = base
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
//---- Handoff ---
|
|
|
|
p.initAutoHandoff()
|
|
|
|
|
|
|
|
//---- Balance ---
|
|
|
|
p.initAutoBalance()
|
|
|
|
p.initOverloadedMemoryThresholdPercentage()
|
|
|
|
p.initBalanceIntervalSeconds()
|
|
|
|
p.initMemoryUsageMaxDifferencePercentage()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryCoordConfig) initAutoHandoff() {
|
2022-02-08 20:57:47 +08:00
|
|
|
handoff, err := p.Base.Load("queryCoord.autoHandoff")
|
2021-12-23 18:39:11 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.AutoHandoff, err = strconv.ParseBool(handoff)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryCoordConfig) initAutoBalance() {
|
2022-02-08 20:57:47 +08:00
|
|
|
balanceStr := p.Base.LoadWithDefault("queryCoord.autoBalance", "false")
|
2021-12-23 18:39:11 +08:00
|
|
|
autoBalance, err := strconv.ParseBool(balanceStr)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.AutoBalance = autoBalance
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryCoordConfig) initOverloadedMemoryThresholdPercentage() {
|
2022-02-08 20:57:47 +08:00
|
|
|
overloadedMemoryThresholdPercentage := p.Base.LoadWithDefault("queryCoord.overloadedMemoryThresholdPercentage", "90")
|
2021-12-23 18:39:11 +08:00
|
|
|
thresholdPercentage, err := strconv.ParseInt(overloadedMemoryThresholdPercentage, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.OverloadedMemoryThresholdPercentage = float64(thresholdPercentage) / 100
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryCoordConfig) initBalanceIntervalSeconds() {
|
2022-02-08 20:57:47 +08:00
|
|
|
balanceInterval := p.Base.LoadWithDefault("queryCoord.balanceIntervalSeconds", "60")
|
2021-12-23 18:39:11 +08:00
|
|
|
interval, err := strconv.ParseInt(balanceInterval, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.BalanceIntervalSeconds = interval
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryCoordConfig) initMemoryUsageMaxDifferencePercentage() {
|
2022-02-08 20:57:47 +08:00
|
|
|
maxDiff := p.Base.LoadWithDefault("queryCoord.memoryUsageMaxDifferencePercentage", "30")
|
2021-12-23 18:39:11 +08:00
|
|
|
diffPercentage, err := strconv.ParseInt(maxDiff, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.MemoryUsageMaxDifferencePercentage = float64(diffPercentage) / 100
|
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// --- querynode ---
|
|
|
|
type queryNodeConfig struct {
|
2022-02-08 20:57:47 +08:00
|
|
|
Base *BaseTable
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
Alias string
|
|
|
|
QueryNodeIP string
|
|
|
|
QueryNodePort int64
|
|
|
|
QueryNodeID UniqueID
|
|
|
|
// TODO: remove cacheSize
|
|
|
|
CacheSize int64 // deprecated
|
|
|
|
|
|
|
|
FlowGraphMaxQueueLength int32
|
|
|
|
FlowGraphMaxParallelism int32
|
|
|
|
|
|
|
|
// search
|
|
|
|
SearchChannelNames []string
|
|
|
|
SearchResultChannelNames []string
|
|
|
|
SearchReceiveBufSize int64
|
|
|
|
SearchPulsarBufSize int64
|
|
|
|
SearchResultReceiveBufSize int64
|
|
|
|
|
|
|
|
// Retrieve
|
|
|
|
RetrieveChannelNames []string
|
|
|
|
RetrieveResultChannelNames []string
|
|
|
|
RetrieveReceiveBufSize int64
|
|
|
|
RetrievePulsarBufSize int64
|
|
|
|
RetrieveResultReceiveBufSize int64
|
|
|
|
|
|
|
|
// stats
|
|
|
|
StatsPublishInterval int
|
|
|
|
|
|
|
|
GracefulTime int64
|
|
|
|
SliceIndex int
|
|
|
|
|
|
|
|
// segcore
|
|
|
|
ChunkRows int64
|
|
|
|
|
|
|
|
CreatedTime time.Time
|
|
|
|
UpdatedTime time.Time
|
|
|
|
|
|
|
|
// memory limit
|
|
|
|
OverloadedMemoryThresholdPercentage float64
|
2022-03-26 22:05:26 +08:00
|
|
|
|
|
|
|
// cache limit
|
2022-04-07 22:05:32 +08:00
|
|
|
CacheEnabled bool
|
|
|
|
CacheMemoryLimit int64
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *queryNodeConfig) init(base *BaseTable) {
|
|
|
|
p.Base = base
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
p.initCacheSize()
|
|
|
|
p.initGracefulTime()
|
|
|
|
|
|
|
|
p.initFlowGraphMaxQueueLength()
|
|
|
|
p.initFlowGraphMaxParallelism()
|
|
|
|
|
|
|
|
p.initSearchReceiveBufSize()
|
|
|
|
p.initSearchPulsarBufSize()
|
|
|
|
p.initSearchResultReceiveBufSize()
|
|
|
|
|
|
|
|
p.initStatsPublishInterval()
|
|
|
|
|
|
|
|
p.initSegcoreChunkRows()
|
|
|
|
|
|
|
|
p.initOverloadedMemoryThresholdPercentage()
|
2022-03-26 22:05:26 +08:00
|
|
|
|
2022-04-07 22:05:32 +08:00
|
|
|
p.initCacheMemoryLimit()
|
|
|
|
p.initCacheEnabled()
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// InitAlias initializes an alias for the QueryNode role.
|
|
|
|
func (p *queryNodeConfig) InitAlias(alias string) {
|
|
|
|
p.Alias = alias
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryNodeConfig) initCacheSize() {
|
|
|
|
defer log.Debug("init cacheSize", zap.Any("cacheSize (GB)", p.CacheSize))
|
|
|
|
|
|
|
|
const defaultCacheSize = 32 // GB
|
|
|
|
p.CacheSize = defaultCacheSize
|
|
|
|
|
|
|
|
var err error
|
|
|
|
cacheSize := os.Getenv("CACHE_SIZE")
|
|
|
|
if cacheSize == "" {
|
2022-02-08 20:57:47 +08:00
|
|
|
cacheSize, err = p.Base.Load("queryNode.cacheSize")
|
2021-12-23 18:39:11 +08:00
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
value, err := strconv.ParseInt(cacheSize, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
p.CacheSize = value
|
|
|
|
}
|
|
|
|
|
|
|
|
// advanced params
|
|
|
|
// stats
|
|
|
|
func (p *queryNodeConfig) initStatsPublishInterval() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.StatsPublishInterval = p.Base.ParseIntWithDefault("queryNode.stats.publishInterval", 1000)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// dataSync:
|
|
|
|
func (p *queryNodeConfig) initFlowGraphMaxQueueLength() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.FlowGraphMaxQueueLength = p.Base.ParseInt32WithDefault("queryNode.dataSync.flowGraph.maxQueueLength", 1024)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryNodeConfig) initFlowGraphMaxParallelism() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.FlowGraphMaxParallelism = p.Base.ParseInt32WithDefault("queryNode.dataSync.flowGraph.maxParallelism", 1024)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// msgStream
|
|
|
|
func (p *queryNodeConfig) initSearchReceiveBufSize() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.SearchReceiveBufSize = p.Base.ParseInt64WithDefault("queryNode.msgStream.search.recvBufSize", 512)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryNodeConfig) initSearchPulsarBufSize() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.SearchPulsarBufSize = p.Base.ParseInt64WithDefault("queryNode.msgStream.search.pulsarBufSize", 512)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryNodeConfig) initSearchResultReceiveBufSize() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.SearchResultReceiveBufSize = p.Base.ParseInt64WithDefault("queryNode.msgStream.searchResult.recvBufSize", 64)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryNodeConfig) initGracefulTime() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.GracefulTime = p.Base.ParseInt64("queryNode.gracefulTime")
|
2021-12-23 18:39:11 +08:00
|
|
|
log.Debug("query node init gracefulTime", zap.Any("gracefulTime", p.GracefulTime))
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryNodeConfig) initSegcoreChunkRows() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.ChunkRows = p.Base.ParseInt64WithDefault("queryNode.segcore.chunkRows", 32768)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *queryNodeConfig) initOverloadedMemoryThresholdPercentage() {
|
2022-02-08 20:57:47 +08:00
|
|
|
overloadedMemoryThresholdPercentage := p.Base.LoadWithDefault("queryCoord.overloadedMemoryThresholdPercentage", "90")
|
2021-12-23 18:39:11 +08:00
|
|
|
thresholdPercentage, err := strconv.ParseInt(overloadedMemoryThresholdPercentage, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.OverloadedMemoryThresholdPercentage = float64(thresholdPercentage) / 100
|
|
|
|
}
|
|
|
|
|
2022-04-07 22:05:32 +08:00
|
|
|
func (p *queryNodeConfig) initCacheMemoryLimit() {
|
|
|
|
overloadedMemoryThresholdPercentage := p.Base.LoadWithDefault("queryNode.cache.memoryLimit", "2147483648")
|
|
|
|
cacheMemoryLimit, err := strconv.ParseInt(overloadedMemoryThresholdPercentage, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.CacheMemoryLimit = cacheMemoryLimit
|
|
|
|
}
|
|
|
|
func (p *queryNodeConfig) initCacheEnabled() {
|
|
|
|
var err error
|
|
|
|
cacheEnabled := p.Base.LoadWithDefault("queryNode.cache.enabled", "true")
|
|
|
|
p.CacheEnabled, err = strconv.ParseBool(cacheEnabled)
|
2022-03-26 22:05:26 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-23 18:39:11 +08:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// --- datacoord ---
|
|
|
|
type dataCoordConfig struct {
|
2022-02-08 20:57:47 +08:00
|
|
|
Base *BaseTable
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
NodeID int64
|
|
|
|
|
|
|
|
IP string
|
|
|
|
Port int
|
|
|
|
Address string
|
|
|
|
|
|
|
|
// --- ETCD ---
|
2022-01-06 14:49:20 +08:00
|
|
|
ChannelWatchSubPath string
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
// --- SEGMENTS ---
|
|
|
|
SegmentMaxSize float64
|
|
|
|
SegmentSealProportion float64
|
|
|
|
SegAssignmentExpiration int64
|
|
|
|
|
|
|
|
CreatedTime time.Time
|
|
|
|
UpdatedTime time.Time
|
|
|
|
|
|
|
|
EnableCompaction bool
|
2022-01-10 19:03:35 +08:00
|
|
|
EnableAutoCompaction bool
|
2021-12-23 18:39:11 +08:00
|
|
|
EnableGarbageCollection bool
|
|
|
|
|
2022-02-07 22:45:46 +08:00
|
|
|
RetentionDuration int64
|
|
|
|
CompactionEntityExpiration int64
|
|
|
|
|
2021-12-23 18:39:11 +08:00
|
|
|
// Garbage Collection
|
|
|
|
GCInterval time.Duration
|
|
|
|
GCMissingTolerance time.Duration
|
|
|
|
GCDropTolerance time.Duration
|
|
|
|
}
|
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *dataCoordConfig) init(base *BaseTable) {
|
|
|
|
p.Base = base
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
p.initChannelWatchPrefix()
|
|
|
|
|
|
|
|
p.initSegmentMaxSize()
|
|
|
|
p.initSegmentSealProportion()
|
|
|
|
p.initSegAssignmentExpiration()
|
|
|
|
|
|
|
|
p.initEnableCompaction()
|
|
|
|
p.initEnableAutoCompaction()
|
2022-02-07 22:45:46 +08:00
|
|
|
p.initCompactionEntityExpiration()
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
p.initEnableGarbageCollection()
|
|
|
|
p.initGCInterval()
|
|
|
|
p.initGCMissingTolerance()
|
|
|
|
p.initGCDropTolerance()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataCoordConfig) initSegmentMaxSize() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.SegmentMaxSize = p.Base.ParseFloatWithDefault("dataCoord.segment.maxSize", 512.0)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataCoordConfig) initSegmentSealProportion() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.SegmentSealProportion = p.Base.ParseFloatWithDefault("dataCoord.segment.sealProportion", 0.75)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataCoordConfig) initSegAssignmentExpiration() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.SegAssignmentExpiration = p.Base.ParseInt64WithDefault("dataCoord.segment.assignmentExpiration", 2000)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataCoordConfig) initChannelWatchPrefix() {
|
|
|
|
// WARN: this value should not be put to milvus.yaml. It's a default value for channel watch path.
|
|
|
|
// This will be removed after we reconstruct our config module.
|
|
|
|
p.ChannelWatchSubPath = "channelwatch"
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataCoordConfig) initEnableCompaction() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.EnableCompaction = p.Base.ParseBool("dataCoord.enableCompaction", false)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// -- GC --
|
|
|
|
func (p *dataCoordConfig) initEnableGarbageCollection() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.EnableGarbageCollection = p.Base.ParseBool("dataCoord.enableGarbageCollection", false)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataCoordConfig) initGCInterval() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.GCInterval = time.Duration(p.Base.ParseInt64WithDefault("dataCoord.gc.interval", 60*60)) * time.Second
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataCoordConfig) initGCMissingTolerance() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.GCMissingTolerance = time.Duration(p.Base.ParseInt64WithDefault("dataCoord.gc.missingTolerance", 24*60*60)) * time.Second
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataCoordConfig) initGCDropTolerance() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.GCDropTolerance = time.Duration(p.Base.ParseInt64WithDefault("dataCoord.gc.dropTolerance", 24*60*60)) * time.Second
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataCoordConfig) initEnableAutoCompaction() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.EnableAutoCompaction = p.Base.ParseBool("dataCoord.compaction.enableAutoCompaction", false)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
2022-02-07 22:45:46 +08:00
|
|
|
func (p *dataCoordConfig) initCompactionEntityExpiration() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.CompactionEntityExpiration = p.Base.ParseInt64WithDefault("dataCoord.compaction.entityExpiration", math.MaxInt64)
|
2022-02-07 22:45:46 +08:00
|
|
|
p.CompactionEntityExpiration = func(x, y int64) int64 {
|
|
|
|
if x > y {
|
|
|
|
return x
|
|
|
|
}
|
|
|
|
return y
|
|
|
|
}(p.CompactionEntityExpiration, p.RetentionDuration)
|
|
|
|
}
|
|
|
|
|
2021-12-23 18:39:11 +08:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// --- datanode ---
|
|
|
|
type dataNodeConfig struct {
|
2022-02-08 20:57:47 +08:00
|
|
|
Base *BaseTable
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
// ID of the current DataNode
|
|
|
|
NodeID UniqueID
|
|
|
|
|
|
|
|
// IP of the current DataNode
|
|
|
|
IP string
|
|
|
|
|
|
|
|
// Port of the current DataNode
|
|
|
|
Port int
|
|
|
|
FlowGraphMaxQueueLength int32
|
|
|
|
FlowGraphMaxParallelism int32
|
|
|
|
FlushInsertBufferSize int64
|
|
|
|
InsertBinlogRootPath string
|
|
|
|
StatsBinlogRootPath string
|
|
|
|
DeleteBinlogRootPath string
|
|
|
|
Alias string // Different datanode in one machine
|
|
|
|
|
|
|
|
// etcd
|
|
|
|
ChannelWatchSubPath string
|
|
|
|
|
|
|
|
CreatedTime time.Time
|
|
|
|
UpdatedTime time.Time
|
|
|
|
}
|
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *dataNodeConfig) init(base *BaseTable) {
|
|
|
|
p.Base = base
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
p.initFlowGraphMaxQueueLength()
|
|
|
|
p.initFlowGraphMaxParallelism()
|
|
|
|
p.initFlushInsertBufferSize()
|
|
|
|
p.initInsertBinlogRootPath()
|
|
|
|
p.initStatsBinlogRootPath()
|
|
|
|
p.initDeleteBinlogRootPath()
|
|
|
|
|
|
|
|
p.initChannelWatchPath()
|
|
|
|
}
|
|
|
|
|
|
|
|
// InitAlias init this DataNode alias
|
|
|
|
func (p *dataNodeConfig) InitAlias(alias string) {
|
|
|
|
p.Alias = alias
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataNodeConfig) initFlowGraphMaxQueueLength() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.FlowGraphMaxQueueLength = p.Base.ParseInt32WithDefault("dataNode.dataSync.flowGraph.maxQueueLength", 1024)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataNodeConfig) initFlowGraphMaxParallelism() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.FlowGraphMaxParallelism = p.Base.ParseInt32WithDefault("dataNode.dataSync.flowGraph.maxParallelism", 1024)
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataNodeConfig) initFlushInsertBufferSize() {
|
2022-02-08 20:57:47 +08:00
|
|
|
p.FlushInsertBufferSize = p.Base.ParseInt64("_DATANODE_INSERTBUFSIZE")
|
2021-12-23 18:39:11 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataNodeConfig) initInsertBinlogRootPath() {
|
2022-01-10 17:29:34 +08:00
|
|
|
// GOOSE TODO: rootPath change to TenentID
|
2022-02-08 20:57:47 +08:00
|
|
|
rootPath, err := p.Base.Load("minio.rootPath")
|
2021-12-23 18:39:11 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.InsertBinlogRootPath = path.Join(rootPath, "insert_log")
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataNodeConfig) initStatsBinlogRootPath() {
|
2022-02-08 20:57:47 +08:00
|
|
|
rootPath, err := p.Base.Load("minio.rootPath")
|
2021-12-23 18:39:11 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.StatsBinlogRootPath = path.Join(rootPath, "stats_log")
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataNodeConfig) initDeleteBinlogRootPath() {
|
2022-02-08 20:57:47 +08:00
|
|
|
rootPath, err := p.Base.Load("minio.rootPath")
|
2021-12-23 18:39:11 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.DeleteBinlogRootPath = path.Join(rootPath, "delta_log")
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *dataNodeConfig) initChannelWatchPath() {
|
|
|
|
p.ChannelWatchSubPath = "channelwatch"
|
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// --- indexcoord ---
|
|
|
|
type indexCoordConfig struct {
|
2022-02-08 20:57:47 +08:00
|
|
|
Base *BaseTable
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
Address string
|
|
|
|
Port int
|
|
|
|
|
|
|
|
IndexStorageRootPath string
|
|
|
|
|
|
|
|
CreatedTime time.Time
|
|
|
|
UpdatedTime time.Time
|
|
|
|
}
|
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *indexCoordConfig) init(base *BaseTable) {
|
|
|
|
p.Base = base
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
p.initIndexStorageRootPath()
|
|
|
|
}
|
|
|
|
|
|
|
|
// initIndexStorageRootPath initializes the root path of index files.
|
|
|
|
func (p *indexCoordConfig) initIndexStorageRootPath() {
|
2022-02-08 20:57:47 +08:00
|
|
|
rootPath, err := p.Base.Load("minio.rootPath")
|
2021-12-23 18:39:11 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.IndexStorageRootPath = path.Join(rootPath, "index_files")
|
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// --- indexnode ---
|
|
|
|
type indexNodeConfig struct {
|
2022-02-08 20:57:47 +08:00
|
|
|
Base *BaseTable
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
IP string
|
|
|
|
Address string
|
|
|
|
Port int
|
|
|
|
|
|
|
|
NodeID int64
|
|
|
|
Alias string
|
|
|
|
|
|
|
|
IndexStorageRootPath string
|
|
|
|
|
|
|
|
CreatedTime time.Time
|
|
|
|
UpdatedTime time.Time
|
|
|
|
}
|
|
|
|
|
2022-02-08 20:57:47 +08:00
|
|
|
func (p *indexNodeConfig) init(base *BaseTable) {
|
|
|
|
p.Base = base
|
2021-12-23 18:39:11 +08:00
|
|
|
|
|
|
|
p.initIndexStorageRootPath()
|
|
|
|
}
|
|
|
|
|
|
|
|
// InitAlias initializes an alias for the IndexNode role.
|
|
|
|
func (p *indexNodeConfig) InitAlias(alias string) {
|
|
|
|
p.Alias = alias
|
|
|
|
}
|
|
|
|
|
|
|
|
func (p *indexNodeConfig) initIndexStorageRootPath() {
|
2022-02-08 20:57:47 +08:00
|
|
|
rootPath, err := p.Base.Load("minio.rootPath")
|
2021-12-23 18:39:11 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
p.IndexStorageRootPath = path.Join(rootPath, "index_files")
|
|
|
|
}
|