Add GetIntSlice functions

viper.go

@@ -684,6 +684,12 @@ func (v *Viper) GetStringSlice(key string) []string {
 	return cast.ToStringSlice(v.Get(key))
 }
 
+// GetIntSlice returns the value associated with the key as a slice of ints.
+func GetIntSlice(key string) []int { return v.GetIntSlice(key) }
+func (v *Viper) GetIntSlice(key string) []int {
+	return cast.ToIntSlice(v.Get(key))
+}
+
 // GetStringMap returns the value associated with the key as a map of interfaces.
 func GetStringMap(key string) map[string]interface{} { return v.GetStringMap(key) }
 func (v *Viper) GetStringMap(key string) map[string]interface{} {

viper_test.go

@@ -818,6 +818,10 @@ hello:
     - uk
     - fr
     - de
+    groups:
+    - 1
+    - 2
+    - 3
 `)
 
 var yamlMergeExampleSrc = []byte(`
@@ -827,6 +831,9 @@ hello:
     universe:
     - mw
     - ad
+    ages:
+    - 21
+    - 34
 fu: bar
 `)
 
@@ -853,6 +860,10 @@ func TestMergeConfig(t *testing.T) {
 		t.Fatalf("len(world) != 4, = %d", len(world))
 	}
 
+	if groups := v.GetIntSlice("hello.groups"); len(groups) != 3 {
+		t.Fatalf("len(groups) != 3, = %d", len(groups))
+	}
+
 	if fu := v.GetString("fu"); fu != "" {
 		t.Fatalf("fu != \"\", = %s", fu)
 	}
@@ -881,6 +892,14 @@ func TestMergeConfig(t *testing.T) {
 		t.Fatalf("len(universe) != 2, = %d", len(universe))
 	}
 
+	if groups := v.GetIntSlice("hello.groups"); len(groups) != 3 {
+		t.Fatalf("len(groups) != 3, = %d", len(groups))
+	}
+
+	if ages := v.GetIntSlice("hello.ages"); len(ages) != 2 {
+		t.Fatalf("len(ages) != 2, = %d", len(ages))
+	}
+
 	if fu := v.GetString("fu"); fu != "bar" {
 		t.Fatalf("fu != \"bar\", = %s", fu)
 	}
@@ -901,6 +920,10 @@ func TestMergeConfigNoMerge(t *testing.T) {
 		t.Fatalf("len(world) != 4, = %d", len(world))
 	}
 
+	if groups := v.GetIntSlice("hello.groups"); len(groups) != 3 {
+		t.Fatalf("len(groups) != 3, = %d", len(groups))
+	}
+
 	if fu := v.GetString("fu"); fu != "" {
 		t.Fatalf("fu != \"\", = %s", fu)
 	}
@@ -921,6 +944,14 @@ func TestMergeConfigNoMerge(t *testing.T) {
 		t.Fatalf("len(universe) != 2, = %d", len(universe))
 	}
 
+	if groups := v.GetIntSlice("hello.groups"); len(groups) != 0 {
+		t.Fatalf("len(groups) != 0, = %d", len(groups))
+	}
+
+	if ages := v.GetIntSlice("hello.ages"); len(ages) != 2 {
+		t.Fatalf("len(groups) != 2, = %d", len(ages))
+	}
+
 	if fu := v.GetString("fu"); fu != "bar" {
 		t.Fatalf("fu != \"bar\", = %s", fu)
 	}