deployer.go

package main

import (
	"flag"
	"os"

	"github.com/golang/glog"

	"github.com/spf13/viper"
)

// Run start the web server
func Run(fileConfig string) error {
	viper := viper.New()
	viper.SetConfigType("json")
	viper.SetDefault("restartCount", 5)

	if fileConfig == "" {
		viper.SetConfigName("config")
		viper.AddConfigPath("/etc/deployer")
		viper.Set("awsId", os.Getenv("AWS_ACCESS_KEY_ID"))
		viper.Set("awsSecret", os.Getenv("AWS_SECRET_ACCESS_KEY"))
	} else {
		viper.SetConfigFile(fileConfig)
	}

	err := viper.ReadInConfig()
	if err != nil {
		return err
	}

	server := NewServer(viper)
	return server.StartServer()
}

func main() {
	configPath := flag.String("config", "", "The file path to a config file")
	flag.Parse()

	err := Run(*configPath)
	glog.Errorln(err)
}