Minor improvements

config/config.go

@@ -15,24 +15,27 @@ var config *Config
 
 func Get() *Config {
 	if config == nil {
-		ReadConfigurationFile("config.yaml")
+		cfg, err := readConfigurationFile("config.yaml")
+		if err != nil {
+			panic(err)
+		}
+		config = cfg
 	}
 	return config
 }
 
-func ReadConfigurationFile(fileName string) *Config {
+func readConfigurationFile(fileName string) (*Config, error) {
-	config = &Config{}
+	if bytes, err := ioutil.ReadFile(fileName); err == nil {
-	if bytes, err := ioutil.ReadFile(fileName); err == nil { // file exists
+		// file exists, so we'll parse it and return it
-		return ParseConfigBytes(bytes)
+		return parseConfigBytes(bytes), nil
 	} else {
-		panic(err)
+		return nil, err
 	}
-	return config
+	return &Config{}, nil
 }
 
-func ParseConfigBytes(yamlBytes []byte) *Config {
+func parseConfigBytes(yamlBytes []byte) (config *Config) {
-	config = &Config{}
+	yaml.Unmarshal(yamlBytes, &config)
-	yaml.Unmarshal(yamlBytes, config)
 	for _, service := range config.Services {
 		if service.Interval == 0 {
 			service.Interval = 10 * time.Second

config/config_test.go

@@ -7,7 +7,7 @@ import (
 )
 
 func TestParseConfigBytes(t *testing.T) {
-	config := ParseConfigBytes([]byte(`
+	config := parseConfigBytes([]byte(`
 services:
   - name: twinnation
     url: https://twinnation.org/actuator/health
@@ -45,7 +45,7 @@ services:
 }
 
 func TestParseConfigBytesDefault(t *testing.T) {
-	config := ParseConfigBytes([]byte(`
+	config := parseConfigBytes([]byte(`
 services:
   - name: twinnation
     url: https://twinnation.org/actuator/health