super-graph/internal/serv/serv.go

package serv

import (
	"context"
	"fmt"
	"net/http"
	"os"
	"os/signal"
	"path"
	"strings"
	"time"

	rice "github.com/GeertJohan/go.rice"
	"github.com/NYTimes/gziphandler"
	"github.com/dosco/super-graph/internal/serv/internal/auth"
	"go.opencensus.io/plugin/ochttp"
)

var (
	apiRoute string = "/api/v1/graphql"
)

func initWatcher() {
	cpath := conf.cpath
	if conf != nil && !conf.WatchAndReload {
		return
	}

	var d dir
	if len(cpath) == 0 || cpath == "./" {
		d = Dir("./config", ReExec)
	} else {
		d = Dir(cpath, ReExec)
	}

	go func() {
		err := Do(log.Printf, d)
		if err != nil {
			log.Fatalf("ERR %s", err)
		}
	}()
}

func startHTTP() {
	var appName string

	defaultHP := "0.0.0.0:8080"
	env := os.Getenv("GO_ENV")

	if conf != nil {
		appName = conf.AppName
		hp := strings.SplitN(conf.HostPort, ":", 2)

		if len(hp) == 2 {
			if len(conf.Host) != 0 {
				hp[0] = conf.Host
			}

			if len(conf.Port) != 0 {
				hp[1] = conf.Port
			}

			conf.hostPort = fmt.Sprintf("%s:%s", hp[0], hp[1])
		}
	}

	if len(conf.hostPort) == 0 {
		conf.hostPort = defaultHP
	}

	routes, err := routeHandler()
	if err != nil {
		log.Fatalf("ERR %s", err)
	}

	srv := &http.Server{
		Addr:           conf.hostPort,
		Handler:        routes,
		ReadTimeout:    5 * time.Second,
		WriteTimeout:   10 * time.Second,
		MaxHeaderBytes: 1 << 20,
	}

	if conf.telemetryEnabled() {
		srv.Handler = &ochttp.Handler{Handler: routes}
	}

	idleConnsClosed := make(chan struct{})
	go func() {
		sigint := make(chan os.Signal, 1)
		signal.Notify(sigint, os.Interrupt)
		<-sigint

		if err := srv.Shutdown(context.Background()); err != nil {
			log.Fatalln("INF shutdown signal received")
		}
		close(idleConnsClosed)
	}()

	srv.RegisterOnShutdown(func() {
		if conf.closeFn != nil {
			conf.closeFn()
		}
		db.Close()
		log.Fatalln("INF shutdown complete")
	})

	log.Printf("INF Super Graph started, version: %s, git-branch: %s, host-port: %s, app-name: %s, env: %s\n",
		version, gitBranch, conf.hostPort, appName, env)

	if err := srv.ListenAndServe(); err != http.ErrServerClosed {
		log.Fatalln("INF server closed")
	}

	<-idleConnsClosed
}

func routeHandler() (http.Handler, error) {
	var err error
	mux := http.NewServeMux()

	if conf == nil {
		return mux, nil
	}

	if len(conf.APIPath) != 0 {
		apiRoute = path.Join("/", conf.APIPath, "/v1/graphql")
	}

	routes := map[string]http.Handler{
		"/health": http.HandlerFunc(health),
		apiRoute:  apiV1Handler(),
	}

	if err := setActionRoutes(routes); err != nil {
		return nil, err
	}

	if conf.WebUI {
		routes["/"] = http.FileServer(rice.MustFindBox("./web/build").HTTPBox())
	}

	if conf.HTTPGZip {
		gz := gziphandler.MustNewGzipLevelHandler(6)
		for k, v := range routes {
			routes[k] = gz(v)
		}
	}

	for k, v := range routes {
		mux.Handle(k, v)
	}

	if conf.telemetryEnabled() {
		conf.closeFn, err = enableObservability(mux)
		if err != nil {
			return nil, err
		}
	}

	fn := func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Server", serverName)
		mux.ServeHTTP(w, r)
	}

	return http.HandlerFunc(fn), nil
}

func setActionRoutes(routes map[string]http.Handler) error {
	var err error

	for _, a := range conf.Actions {
		var fn http.Handler

		fn, err = newAction(&a)
		if err != nil {
			break
		}

		p := fmt.Sprintf("/api/v1/actions/%s", strings.ToLower(a.Name))

		if ac := findAuth(a.AuthName); ac != nil {
			routes[p], err = auth.WithAuth(fn, ac)
		} else {
			routes[p] = fn
		}

		if conf.telemetryEnabled() {
			routes[p] = ochttp.WithRouteTag(routes[p], p)
		}

		if err != nil {
			return err
		}
	}
	return nil
}

func findAuth(name string) *auth.Auth {
	for _, a := range conf.Auths {
		if strings.EqualFold(a.Name, name) {
			return &a
		}
	}
	return nil
}