mirror of https://github.com/milvus-io/milvus.git
108 lines
2.4 KiB
Go
108 lines
2.4 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"
|
|
|
|
"github.com/milvus-io/milvus/pkg/util/typeutil"
|
|
)
|
|
|
|
var (
|
|
once sync.Once
|
|
params ComponentParam
|
|
runtimeParam = runtimeConfig{
|
|
components: typeutil.ConcurrentSet[string]{},
|
|
}
|
|
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) {
|
|
runtimeParam.nodeID.Store(newID)
|
|
}
|
|
|
|
func GetNodeID() UniqueID {
|
|
return runtimeParam.nodeID.Load()
|
|
}
|
|
|
|
func GetStringNodeID() string {
|
|
return strconv.FormatInt(GetNodeID(), 10)
|
|
}
|
|
|
|
func SetRole(role string) {
|
|
runtimeParam.role.Store(role)
|
|
}
|
|
|
|
func GetRole() string {
|
|
return runtimeParam.role.Load()
|
|
}
|
|
|
|
func SetCreateTime(d time.Time) {
|
|
runtimeParam.createTime.Store(d)
|
|
}
|
|
|
|
func GetCreateTime() time.Time {
|
|
return runtimeParam.createTime.Load()
|
|
}
|
|
|
|
func SetUpdateTime(d time.Time) {
|
|
runtimeParam.updateTime.Store(d)
|
|
}
|
|
|
|
func GetUpdateTime() time.Time {
|
|
return runtimeParam.updateTime.Load()
|
|
}
|
|
|
|
func SetLocalComponentEnabled(component string) {
|
|
runtimeParam.components.Insert(component)
|
|
}
|
|
|
|
func IsLocalComponentEnabled(component string) bool {
|
|
return runtimeParam.components.Contain(component)
|
|
}
|