Merge pull request #3 from snorwin/additional-env-vars

Load additional environment vars dynamically from file

Author: snorwin

main.go

@@ -32,6 +32,7 @@ func main() {
 	cmd := exec.Command(executable, os.Args[1:]...)
 	cmd.Stdout = os.Stdout
 	cmd.Stderr = os.Stderr
+	cmd.Env = utils.LoadEnvFile()
 	if err := cmd.AsyncRun(); err != nil {
 		log.Emergency(err.Error())
 		os.Exit(1)
@@ -72,6 +73,7 @@ func main() {
 			tmp := exec.Command(executable, append([]string{"-x", utils.LookupHAProxySocketPath(), "-sf", strconv.Itoa(cmd.Process.Pid)}, os.Args[1:]...)...)
 			tmp.Stdout = os.Stdout
 			tmp.Stderr = os.Stderr
+			tmp.Env = utils.LoadEnvFile()
 			if err := tmp.AsyncRun(); err != nil {
 				log.Warning(err.Error())
 				log.Warning("reload failed")

pkg/utils/utils.go

@@ -4,6 +4,7 @@ import (
 	"os"
 	"os/exec"
 	"path/filepath"
+	"strings"
 )
 
 // LookupExecutablePathAbs lookup the $PATH to find the absolut path of an executable
@@ -36,3 +37,16 @@ func LookupHAProxySocketPath() string {
 
 	return "/var/run/haproxy.sock"
 }
+
+// LoadEnvFile load additional dynamic environment variables from a file which contains them in the form "key=value".
+func LoadEnvFile() []string {
+	env := os.Environ()
+
+	if file, ok := os.LookupEnv("ENV_FILE"); ok {
+		if data, err := os.ReadFile(file); err != nil {
+			env = append(env, strings.Split(string(data), "/n")...)
+		}
+	}
+
+	return env
+}