func LoadConfiguration() {
if CfgPath != "" {
if err := config.InitConfig(CfgBackend, CfgPath); err != nil {
fmt.Fprintf(os.Stderr, "%v\n", err)
os.Exit(1)
}
}
}
func InitConfig(t *testing.T, conf string, params ...HelperParams) {
f, err := ioutil.TempFile("", "skydive_agent")
if err != nil {
t.Fatal(err.Error())
}
if len(params) == 0 {
params = []HelperParams{make(HelperParams)}
}
params[0]["AnalyzerPort"] = 64500
if testing.Verbose() {
params[0]["LogLevel"] = "DEBUG"
} else {
params[0]["LogLevel"] = "INFO"
}
if etcdServer != "" {
params[0]["EmbeddedEtcd"] = "false"
params[0]["EtcdServer"] = etcdServer
} else {
params[0]["EmbeddedEtcd"] = "true"
params[0]["EtcdServer"] = "http://localhost:2374"
}
if storageBackend != "" {
params[0]["Storage"] = storageBackend
}
if storageBackend == "orientdb" {
orientDBPassword := os.Getenv("ORIENTDB_ROOT_PASSWORD")
if orientDBPassword == "" {
orientDBPassword = "******"
}
params[0]["OrientDBRootPassword"] = orientDBPassword
}
tmpl, err := template.New("config").Parse(conf)
if err != nil {
t.Fatal(err.Error())
}
buff := bytes.NewBufferString("")
tmpl.Execute(buff, params[0])
f.Write(buff.Bytes())
f.Close()
t.Logf("Configuration: %s", buff.String())
err = config.InitConfig("file", f.Name())
if err != nil {
t.Fatal(err.Error())
}
err = logging.InitLogger()
if err != nil {
t.Fatal(err)
}
}
