加入网络层
This commit is contained in:
@@ -1,64 +1,14 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/spf13/viper"
|
||||
"go.uber.org/zap"
|
||||
"golang.org/x/sync/errgroup"
|
||||
"gorm.io/gorm/logger"
|
||||
"os"
|
||||
"os/signal"
|
||||
"fmt"
|
||||
"gateway/app"
|
||||
"github.com/judwhite/go-svc"
|
||||
"syscall"
|
||||
"time"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// 初始化配置与日志
|
||||
initConfig()
|
||||
// 数据库初始化
|
||||
initDB()
|
||||
|
||||
// HTTP 服务配置
|
||||
server := initServer(viper.GetString("PORT"), logger)
|
||||
|
||||
// 生命周期管理
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
g, gCtx := errgroup.WithContext(ctx)
|
||||
|
||||
// 启动 HTTP 服务
|
||||
g.Go(func() error {
|
||||
return server.Start()
|
||||
})
|
||||
|
||||
// 优雅关闭
|
||||
g.Go(func() error {
|
||||
<-gCtx.Done()
|
||||
return server.Shutdown(30 * time.Second)
|
||||
})
|
||||
|
||||
// 信号监听
|
||||
signals := make(chan os.Signal, 1)
|
||||
signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
|
||||
g.Go(func() error {
|
||||
select {
|
||||
case <-gCtx.Done():
|
||||
return gCtx.Err()
|
||||
case sig := <-signals:
|
||||
logger.Info("Received signal", zap.String("signal", sig.String()))
|
||||
cancel() // 触发关闭
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
if err := g.Wait(); err != nil {
|
||||
logger.Fatal("Server exited with error", zap.Error(err))
|
||||
if err := svc.Run(&app.Program{}, syscall.SIGINT, syscall.SIGTERM, syscall.SIGKILL); err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
}
|
||||
|
||||
func initConfig() {
|
||||
|
||||
}
|
||||
|
||||
func initDB() {
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user