96 lines
2.0 KiB
Go
96 lines
2.0 KiB
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
"net"
|
||
|
|
"net/http"
|
||
|
|
"packet-ui/internal/model"
|
||
|
|
"packet-ui/internal/model/server"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Server struct {
|
||
|
|
cfg *model.Configuration
|
||
|
|
*gin.Engine
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
svr *Server
|
||
|
|
)
|
||
|
|
|
||
|
|
func GetSvr() *Server {
|
||
|
|
return svr
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewSvr(cfg *model.Configuration) *Server {
|
||
|
|
s := &Server{
|
||
|
|
cfg: cfg,
|
||
|
|
}
|
||
|
|
s.Init(InitEngine, InitRouter)
|
||
|
|
|
||
|
|
svr = s
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
|
||
|
|
// RunServer 如果初始化失败 cancel
|
||
|
|
func RunServer(cfg *model.Configuration, cancel context.CancelFunc) (httpServer *http.Server) {
|
||
|
|
s := NewSvr(cfg)
|
||
|
|
|
||
|
|
svrConfig := cfg.Svr
|
||
|
|
defer func() {
|
||
|
|
if r := recover(); r != nil {
|
||
|
|
cancel()
|
||
|
|
} else {
|
||
|
|
serverRunSuccess(svrConfig)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
httpSvr := &http.Server{
|
||
|
|
Addr: svrConfig.Address,
|
||
|
|
Handler: s.Engine,
|
||
|
|
ReadTimeout: 120 * time.Second,
|
||
|
|
WriteTimeout: 120 * time.Second,
|
||
|
|
MaxHeaderBytes: 1 << 20,
|
||
|
|
}
|
||
|
|
|
||
|
|
httpServer = httpSvr
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
if err := httpSvr.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||
|
|
zap.S().Panicf("run server err! %v", err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func serverRunSuccess(svr *server.Config) {
|
||
|
|
printIPInfo := func(adds ...string) {
|
||
|
|
info := "svr running, via"
|
||
|
|
for _, add := range adds {
|
||
|
|
info += "\n\thttp://" + add
|
||
|
|
}
|
||
|
|
zap.S().Info(info)
|
||
|
|
}
|
||
|
|
ip, port, _ := net.SplitHostPort(svr.Address)
|
||
|
|
adders, err := net.InterfaceAddrs()
|
||
|
|
if net.ParseIP(ip).IsUnspecified() && err == nil {
|
||
|
|
var successAddress []string
|
||
|
|
for _, addr := range adders {
|
||
|
|
if ipNet, ok := addr.(*net.IPNet); ok {
|
||
|
|
successAddress = append(successAddress, net.JoinHostPort(ipNet.IP.String(), port))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
printIPInfo(successAddress...)
|
||
|
|
} else {
|
||
|
|
printIPInfo(svr.Address)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Dispatch 当配置文件更新的时候可以去调用
|
||
|
|
// 目前使用的全局 config 指针的方式如果发生更改，理论上 server 关联的 model 内置指针对象也会被更改
|
||
|
|
// 需要重新初始化的部分将由本函数予以实现
|
||
|
|
func (s *Server) Dispatch(_ *model.Configuration) {
|
||
|
|
}
|