Blizzard
80 lines
2.8 KiB
Go
80 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"sundynix-micro-go/app/gateway/internal/config"
|
|
"sundynix-micro-go/app/gateway/internal/handler"
|
|
"sundynix-micro-go/app/gateway/internal/middleware"
|
|
"sundynix-micro-go/app/system/rpc/systemservice"
|
|
|
|
"github.com/zeromicro/go-zero/core/conf"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"github.com/zeromicro/go-zero/core/stat"
|
|
"github.com/zeromicro/go-zero/zrpc"
|
|
)
|
|
|
|
var configFile = flag.String("f", "etc/gateway.yaml", "the config file")
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
stat.DisableLog()
|
|
|
|
var c config.Config
|
|
conf.MustLoad(*configFile, &c)
|
|
|
|
// 手动初始化日志（自定义 gateway 没用 rest.MustNewServer，需要显式设置）
|
|
logx.MustSetup(c.Log)
|
|
|
|
// 初始化 system-rpc 客户端（用于写操作日志）
|
|
systemRpc := systemservice.NewSystemService(zrpc.MustNewClient(c.SystemRpc))
|
|
|
|
// 构建反向代理路由器
|
|
proxyRouter := handler.NewProxyRouter(c.Upstreams)
|
|
|
|
// 构建中间件链（执行顺序：操作日志 → 鉴权 → CORS → 代理）
|
|
// 注意：由于是外层包内层，实际请求流程是：OpLog → Auth → CORS → Proxy → CORS → Auth → OpLog
|
|
var finalHandler http.Handler = proxyRouter
|
|
|
|
// 注入 CORS 中间件
|
|
if c.Cors.Enable {
|
|
corsMiddleware := middleware.NewCorsMiddleware(c.Cors.AllowOrigins, c.Cors.AllowMethods, c.Cors.AllowHeaders)
|
|
finalHandler = http.HandlerFunc(corsMiddleware.Handle(func(w http.ResponseWriter, r *http.Request) {
|
|
proxyRouter.ServeHTTP(w, r)
|
|
}))
|
|
}
|
|
|
|
// 注入 JWT 鉴权中间件（含滑动窗口续期）
|
|
authMiddleware := middleware.NewAuthMiddleware(c.JwtSecret, c.AuthWhitelist)
|
|
finalHandler = authMiddleware.Handle(finalHandler)
|
|
|
|
// 注入操作日志中间件（在最外层，记录完整的请求-响应周期，含鉴权状态）
|
|
opLogMiddleware := middleware.NewOperationLogMiddleware(systemRpc, c.JwtSecret)
|
|
finalHandler = opLogMiddleware.Handle(finalHandler)
|
|
|
|
// 健康检查
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
fmt.Fprintf(w, `{"code":200,"msg":"gateway is healthy","upstreams":%d}`, len(c.Upstreams))
|
|
})
|
|
mux.Handle("/", finalHandler)
|
|
|
|
addr := fmt.Sprintf("%s:%d", c.Host, c.Port)
|
|
logx.Infof("===== Sundynix Gateway 启动 =====")
|
|
logx.Infof("监听地址: %s", addr)
|
|
logx.Infof("路由数量: %d", len(c.Upstreams))
|
|
for _, u := range c.Upstreams {
|
|
logx.Infof(" %s -> %s", u.Prefix, u.Target)
|
|
}
|
|
logx.Infof("中间件: CORS | JWT鉴权(滑动续期) | 操作日志→system-rpc")
|
|
logx.Infof("鉴权白名单: %d 条", len(c.AuthWhitelist))
|
|
logx.Infof("================================")
|
|
|
|
if err := http.ListenAndServe(addr, mux); err != nil {
|
|
logx.Errorf("网关启动失败: %v", err)
|
|
}
|
|
}
|