about summary refs log tree commit diff
path: root/util/util.go
diff options
context:
space:
mode:
authorVincent Ambo <tazjin@gmail.com>2017-06-22T13·52+0200
committerVincent Ambo <tazjin@gmail.com>2017-07-03T12·27+0200
commit3728d0ae2eb957c8ba4251ee0f562c4941163816 (patch)
tree11e1960127024569d3ee77b8ed24039acda6ff13 /util/util.go
parente2f7cf6258f469ae9b69d0b5cce57b397c08c36d (diff)
refactor context: Extract loadJsonOrYaml to util package
The logic to deserialise a structure from *either* JSON or YAML is reused several
times and can be easily extracted, which this commit does.
Diffstat (limited to 'util/util.go')
-rw-r--r--util/util.go27
1 files changed, 27 insertions, 0 deletions
diff --git a/util/util.go b/util/util.go
index ea9f636ac89a..0c5815a38b88 100644
--- a/util/util.go
+++ b/util/util.go
@@ -1,5 +1,14 @@
 package util
 
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"strings"
+
+	"github.com/ghodss/yaml"
+)
+
 // Merges two maps together. Values from the second map override values in the first map.
 // The returned map is new if anything was changed.
 func Merge(in1 *map[string]interface{}, in2 *map[string]interface{}) *map[string]interface{} {
@@ -22,3 +31,21 @@ func Merge(in1 *map[string]interface{}, in2 *map[string]interface{}) *map[string
 
 	return &new
 }
+
+// Loads either a YAML or JSON file from the specified path and deserialises it into the provided interface.
+func LoadJsonOrYaml(filename string, addr interface{}) error {
+	file, err := ioutil.ReadFile(filename)
+	if err != nil {
+		return err
+	}
+
+	if strings.HasSuffix(filename, "json") {
+		err = json.Unmarshal(file, addr)
+	} else if strings.HasSuffix(filename, "yaml") || strings.HasSuffix(filename, "yml") {
+		err = yaml.Unmarshal(file, addr)
+	} else {
+		return fmt.Errorf("File format not supported. Must be JSON or YAML.")
+	}
+
+	return nil
+}