StartAPIServer.go

package main

import (
	"context"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/DataManager-Go/DataManagerServer/services"
	"gorm.io/gorm"

	log "github.com/sirupsen/logrus"
)

// Services
var (
	apiService     *services.APIService     // Handle endpoints
	cleanupService *services.CleanupService // Cleanup db stuff
)

func startAPI() {
	log.Info("Starting version " + version)

	// Create and start required services
	apiService = services.NewAPIService(config, db)
	apiService.Start()

	cleanupService = services.NewClienupService(config, db)
	cleanupService.Start()

	if config.Webserver.Profiling {
		log.Info("Starting in profiling mode")
	}

	// Startup done
	log.Info("Startup completed")

	awaitExit(apiService, db)
}

// Shutdown server gracefully
func awaitExit(httpServer *services.APIService, db *gorm.DB) {
	signalChan := make(chan os.Signal, 1)
	signal.Notify(signalChan, syscall.SIGINT, os.Interrupt, syscall.SIGKILL, syscall.SIGTERM)

	// await os signal
	<-signalChan

	// Create a deadline for the await
	ctx, cancel := context.WithTimeout(context.Background(), time.Second*15)
	defer cancel()

	log.Info("Shutting down server")

	if httpServer.HTTPServer != nil {
		err := httpServer.HTTPServer.Shutdown(ctx)
		if err != nil {
			log.Warn(err)
		}
		log.Info("HTTP server shutdown complete")
	}

	if httpServer.HTTPTLSServer != nil {
		err := httpServer.HTTPTLSServer.Shutdown(ctx)
		if err != nil {
			log.Warn(err)
		}
		log.Info("HTTPs server shutdown complete")
	}

	log.Info("Shutting down complete")
}