photoprism/internal/server/server.go

package server

import (
	"fmt"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/photoprism/photoprism/internal/config"
	log "github.com/sirupsen/logrus"
)

// Start the REST API server using the configuration provided
func Start(conf *config.Config) {
	if conf.HttpServerMode() != "" {
		gin.SetMode(conf.HttpServerMode())
	} else if conf.Debug() == false {
		gin.SetMode(gin.ReleaseMode)
	}

	router := gin.New()
	router.Use(gin.Logger(), gin.Recovery())

	// Set template directory
	router.LoadHTMLGlob(conf.HttpTemplatesPath() + "/*")

	registerRoutes(router, conf)

	server := &http.Server{
		Addr:    fmt.Sprintf("%s:%d", conf.HttpServerHost(), conf.HttpServerPort()),
		Handler: router,
	}

	quit := make(chan os.Signal)

	/*
		    TODO: Use a Context for graceful shutdown of web and database servers (and other goroutines)

			TODO: Add web server tests

			See
			- https://github.com/gin-gonic/gin/blob/dfe37ea6f1b9127be4cff4822a1308b4349444e0/examples/graceful-shutdown/graceful-shutdown/server.go
			- https://stackoverflow.com/questions/45500836/close-multiple-goroutine-if-an-error-occurs-in-one-in-go
	*/

	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)

	go func() {
		<-quit
		log.Info("received interrupt signal - shutting down")

		conf.Shutdown()

		if err := server.Close(); err != nil {
			log.Errorf("server close: %s", err)
		}
	}()

	if err := server.ListenAndServe(); err != nil {
		if err == http.ErrServerClosed {
			log.Info("web server closed")
		} else {
			log.Errorf("web server closed unexpect: %s", err)
		}
	}

	log.Info("please come back another time")

	time.Sleep(2 * time.Second)
}