45 lines
875 B
Go
45 lines
875 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"gopkg.in/yaml.v2"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ClusterConfig struct {
|
||
|
|
ClusterLabel string `yaml:"cluster_label"`
|
||
|
|
User string `yaml:"user"`
|
||
|
|
ConfigFile string `yaml:"config_file"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Config is the top-level configuration for Metastord.
|
||
|
|
type Config struct {
|
||
|
|
Cluster []*ClusterConfig
|
||
|
|
}
|
||
|
|
|
||
|
|
// fileExists returns true if the path exists and is a file.
|
||
|
|
func fileExists(path string) bool {
|
||
|
|
stat, err := os.Stat(path)
|
||
|
|
return !os.IsNotExist(err) && !stat.IsDir()
|
||
|
|
}
|
||
|
|
|
||
|
|
func ParseConfig(p string) (*Config, error) {
|
||
|
|
if !fileExists(p) {
|
||
|
|
return nil, fmt.Errorf("Config file does not exist.")
|
||
|
|
}
|
||
|
|
|
||
|
|
cfgData, err := ioutil.ReadFile(p)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("ReadFile: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var cfg Config
|
||
|
|
err = yaml.Unmarshal(cfgData, &cfg)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("yaml parse: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &cfg, nil
|
||
|
|
}
|