mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 21:39:19 +08:00
90bed1caf9
issue: #30436 origin pr: #30438 related pr: #31772 --------- Signed-off-by: SimFG <bang.fu@zilliz.com>
90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
// 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 (
|
|
"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 ¶ms
|
|
}
|
|
|
|
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()
|
|
}
|