settings: support ENV

Author: kornrunner

agent/agent.go

@@ -3,6 +3,7 @@ package agent
 import (
 	"fmt"
 	"github.com/garyburd/redigo/redis"
+	"os"
 	"os/exec"
 	"strings"
 )
@@ -58,7 +59,9 @@ func SaveTwemproxyConfig() {
 func RestartTwemproxy() error {
 	Debug("Restarting Twemproxy.")
 	args := strings.Split(Settings.RestartArgs, string(' '))
-	out, err := exec.Command(Settings.RestartCommand, args...).Output()
+	cmd := exec.Command(Settings.RestartCommand, args...)
+	cmd.Env = append (os.Environ(), Settings.RestartEnv)
+	out, err := cmd.Output()
 
 	if err != nil {
 		Debug(fmt.Sprintf("Cannot restart twemproxy. output: %s. error: %s", out, err))

agent/settings.go

@@ -13,6 +13,7 @@ type AgentSettings struct {
 	Sentinels           []string `yaml:"sentinels"`
 	RestartCommand      string `yaml:"restart_command"`
 	RestartArgs         string `yaml:"restart_args"`
+	RestartEnv          string `yaml:"restart_env"`
 	LogFile             string `yaml:"log_file"`
 }