mirror of https://github.com/milvus-io/milvus.git
121 lines
2.9 KiB
Go
121 lines
2.9 KiB
Go
package configs
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/milvus-io/milvus/cmd/tools/migration/console"
|
|
"github.com/milvus-io/milvus/pkg/util"
|
|
"github.com/milvus-io/milvus/pkg/util/paramtable"
|
|
)
|
|
|
|
const (
|
|
RunCmd = "run"
|
|
BackupCmd = "backup"
|
|
RollbackCmd = "rollback"
|
|
)
|
|
|
|
type RunConfig struct {
|
|
base *paramtable.BaseTable
|
|
Cmd string
|
|
RunWithBackup bool
|
|
SourceVersion string
|
|
TargetVersion string
|
|
BackupFilePath string
|
|
}
|
|
|
|
func newRunConfig(base *paramtable.BaseTable) *RunConfig {
|
|
c := &RunConfig{}
|
|
c.init(base)
|
|
return c
|
|
}
|
|
|
|
func (c *RunConfig) String() string {
|
|
if c == nil {
|
|
return ""
|
|
}
|
|
switch c.Cmd {
|
|
case RunCmd:
|
|
return fmt.Sprintf("Cmd: %s, SourceVersion: %s, TargetVersion: %s, BackupFilePath: %s, RunWithBackup: %v",
|
|
c.Cmd, c.SourceVersion, c.TargetVersion, c.BackupFilePath, c.RunWithBackup)
|
|
case BackupCmd:
|
|
return fmt.Sprintf("Cmd: %s, SourceVersion: %s, BackupFilePath: %s",
|
|
c.Cmd, c.SourceVersion, c.BackupFilePath)
|
|
case RollbackCmd:
|
|
return fmt.Sprintf("Cmd: %s, SourceVersion: %s, TargetVersion: %s, BackupFilePath: %s",
|
|
c.Cmd, c.SourceVersion, c.TargetVersion, c.BackupFilePath)
|
|
default:
|
|
return fmt.Sprintf("invalid cmd: %s", c.Cmd)
|
|
}
|
|
}
|
|
|
|
func (c *RunConfig) show() {
|
|
console.Warning(c.String())
|
|
}
|
|
|
|
func (c *RunConfig) init(base *paramtable.BaseTable) {
|
|
c.base = base
|
|
|
|
c.Cmd = c.base.GetWithDefault("cmd.type", "")
|
|
c.RunWithBackup, _ = strconv.ParseBool(c.base.GetWithDefault("cmd.runWithBackup", "false"))
|
|
c.SourceVersion = c.base.GetWithDefault("config.sourceVersion", "")
|
|
c.TargetVersion = c.base.GetWithDefault("config.targetVersion", "")
|
|
c.BackupFilePath = c.base.GetWithDefault("config.backupFilePath", "")
|
|
}
|
|
|
|
type MilvusConfig struct {
|
|
MetaStoreCfg *paramtable.MetaStoreConfig
|
|
EtcdCfg *paramtable.EtcdConfig
|
|
}
|
|
|
|
func newMilvusConfig(base *paramtable.BaseTable) *MilvusConfig {
|
|
c := &MilvusConfig{}
|
|
c.init(base)
|
|
return c
|
|
}
|
|
|
|
func (c *MilvusConfig) init(base *paramtable.BaseTable) {
|
|
c.MetaStoreCfg = ¶mtable.MetaStoreConfig{}
|
|
c.EtcdCfg = ¶mtable.EtcdConfig{}
|
|
|
|
c.MetaStoreCfg.Init(base)
|
|
c.EtcdCfg.Init(base)
|
|
}
|
|
|
|
func (c *MilvusConfig) String() string {
|
|
if c == nil {
|
|
return ""
|
|
}
|
|
switch c.MetaStoreCfg.MetaStoreType.GetValue() {
|
|
case util.MetaStoreTypeEtcd:
|
|
return fmt.Sprintf("Type: %s, EndPoints: %v, MetaRootPath: %s", c.MetaStoreCfg.MetaStoreType.GetValue(), c.EtcdCfg.Endpoints.GetValue(), c.EtcdCfg.MetaRootPath.GetValue())
|
|
default:
|
|
return fmt.Sprintf("unsupported meta store: %s", c.MetaStoreCfg.MetaStoreType.GetValue())
|
|
}
|
|
}
|
|
|
|
func (c *MilvusConfig) show() {
|
|
console.Warning(c.String())
|
|
}
|
|
|
|
type Config struct {
|
|
base *paramtable.BaseTable
|
|
*RunConfig
|
|
*MilvusConfig
|
|
}
|
|
|
|
func (c *Config) init(yamlFile string) {
|
|
c.base = paramtable.NewBaseTableFromYamlOnly(yamlFile)
|
|
c.RunConfig = newRunConfig(c.base)
|
|
c.MilvusConfig = newMilvusConfig(c.base)
|
|
|
|
c.RunConfig.show()
|
|
c.MilvusConfig.show()
|
|
}
|
|
|
|
func NewConfig(yamlFile string) *Config {
|
|
c := &Config{}
|
|
c.init(yamlFile)
|
|
return c
|
|
}
|