117 lines
2.3 KiB
Go
117 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/joho/godotenv"
|
|
"golang.captainalm.com/GOPackageHeaderServer/conf"
|
|
"golang.captainalm.com/GOPackageHeaderServer/web"
|
|
"gopkg.in/yaml.v3"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"path"
|
|
"path/filepath"
|
|
"sync"
|
|
"syscall"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
buildVersion = "develop"
|
|
buildDate = ""
|
|
)
|
|
|
|
func main() {
|
|
log.Printf("[Main] Starting up GO Package Header Server #%s (%s)\n", buildVersion, buildDate)
|
|
y := time.Now()
|
|
|
|
//Hold main thread till safe shutdown exit:
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(1)
|
|
|
|
//Get working directory:
|
|
cwdDir, err := os.Getwd()
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
//Load environment file:
|
|
err = godotenv.Load()
|
|
if err != nil {
|
|
log.Fatalln("Error loading .env file")
|
|
}
|
|
|
|
//Data directory processing:
|
|
dataDir := os.Getenv("DIR_DATA")
|
|
if dataDir == "" {
|
|
dataDir = path.Join(cwdDir, ".data")
|
|
}
|
|
|
|
check(os.MkdirAll(dataDir, 0777))
|
|
|
|
//Config file processing:
|
|
configLocation := os.Getenv("CONFIG_FILE")
|
|
if configLocation == "" {
|
|
configLocation = path.Join(dataDir, "config.yml")
|
|
} else {
|
|
if !filepath.IsAbs(configLocation) {
|
|
configLocation = path.Join(dataDir, configLocation)
|
|
}
|
|
}
|
|
|
|
//Config loading:
|
|
configFile, err := os.Open(configLocation)
|
|
if err != nil {
|
|
log.Fatalln("Failed to open config.yml")
|
|
}
|
|
|
|
var configYml conf.ConfigYaml
|
|
groupsDecoder := yaml.NewDecoder(configFile)
|
|
err = groupsDecoder.Decode(&configYml)
|
|
if err != nil {
|
|
log.Fatalln("Failed to parse config.yml:", err)
|
|
}
|
|
|
|
//Server definitions:
|
|
log.Printf("[Main] Starting up HTTP server on %s...\n", configYml.Listen.Web)
|
|
webServer, _ := web.New(configYml)
|
|
|
|
//=====================
|
|
// Safe shutdown
|
|
sigs := make(chan os.Signal, 1)
|
|
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
//Startup complete:
|
|
z := time.Now().Sub(y)
|
|
log.Printf("[Main] Took '%s' to fully initialize modules\n", z.String())
|
|
|
|
go func() {
|
|
<-sigs
|
|
fmt.Printf("\n")
|
|
|
|
log.Printf("[Main] Attempting safe shutdown\n")
|
|
a := time.Now()
|
|
|
|
log.Printf("[Main] Shutting down HTTP server...\n")
|
|
err := webServer.Close()
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
log.Printf("[Main] Signalling program exit...\n")
|
|
b := time.Now().Sub(a)
|
|
log.Printf("[Main] Took '%s' to fully shutdown modules\n", b.String())
|
|
wg.Done()
|
|
}()
|
|
//
|
|
//=====================
|
|
wg.Wait()
|
|
log.Println("[Main] Goodbye")
|
|
}
|
|
|
|
func check(err error) {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|