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) } }