158 lines
2.7 KiB
Go
158 lines
2.7 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"os"
|
|
"os/signal"
|
|
"proxy-server/pkg/utils"
|
|
"proxy-server/server/fwd"
|
|
"proxy-server/server/pkg/env"
|
|
"proxy-server/server/pkg/orm"
|
|
"proxy-server/server/web"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/joho/godotenv"
|
|
"github.com/lmittmann/tint"
|
|
"github.com/mattn/go-colorable"
|
|
)
|
|
|
|
type Context struct {
|
|
context.Context
|
|
log *slog.Logger
|
|
}
|
|
|
|
func Start() {
|
|
|
|
// 初始化
|
|
initLog()
|
|
env.Init()
|
|
orm.Init()
|
|
|
|
// 启动服务
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
errQuit := make(chan error)
|
|
defer close(errQuit)
|
|
wg := utils.CountWaitGroup{}
|
|
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
err := startFwdServer(ctx)
|
|
if err != nil {
|
|
slog.Error("代理服务发生错误", "err", err)
|
|
}
|
|
errQuit <- err
|
|
}()
|
|
|
|
// 等待退出信号
|
|
osQuit := make(chan os.Signal)
|
|
signal.Notify(osQuit, os.Interrupt, syscall.SIGTERM)
|
|
|
|
select {
|
|
case <-osQuit:
|
|
slog.Info("服务关闭")
|
|
case <-errQuit:
|
|
slog.Error("服务异常退出")
|
|
}
|
|
|
|
// 等待子服务退出
|
|
cancel()
|
|
|
|
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
wgCh := utils.ChanWgWait(timeout, &wg)
|
|
|
|
select {
|
|
case <-timeout.Done():
|
|
slog.Error("关闭超时，强制关闭")
|
|
case <-wgCh:
|
|
slog.Info("服务已退出")
|
|
}
|
|
}
|
|
|
|
func initLog() {
|
|
slog.SetLogLoggerLevel(slog.LevelDebug)
|
|
}
|
|
|
|
func startFwdServer(ctx context.Context) error {
|
|
server := fwd.New(nil)
|
|
|
|
go func() {
|
|
<-ctx.Done()
|
|
server.Close()
|
|
}()
|
|
|
|
server.Run(ctx)
|
|
return nil
|
|
}
|
|
|
|
func startMntServer(ctx context.Context) {
|
|
|
|
}
|
|
|
|
func startWebServer(ctx context.Context) {
|
|
|
|
}
|
|
|
|
func Start2() {
|
|
defer func() {
|
|
err := recover()
|
|
if err != nil {
|
|
slog.Error("服务由于意外的 panic 导致退出", err)
|
|
}
|
|
}()
|
|
|
|
ctx := context.Background()
|
|
|
|
// 初始化环境变量
|
|
err := godotenv.Load()
|
|
if err != nil {
|
|
slog.Debug("没有本地环境变量文件")
|
|
}
|
|
|
|
// 配置日志
|
|
writer := colorable.NewColorable(os.Stdout)
|
|
logger := slog.New(tint.NewHandler(writer, &tint.Options{
|
|
Level: slog.LevelDebug,
|
|
ReplaceAttr: func(_ []string, attr slog.Attr) slog.Attr {
|
|
err, ok := attr.Value.Any().(error)
|
|
if !ok {
|
|
return attr
|
|
}
|
|
return tint.Err(err)
|
|
},
|
|
}))
|
|
slog.SetDefault(logger)
|
|
|
|
// 初始化公共组件
|
|
orm.Init()
|
|
|
|
// 启动子服务
|
|
goCount := 1
|
|
errChan := make(chan error, goCount)
|
|
ctxC, cancel := context.WithCancel(ctx)
|
|
defer cancel()
|
|
|
|
go web.Start(ctxC, errChan)
|
|
// go monitor.Start2(ctxC, errChan)
|
|
slog.Info("服务启动成功")
|
|
|
|
// 监听异常
|
|
well := true
|
|
for i := 0; i < goCount; i++ {
|
|
err := <-errChan
|
|
if err != nil {
|
|
slog.Error("服务异常退出", err)
|
|
if well { // 第一次出错时取消其他服务
|
|
well = false
|
|
cancel()
|
|
}
|
|
}
|
|
}
|
|
close(errChan)
|
|
slog.Info("服务已全部退出")
|
|
}
|