From ffc50af67e12dbb19167daf2b10b2b4dcc4ce70d Mon Sep 17 00:00:00 2001
From: TwinProduction <twin@twinnation.org>
Date: Sat, 19 Oct 2019 21:39:31 -0400
Subject: [PATCH] Minor improvements

---
 config/config.go      | 23 +++++++++++++----------
 config/config_test.go |  4 ++--
 2 files changed, 15 insertions(+), 12 deletions(-)

diff --git a/config/config.go b/config/config.go
index 96bd5f51..12a96050 100644
--- a/config/config.go
+++ b/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 {
-	config = &Config{}
-	if bytes, err := ioutil.ReadFile(fileName); err == nil { // file exists
-		return ParseConfigBytes(bytes)
+func readConfigurationFile(fileName string) (*Config, error) {
+	if bytes, err := ioutil.ReadFile(fileName); err == nil {
+		// file exists, so we'll parse it and return it
+		return parseConfigBytes(bytes), nil
 	} else {
-		panic(err)
+		return nil, err
 	}
-	return config
+	return &Config{}, nil
 }
 
-func ParseConfigBytes(yamlBytes []byte) *Config {
-	config = &Config{}
-	yaml.Unmarshal(yamlBytes, config)
+func parseConfigBytes(yamlBytes []byte) (config *Config) {
+	yaml.Unmarshal(yamlBytes, &config)
 	for _, service := range config.Services {
 		if service.Interval == 0 {
 			service.Interval = 10 * time.Second
diff --git a/config/config_test.go b/config/config_test.go
index 2b534521..2e222f86 100644
--- a/config/config_test.go
+++ b/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