feat: 初次启动

This commit is contained in:
Blizzard
2026-04-27 21:23:13 +08:00
parent e515f6a287
commit bb8ad4d515
148 changed files with 8602 additions and 5678 deletions
+58
View File
@@ -0,0 +1,58 @@
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"
"github.com/zeromicro/go-zero/core/conf"
"github.com/zeromicro/go-zero/core/logx"
)
var configFile = flag.String("f", "etc/gateway.yaml", "the config file")
func main() {
flag.Parse()
var c config.Config
conf.MustLoad(*configFile, &c)
// 构建反向代理路由器
proxyRouter := handler.NewProxyRouter(c.Upstreams)
// 构建请求处理链
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)
}))
}
// 健康检查
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("================================")
if err := http.ListenAndServe(addr, mux); err != nil {
logx.Errorf("网关启动失败: %v", err)
}
}