DMAC/cmd/server/main.go

package main

import (
	"fmt"
	_ "net/http"
	_ "net/http/pprof"
	"os/signal"
	"syscall"

	//"flag"
	//"log"
	"FRMS/internal/pkg/config"
	"FRMS/internal/pkg/logging"
	"FRMS/internal/pkg/server"
	"os"

	"github.com/spf13/viper"
)

type coordinator interface {
	Start()
}

func NewCoordinator(config *viper.Viper, ch chan error) coordinator {
	return server.NewCentralCoordinator(config, ch)
}

func NewConfig(fname string) *viper.Viper {
	return config.LoadConfig(fname)
}

func main() {
	// lets get this bread

	gracefulShutdown := make(chan os.Signal, 1)
	signal.Notify(gracefulShutdown, syscall.SIGINT, syscall.SIGTERM)
	conf := NewConfig("server")
	errCh := make(chan error)

	//fmt.Printf("Listening on %v\n", lport)
	c := NewCoordinator(conf, errCh)
	go c.Start()
	logging.Debug(logging.DStart, "CCO 01 Server %s started", conf.Get("name"))
	select {
	case err := <-errCh: // blocking to wait for any errors and keep alive otherwise
		panic(err)
	case <-gracefulShutdown:
		// Shutdown via INT
		// storing config
		fmt.Printf("\nStoring config to %s\n", conf.ConfigFileUsed())
		if err := conf.WriteConfig(); err != nil {
			panic(err)
		}
		fmt.Println("Stored config successfully. Exiting...")
		os.Exit(0)
	}
}