server_ui.go

package main

import (
	"github.com/sirupsen/logrus"
	"github.com/spf13/viper"
	"github.com/tylerb/graceful"
	"net/http"
	"time"
        "log"
	"net"
        "os"
	"github.com/CloudInstall/application"
	"github.com/CloudInstall/handlers"
)

func newConfig() (*viper.Viper, error) {
	c := viper.New()
	c.SetDefault("cookie_secret", "00MTi32Qr4BM1u5D")
	serverAddress1 := GetSystemIP()+ ":8888"
//	c.SetDefault("http_addr", "10.73.119.127:8888")
	c.SetDefault("http_addr", serverAddress1)
	c.SetDefault("http_cert_file", "")
	c.SetDefault("http_key_file", "")
	c.SetDefault("http_drain_interval", "1s")

	c.AutomaticEnv()

	return c, nil
}

// Get preferred outbound ip of this machine
func GetSystemIP() string {
	conn, err := net.Dial("udp", "8.8.8.8:80")
	if err != nil {
		log.Fatal(err)
	}
	defer conn.Close()

	localAddr := conn.LocalAddr().(*net.UDPAddr)

	return localAddr.IP.String()
}

func main() {
	handlers.OnboarderAddr = os.Getenv("ONBOARDER_API_ADDRESS")
	if handlers.OnboarderAddr == "" {
		log.Fatalf("Provide \"ONBOARDER_API_ADDRESS\" environment variable")
	}
	config, err := newConfig()
	if err != nil {
		logrus.Fatal(err)
	}

	app, err := application.New(config)
	if err != nil {
		logrus.Fatal(err)
	}

	middle, err := app.MiddlewareStruct()
	if err != nil {
		logrus.Fatal(err)
	}

	//serverAddress := config.Get("http_addr").(string)
	serverAddress := GetSystemIP()+ ":8888"
	certFile := config.Get("http_cert_file").(string)
	keyFile := config.Get("http_key_file").(string)
	drainIntervalString := config.Get("http_drain_interval").(string)

	drainInterval, err := time.ParseDuration(drainIntervalString)
	if err != nil {
		logrus.Fatal(err)
	}

	srv := &graceful.Server{
		Timeout: drainInterval,
		Server:  &http.Server{Addr: serverAddress, Handler: middle},
	}

	logrus.Infoln("Running HTTP server on " + serverAddress)

	if certFile != "" && keyFile != "" {
		err = srv.ListenAndServeTLS(certFile, keyFile)
	} else {
		err = srv.ListenAndServe()
	}

	if err != nil {
		logrus.Fatal(err)
	}
}