package main

import (
	"context"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"
	"wms/conf"
	"wms/models"
	"wms/pkg/logx"
	"wms/router"
)

func main() {
	// 读取配置
	if err := conf.Init(); err != nil {
		logx.Errorf("config init error! ", err.Error())
		return
	}

	// 日志初始化
	logx.Init(*conf.LogConf)
	defer logx.Sync()

	// 数据库初始化
	if err := models.Init(); err != nil {
		logx.Errorf("db init error! ", err.Error())
		return
	}

	logx.Infof("aps-server start serve...")
	server := &http.Server{
		Addr:         ":" + conf.WebConf.Port,
		Handler:      router.NewRouter(),
		ReadTimeout:  5 * time.Second,
		WriteTimeout: 5 * time.Second,
	}
	go shutdown(server)
	logx.Error(server.ListenAndServe().Error())
}

func shutdown(server *http.Server) {
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGKILL, syscall.SIGQUIT, syscall.SIGINT, syscall.SIGTERM)
	<-quit

	logx.Infof("server exiting...")
	// 创建一个上下文，设置超时时间
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	// 关闭HTTP服务器
	if err := server.Shutdown(ctx); err != nil {
		logx.Infof("服务优雅退出失败: %v", err)
		return
	}
	logx.Infof("server exited")
}