package main
|
|
import (
|
"basic.com/valib/logger.git"
|
"context"
|
"flag"
|
"os"
|
"os/signal"
|
"sync"
|
"syscall"
|
"vamicro/config"
|
"vamicro/update-service/models"
|
"vamicro/update-service/routers"
|
"vamicro/update-service/upgrade"
|
"vamicro/update-service/utils"
|
_ "basic.com/valib/version.git"
|
)
|
|
var (
|
env = flag.String("e", "pro", "")
|
)
|
|
func init() {
|
flag.Parse()
|
config.Init(*env)
|
|
var logFile = utils.GetLogFilePath()
|
logger.InitLogger(logFile, config.LogConf.Level, config.LogConf.MaxSize, config.LogConf.MaxBackups, config.LogConf.MaxAge)
|
logger.Info("log init success !")
|
}
|
|
func main() {
|
models.Init()
|
defer models.CloseDB()
|
|
wg := &sync.WaitGroup{}
|
q := make(chan os.Signal, 1)
|
signal.Notify(q, os.Interrupt, os.Kill, syscall.SIGTERM)
|
ctx, cancel := context.WithCancel(context.Background())
|
upgrade.Initialize(ctx, wg)
|
server := routers.InitRouter(wg)
|
<-q
|
logger.Info("user signalled...")
|
|
// shutdown http service gracefully
|
if err := server.Shutdown(ctx); nil != err {
|
logger.Fatal("server shutdown failed, err:", err)
|
}
|
|
cancel()
|
wg.Wait()
|
logger.Info("all sub threads exited...")
|
}
|
