main.go

package main

import (
	"os"
	"log"
	"net/http"
	"time"
	"strconv"

	"github.com/rs/xaccess"
	"github.com/rs/xhandler"
	"github.com/rs/xlog"
	"github.com/rs/xmux"
	"github.com/okzk/sdnotify"
)

var (
	cfg config
	mw  middleware
	uno *unoconv
)

func init() {
	// read config data
	cfg.initDefaultConfig()

	uno = initUnoconv()

	//plug the xlog handler's input to Go's default logger
	log.SetFlags(0)
	xlogger := xlog.New(cfg.loggerConfig)
	log.SetOutput(xlogger)

	//register some middleware handlers
	mw.initCommonHandlers(
		xlog.NewHandler(cfg.loggerConfig),
		xaccess.NewHandler(),
	)
	sdnotify.Ready()
	watchdog(uno)
}


func watchdog(uno *unoconv) {
	var watchdog_usec time.Duration

	if value, ok := os.LookupEnv("WATCHDOG_USEC"); ok {
		if v, err := strconv.Atoi(value); err == nil {
			watchdog_usec = time.Duration(v) * time.Microsecond
		} else {
			return
		}
	} else {
		return
	}

	go func(uno *unoconv, watchdog_usec time.Duration) {
		for {
			time.Sleep(watchdog_usec / 2)
			if checkUnoconv(uno) {
				sdnotify.Watchdog()
			}
		}
	}(uno, watchdog_usec)
}


func getEnv(key, fallback string) string {
	if value, ok := os.LookupEnv(key); ok {
		return value
	}
	return fallback
}

func main() {
	router := xmux.New()
	addr := getEnv("LISTEN_ADDR", ":3000")
	certFile := getEnv("LISTEN_CERTFILE", "")
	keyFile := getEnv("LISTEN_KEYFILE", "")

	router.GET("/unoconv/health", xhandler.HandlerFuncC(healthHandler))
	router.POST("/unoconv/:filetype", xhandler.HandlerFuncC(unoconvHandler))
	if certFile != "" && keyFile != "" {
		log.Fatal(http.ListenAndServeTLS(addr, certFile, keyFile, mw.Handler(router)))
	}
	log.Fatal(http.ListenAndServe(addr, mw.Handler(router)))
}