photoprism/internal/server/server.go

72 lines
1.6 KiB
Go
Raw Normal View History

package server
import (
"fmt"
"net/http"
"os"
"os/signal"
"syscall"
"time"
2018-09-06 14:47:32 +02:00
"github.com/gin-gonic/gin"
"github.com/photoprism/photoprism/internal/context"
log "github.com/sirupsen/logrus"
)
2018-11-06 19:02:03 +01:00
// Start the REST API server using the configuration provided
func Start(ctx *context.Context) {
if ctx.HttpServerMode() != "" {
gin.SetMode(ctx.HttpServerMode())
} else if ctx.Debug() == false {
gin.SetMode(gin.ReleaseMode)
}
router := gin.New()
router.Use(gin.Logger(), gin.Recovery())
// Set template directory
router.LoadHTMLGlob(ctx.HttpTemplatesPath() + "/*")
registerRoutes(router, ctx)
server := &http.Server{
Addr: fmt.Sprintf("%s:%d", ctx.HttpServerHost(), ctx.HttpServerPort()),
Handler: router,
}
quit := make(chan os.Signal)
/*
TODO: Use context for graceful shutdown of all services and add HTTP / TiDB 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")
ctx.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)
}