use configfromenv

Author: akerl

config.go

@@ -6,7 +6,6 @@ import (
 	"os"
 
 	"github.com/akerl/go-lambda/s3"
-	"github.com/ghodss/yaml"
 )
 
 type configFile struct {
@@ -23,19 +22,7 @@ type configFile struct {
 
 func loadConfig() (*configFile, error) {
 	c := configFile{}
-
-	bucket := os.Getenv("S3_BUCKET")
-	path := os.Getenv("S3_KEY")
-	if bucket == "" || path == "" {
-		return &c, fmt.Errorf("config location not provided")
-	}
-
-	obj, err := s3.GetObject(bucket, path)
-	if err != nil {
-		return &c, err
-	}
-
-	err = yaml.Unmarshal(obj, &c)
+	err := s3.GetConfigFromEnv(&c)
 	if err != nil {
 		return &c, err
 	}