// 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 ¶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() }