milvus/pkg/util/paramtable/runtime.go
congqixia 8b5754f7fe
enhance: Add segment seal proportion jitter (#34636)
See also #34574

Add jitter for segment seal proportion to avoid seal operation burst in
short period of time.

This PR also fix license header in paramtable pkg.

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
2024-07-15 14:47:39 +08:00

95 lines
2.1 KiB
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 paramtable
import (
"strconv"
"sync"
"time"
)
var (
once sync.Once
params ComponentParam
hookParams hookConfig
)
func Init() {
once.Do(func() {
baseTable := NewBaseTable()
params.Init(baseTable)
hookBaseTable := NewBaseTableFromYamlOnly(hookYamlFile)
hookParams.init(hookBaseTable)
})
}
func InitWithBaseTable(baseTable *BaseTable) {
once.Do(func() {
params.Init(baseTable)
hookBaseTable := NewBaseTableFromYamlOnly(hookYamlFile)
hookParams.init(hookBaseTable)
})
}
func Get() *ComponentParam {
return &params
}
func GetBaseTable() *BaseTable {
return params.baseTable
}
func GetHookParams() *hookConfig {
return &hookParams
}
func SetNodeID(newID UniqueID) {
params.RuntimeConfig.NodeID.SetValue(newID)
}
func GetNodeID() UniqueID {
return params.RuntimeConfig.NodeID.GetAsInt64()
}
func GetStringNodeID() string {
return strconv.FormatInt(GetNodeID(), 10)
}
func SetRole(role string) {
params.RuntimeConfig.Role.SetValue(role)
}
func GetRole() string {
return params.RuntimeConfig.Role.GetAsString()
}
func SetCreateTime(d time.Time) {
params.RuntimeConfig.CreateTime.SetValue(d)
}
func GetCreateTime() time.Time {
return params.RuntimeConfig.CreateTime.GetAsTime()
}
func SetUpdateTime(d time.Time) {
params.RuntimeConfig.UpdateTime.SetValue(d)
}
func GetUpdateTime() time.Time {
return params.RuntimeConfig.UpdateTime.GetAsTime()
}