127 lines
2.2 KiB
Go
127 lines
2.2 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"proxy-server/pkg/utils"
|
|
"proxy-server/server/debug"
|
|
"proxy-server/server/fwd"
|
|
"proxy-server/server/pkg/env"
|
|
"proxy-server/server/pkg/log"
|
|
"proxy-server/server/pkg/orm"
|
|
"proxy-server/server/web"
|
|
"runtime"
|
|
"sync"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
import _ "net/http/pprof"
|
|
|
|
type Context struct {
|
|
context.Context
|
|
log *slog.Logger
|
|
}
|
|
|
|
func Start() {
|
|
|
|
// 初始化
|
|
err := godotenv.Load()
|
|
if err != nil {
|
|
println("没有本地环境变量文件")
|
|
}
|
|
|
|
log.Init()
|
|
env.Init()
|
|
orm.Init()
|
|
|
|
// 退出信号
|
|
osQuit := make(chan os.Signal)
|
|
signal.Notify(osQuit, os.Interrupt, syscall.SIGTERM)
|
|
|
|
// 启动服务
|
|
slog.Info("启动服务")
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
wg := sync.WaitGroup{}
|
|
|
|
wg.Add(1)
|
|
fwdQuit := make(chan struct{}, 1)
|
|
go func() {
|
|
defer wg.Done()
|
|
defer close(fwdQuit)
|
|
err := startFwdServer(ctx)
|
|
if err != nil {
|
|
slog.Error("代理服务发生错误", "err", err)
|
|
}
|
|
fwdQuit <- struct{}{}
|
|
}()
|
|
|
|
// 启动 web 服务
|
|
wg.Add(1)
|
|
apiQuit := make(chan struct{}, 1)
|
|
go func() {
|
|
defer wg.Done()
|
|
err := startWebServer(ctx)
|
|
if err != nil {
|
|
slog.Error("web 服务发生错误", "err", err)
|
|
}
|
|
apiQuit <- struct{}{}
|
|
}()
|
|
|
|
// debug
|
|
go func() {
|
|
debug.Start(ctx)
|
|
}()
|
|
|
|
// 性能监控
|
|
go func() {
|
|
runtime.SetBlockProfileRate(1)
|
|
err := http.ListenAndServe(":6060", nil)
|
|
if err != nil {
|
|
slog.Error("性能监控服务发生错误", "err", err)
|
|
}
|
|
}()
|
|
|
|
// 等待退出信号
|
|
select {
|
|
case <-osQuit:
|
|
slog.Info("服务主动退出")
|
|
case <-fwdQuit:
|
|
slog.Warn("fwd 服务异常退出")
|
|
case <-apiQuit:
|
|
slog.Warn("web 服务异常退出")
|
|
}
|
|
// 退出其他服务
|
|
cancel()
|
|
|
|
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
select {
|
|
case <-utils.ChanWgWait(timeout, &wg):
|
|
slog.Info("服务已退出")
|
|
case <-timeout.Done():
|
|
slog.Warn("退出超时，强制退出")
|
|
}
|
|
}
|
|
|
|
func startFwdServer(ctx context.Context) error {
|
|
server := fwd.New(nil)
|
|
go func() {
|
|
<-ctx.Done()
|
|
server.Close()
|
|
}()
|
|
server.Run()
|
|
return nil
|
|
}
|
|
|
|
func startWebServer(ctx context.Context) error {
|
|
return web.Start(ctx)
|
|
}
|