Support for slices

README.md

@@ -47,4 +47,8 @@ Env currently supports the following data types.  If you'd like to have more, pl
 - int, int8, int16, int32, int64
 - uint, uint8, uint16, uint32, uint64
 - float32, float64
-- time.Duration
+- time.Duration
+
+## Todo
+
+- [ ] Allow user to provide custom delimiter for slice types

env.go

@@ -98,6 +98,12 @@ func setField(t reflect.StructField, v reflect.Value, value string) (err error)
 		return
 	}
 
+	// if the given type is a slice, create a slice and return,
+	// otherwise, we're dealing with a primitive type
+	if v.Kind() == reflect.Slice {
+		return setSlice(t, v, value)
+	}
+
 	if err = setBuiltInField(v, value); err != nil {
 		return errors.Wrapf(err, "error setting %s", t.Name)
 	}

envExampleSet_test.go

@@ -0,0 +1,24 @@
+package env
+
+import (
+	"fmt"
+	"os"
+	"time"
+)
+
+func ExampleSet() {
+	os.Setenv("HOSTS", "NEHOST1:1234, NEHOST2:1234")
+	os.Setenv("PORT", "1234")
+	os.Setenv("PEER_TIMEOUT", "2m500ms")
+
+	config := struct {
+		Hosts              []string      `env:"HOSTS" required:"true"`
+		Port               int16         `env:"PORT" required:"true"`
+		PeerConnectTimeout time.Duration `env:"PEER_TIMEOUT" default:"1s500ms"`
+	}{}
+
+	err := Set(&config)
+
+	fmt.Println(config, err)
+	// OUTPUT: {[NEHOST1:1234 NEHOST2:1234] 1234 2m0.5s} <nil>
+}

env_test.go

@@ -188,6 +188,16 @@ func TestEnvDuration(t *testing.T) {
 	Equals(t, "1m30s", config.Prop.String())
 }
 
+func TestEnvSlice(t *testing.T) {
+	os.Setenv("PROPS", "a, b, c")
+	config := struct {
+		Items []string `env:"PROPS"`
+	}{}
+
+	ErrorNil(t, Set(&config))
+	Equals(t, []string{"a", "b", "c"}, config.Items)
+}
+
 func TestEnvSetUnexportedProperty(t *testing.T) {
 	os.Setenv("PROP", "hello")
 

setter.go

@@ -3,6 +3,7 @@ package env
 import (
 	"reflect"
 	"strconv"
+	"strings"
 	"time"
 )
 
@@ -59,3 +60,23 @@ func setDuration(fieldValue reflect.Value, value string) (err error) {
 	fieldValue.SetInt(d.Nanoseconds())
 	return
 }
+
+func setSlice(t reflect.StructField, v reflect.Value, value string) (err error) {
+	switch v.Type() {
+	case reflect.TypeOf([]string{}):
+		out := setStringSlice(v, value)
+		v.Set(out)
+		return
+	}
+	return
+}
+
+func setStringSlice(v reflect.Value, value string) (out reflect.Value) {
+	out = reflect.MakeSlice(reflect.TypeOf([]string{}), 0, 0)
+
+	for _, part := range strings.Split(value, ",") {
+		out = reflect.Append(out, reflect.ValueOf(strings.Trim(part, " ")))
+	}
+
+	return
+}