Minimal graceful Go 1.8+ server example

minimal-graceful-server.go

package main

import (
	"encoding/json"
	"log"
	"net/http"
	"os"

	env "github.com/TV4/env"
	graceful "github.com/TV4/graceful"
)

const defaultPort = "2017"

func main() {
	graceful.LogListenAndServe(setup(env.DefaultClient))
}

func setup(e env.Client) *http.Server {
	port := e.String("PORT", defaultPort)

	return &http.Server{
		Addr: ":" + port,
		Handler: newServer(func(s *server) {
			s.logger = log.New(os.Stdout, "", 0)
		}),
	}
}

type logger interface {
	Printf(format string, v ...interface{})
}

type response map[string]interface{}

type server struct {
	logger  logger
	handler http.Handler
}

func newServer(options ...func(*server)) *server {
	s := &server{logger: graceful.DefaultLogger}

	for _, f := range options {
		f(s)
	}

	s.register()

	return s
}

func (s *server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	s.handler.ServeHTTP(w, r)
}

func (s *server) register() {
	mux := http.NewServeMux()

	mux.Handle("/", http.HandlerFunc(s.hello))

	s.handler = mux
}

func (s *server) hello(w http.ResponseWriter, r *http.Request) {
	s.logger.Printf("Hello request request_id=%q count#requests.hello=1\n",
		r.Header.Get("X-Request-ID"))

	if name := r.URL.Query().Get("name"); name != "" {
		writeJSON(w, response{"hello": name})
	} else {
		writeJSON(w, response{"error": "missing name"}, http.StatusBadRequest)
	}
}

func writeJSON(w http.ResponseWriter, v interface{}, statuses ...int) {
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Header().Add("Vary", "Accept-Encoding")

	if len(statuses) > 0 {
		w.WriteHeader(statuses[0])
	}

	json.NewEncoder(w).Encode(v)
}

minimal graceful server, with handler that implements the Shutdowner interface

package main

import (
	"context"
	"fmt"
	"net/http"

	graceful "github.com/TV4/graceful"
)

func main() {
	graceful.LogListenAndServe(&http.Server{
		Addr:    ":8080",
		Handler: &server{},
	})
}

type server struct{}

func (s *server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "Hello world!")
}

func (s *server) Shutdown(ctx context.Context) error {
	// Here you can do anything that you need to do
	// after the *http.Server has stopped accepting
	// new connections and finished its Shutdown

	// The ctx is the same as the context used to
	// perform *https.Server.Shutdown and thus
	// shares the timeout (15 seconds by default)

	fmt.Println("Finished *server.Shutdown")

	return nil
}

$ go run minimal-shutdown.go
Listening on http://0.0.0.0:8080
^C
Server shutdown with timeout: 15s
Finished all in-flight HTTP requests
Shutting down handler with timeout: 15s
Finished *server.Shutdown
Shutdown finished 15s before deadline