Blizzard
69 lines
2.1 KiB
Go
69 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
|
|
"sundynix-micro-go/app/system/rpc/systemservice"
|
|
"sundynix-micro-go/app/zero-gateway/internal/config"
|
|
"sundynix-micro-go/app/zero-gateway/internal/middleware"
|
|
|
|
"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/gateway"
|
|
"github.com/zeromicro/go-zero/rest"
|
|
"github.com/zeromicro/go-zero/zrpc"
|
|
)
|
|
|
|
var configFile = flag.String("f", "etc/zero-gateway.yaml", "the config file")
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
stat.DisableLog()
|
|
|
|
var c config.Config
|
|
conf.MustLoad(*configFile, &c)
|
|
|
|
// 初始化 system-rpc 客户端（用于异步写入操作日志）
|
|
systemRpc := systemservice.NewSystemService(zrpc.MustNewClient(c.SystemRpc))
|
|
|
|
// 构建各中间件实例
|
|
corsMiddleware := middleware.NewCorsMiddleware(
|
|
c.Cors.AllowOrigins,
|
|
c.Cors.AllowMethods,
|
|
c.Cors.AllowHeaders,
|
|
)
|
|
authMiddleware := middleware.NewAuthMiddleware(c.JwtSecret, c.AuthWhitelist)
|
|
opLogMiddleware := middleware.NewOperationLogMiddleware(systemRpc, c.JwtSecret)
|
|
|
|
// 构建官方 Gateway，注入中间件（执行顺序：CORS → Auth → OpLog → 上游转发）
|
|
gw := gateway.MustNewServer(c.GatewayConf,
|
|
gateway.WithMiddleware(
|
|
rest.Middleware(corsMiddleware.Handle), // 跨域
|
|
rest.Middleware(authMiddleware.Handle), // JWT 鉴权
|
|
rest.Middleware(opLogMiddleware.Handle), // 操作日志（异步写入 system-rpc）
|
|
),
|
|
)
|
|
defer gw.Stop()
|
|
|
|
fmt.Println("===== Sundynix Zero-Gateway (Official) =====")
|
|
logx.Infof("监听地址: %s:%d", c.Host, c.Port)
|
|
logx.Infof("上游服务: %d 个", len(c.Upstreams))
|
|
logx.Infof("中间件: CORS | JWT鉴权 | 操作日志→system-rpc")
|
|
logx.Infof("鉴权白名单: %d 条", len(c.AuthWhitelist))
|
|
for _, u := range c.Upstreams {
|
|
if u.Http.Target != "" {
|
|
logx.Infof(" [HTTP] %-15s -> %s (prefix: %s, routes: %d)",
|
|
u.Name, u.Http.Target, u.Http.Prefix, len(u.Mappings))
|
|
}
|
|
if u.Grpc.Target != "" {
|
|
logx.Infof(" [gRPC] %-15s -> %s (routes: %d)",
|
|
u.Name, u.Grpc.Target, len(u.Mappings))
|
|
}
|
|
}
|
|
fmt.Println("=============================================")
|
|
|
|
gw.Start()
|
|
}
|