main.go

package main

import (
	"context"
	"fmt"
	"github.com/danthegoodman1/SQLGateway/pg"
	"github.com/danthegoodman1/SQLGateway/red"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/danthegoodman1/SQLGateway/gologger"
	"github.com/danthegoodman1/SQLGateway/http_server"
	"github.com/danthegoodman1/SQLGateway/utils"
)

var logger = gologger.NewLogger()

func main() {
	logger.Info().Msg("starting SQLGateway")

	if err := pg.ConnectToDB(); err != nil {
		logger.Error().Err(err).Msg("error connecting to PG Pool")
		os.Exit(1)
	}

	if utils.REDIS_ADDR != "" {
		if err := red.ConnectRedis(); err != nil {
			logger.Error().Err(err).Msg("error connecting to Redis")
			os.Exit(1)
		}
	}

	//if utils.K8S_SD {
	//	k, err := ksd.NewPodKSD("default", "app=SQLGateway", func(obj *v1.Pod) {
	//		//fmt.Println("ADD EVENT: %+v", obj)
	//	}, func(obj *v1.Pod) {
	//		//fmt.Println("DELETE EVENT: %+v", obj)
	//	}, func(oldObj, newObj *v1.Pod) {
	//		//fmt.Println("UPDATE EVENT OLD: %+v", oldObj)
	//		//fmt.Println("UPDATE EVENT NEW: %+v", newObj)
	//	})
	//	if err != nil {
	//		logger.Error().Err(err).Msg("failed to create new pod ksd")
	//		os.Exit(1)
	//	}
	//	defer k.Stop()
	//}

	pg.Manager = pg.NewTxManager()

	httpServer := http_server.StartHTTPServer()

	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
	<-c
	logger.Warn().Msg("received shutdown signal!")

	// Convert the time to seconds
	sleepTime := utils.GetEnvOrDefaultInt("SHUTDOWN_SLEEP_SEC", 0)
	logger.Info().Msg(fmt.Sprintf("sleeping for %ds before exiting", sleepTime))

	time.Sleep(time.Second * time.Duration(sleepTime))
	logger.Info().Msg(fmt.Sprintf("slept for %ds, exiting", sleepTime))

	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
	defer cancel()
	if err := httpServer.Shutdown(ctx); err != nil {
		logger.Error().Err(err).Msg("failed to shutdown HTTP server")
	} else {
		logger.Info().Msg("successfully shutdown HTTP server")
	}
	if utils.REDIS_ADDR != "" {
		if err := red.Shutdown(ctx); err != nil {
			logger.Error().Err(err).Msg("error shutting down redis connection")
		} else {
			logger.Info().Msg("shut down redis")
		}
	}
	pg.Manager.Shutdown()
	logger.Info().Msg("shut down tx manager")
	os.Exit(0)
}