Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
0630734119 |
17
app.go
17
app.go
@ -22,7 +22,6 @@ var GatewayConf = Config{
|
||||
|
||||
// GatewayApp 定义网关应用，融合了 WebServer 的原生能力
|
||||
type GatewayApp struct {
|
||||
*service.WebServer
|
||||
rd *redis.Redis
|
||||
pubsubChannel string
|
||||
cancelPubSub context.CancelFunc
|
||||
@ -30,9 +29,7 @@ type GatewayApp struct {
|
||||
|
||||
// NewGatewayApp 创建 Gateway
|
||||
func NewGatewayApp() *GatewayApp {
|
||||
return &GatewayApp{
|
||||
WebServer: service.NewWebServer(),
|
||||
}
|
||||
return &GatewayApp{}
|
||||
}
|
||||
|
||||
// Start 启动网关服务 (实现 starter.Service)
|
||||
@ -62,7 +59,8 @@ func (g *GatewayApp) Start(ctx context.Context, logger *log.Logger) error {
|
||||
}
|
||||
|
||||
// 启动底层的 WebServer，处理所有实际的 HTTP 连接和发现注册
|
||||
return g.WebServer.Start(ctx, logger)
|
||||
service.Start()
|
||||
return nil
|
||||
}
|
||||
|
||||
// Stop 停止网关服务 (实现 starter.Service)
|
||||
@ -74,13 +72,14 @@ func (g *GatewayApp) Stop(ctx context.Context) error {
|
||||
g.rd.Unsubscribe(g.pubsubChannel)
|
||||
}
|
||||
// 停止底层的 WebServer
|
||||
return g.WebServer.Stop(ctx)
|
||||
// 注意：全局 service.Stop 目前在框架中可能由 starter 统一管理，或者手动调用 service.DefaultServer.Stop(ctx)
|
||||
return service.DefaultServer.Stop(ctx)
|
||||
}
|
||||
|
||||
// Reload 重载网关配置 (实现 starter.Reloader)
|
||||
func (g *GatewayApp) Reload() error {
|
||||
// 1. 触发底层 WebServer 的重载 (会重新加载本地 yaml 配置文件中的固定路由策略，并触发 OnReload 钩子)
|
||||
err := g.WebServer.Reload()
|
||||
err := service.DefaultServer.Reload()
|
||||
|
||||
// 2. 重新全量拉取 Redis 中的动态配置
|
||||
logger := log.DefaultLogger
|
||||
@ -92,11 +91,11 @@ func (g *GatewayApp) Reload() error {
|
||||
|
||||
// Status 返回网关运行状态
|
||||
func (g *GatewayApp) Status() (string, error) {
|
||||
addr, err := g.WebServer.Status()
|
||||
addr, err := service.DefaultServer.Status()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
|
||||
hostCount := 0
|
||||
if g.rd != nil {
|
||||
hostCount = len(g.rd.Do("KEYS", GatewayConf.Prefix+":host:*").Strings())
|
||||
|
||||
80
app_test.go
80
app_test.go
@ -1,16 +1,15 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"apigo.cc/go/log"
|
||||
"apigo.cc/go/redis"
|
||||
"apigo.cc/go/service"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"net"
|
||||
gohttp "net/http"
|
||||
"os"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
@ -27,23 +26,27 @@ func TestGateway(t *testing.T) {
|
||||
// 清理测试数据
|
||||
rd.Do("DEL", "gateway:host:gw.test", "gateway:hosts")
|
||||
|
||||
// 2. 启动一个后端测试服务 test-backend
|
||||
service.Config.App = "test-backend"
|
||||
service.Config.Listen = ":0"
|
||||
service.Config.Register = registry
|
||||
// 2. 启动一个独立的后端测试服务 test-backend (不使用全局 DefaultServer)
|
||||
backend := service.NewWebServer()
|
||||
backend.Config.App = "test-backend"
|
||||
backend.Config.Listen = ":0"
|
||||
backend.Config.Register = registry
|
||||
|
||||
service.Host("*").GET("/hello", func(req *service.Request) string {
|
||||
backend.Host("*").GET("/hello", func(req *service.Request) string {
|
||||
return "hello from backend, path: " + req.RequestURI
|
||||
})
|
||||
service.Host("*").GET("/hello/world", func(req *service.Request) string {
|
||||
backend.Host("*").GET("/hello/world", func(req *service.Request) string {
|
||||
return "hello from backend, path: " + req.RequestURI
|
||||
})
|
||||
|
||||
asBackend := service.AsyncStart()
|
||||
defer asBackend.Stop()
|
||||
time.Sleep(200 * time.Millisecond) // 等待后端启动和注册
|
||||
backendCtx, backendCancel := context.WithCancel(context.Background())
|
||||
defer backendCancel()
|
||||
go backend.Start(backendCtx, log.DefaultLogger)
|
||||
defer backend.Stop(backendCtx)
|
||||
|
||||
// 3. 配置 Gateway
|
||||
time.Sleep(500 * time.Millisecond) // 等待后端启动和注册
|
||||
|
||||
// 3. 配置 Gateway (使用全局 service)
|
||||
GatewayConf.Redis = registry
|
||||
GatewayConf.Prefix = "gateway"
|
||||
|
||||
@ -56,10 +59,10 @@ func TestGateway(t *testing.T) {
|
||||
rewriteRules := []service.RewriteRule{
|
||||
{Path: "^/old-api/(.*)$", ToPath: "/api/$1"},
|
||||
}
|
||||
|
||||
|
||||
proxyJson, _ := json.Marshal(proxyRules)
|
||||
rewriteJson, _ := json.Marshal(rewriteRules)
|
||||
|
||||
|
||||
rd.Do("HSET", "gateway:host:gw.test", "proxies", string(proxyJson))
|
||||
rd.Do("HSET", "gateway:host:gw.test", "rewrites", string(rewriteJson))
|
||||
|
||||
@ -67,7 +70,7 @@ func TestGateway(t *testing.T) {
|
||||
tmpDir, _ := os.MkdirTemp("", "gateway-static")
|
||||
defer os.RemoveAll(tmpDir)
|
||||
_ = os.WriteFile(tmpDir+"/index.html", []byte("static content"), 0644)
|
||||
|
||||
|
||||
staticConfig := map[string]string{
|
||||
"/ui": tmpDir,
|
||||
}
|
||||
@ -76,30 +79,32 @@ func TestGateway(t *testing.T) {
|
||||
|
||||
// 4. 启动 Gateway 应用
|
||||
app := NewGatewayApp()
|
||||
|
||||
// 设置独立的端口给 Gateway 的 WebServer 避免与 backend 冲突
|
||||
|
||||
// 设置独立的端口给 Gateway 的 WebServer
|
||||
service.Config.App = "gateway"
|
||||
service.Config.Listen = ":0"
|
||||
// 重置发现，确保网关独立
|
||||
service.SetDiscovererForTest(nil)
|
||||
// 配置网关可以通过 discover 找到 test-backend (网关也需要开启 discover)
|
||||
service.Config.Register = registry
|
||||
service.Config.Calls = map[string]service.CallConfig{
|
||||
"test-backend": {Timeout: 1000},
|
||||
}
|
||||
|
||||
ctx := context.Background()
|
||||
err := app.Start(ctx, log.DefaultLogger)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to start gateway: %v", err)
|
||||
}
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
// 这里假设 GatewayApp.Start 内部会调用 service.Start()，
|
||||
// 而 service.Start() 在新版本中如果是阻塞的，我们需要在 goroutine 中运行
|
||||
go app.Start(ctx, log.DefaultLogger)
|
||||
defer app.Stop(ctx)
|
||||
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
|
||||
_, gwPort, _ := net.SplitHostPort(app.Addr)
|
||||
time.Sleep(500 * time.Millisecond)
|
||||
|
||||
client := &gohttp.Client{Timeout: 2 * time.Second}
|
||||
gwAddr := service.DefaultServer.Addr
|
||||
if gwAddr == "" {
|
||||
t.Fatalf("Gateway address is empty")
|
||||
}
|
||||
_, gwPort, _ := net.SplitHostPort(gwAddr)
|
||||
|
||||
client := &gohttp.Client{Timeout: 5 * time.Second}
|
||||
|
||||
// ============================================
|
||||
// 测试 1: 直接代理匹配 (Discover)
|
||||
@ -129,8 +134,8 @@ func TestGateway(t *testing.T) {
|
||||
res.Body.Close()
|
||||
body = string(b)
|
||||
}
|
||||
if err != nil || !strings.Contains(body, "hello from backend, path: /hello?a=1") {
|
||||
t.Fatalf("Proxy regexp failed, got: %s", body)
|
||||
if err != nil || body != "hello from backend, path: /hello" {
|
||||
t.Fatalf("Proxy /api/ failed, got: %s", body)
|
||||
}
|
||||
|
||||
// ============================================
|
||||
@ -145,8 +150,8 @@ func TestGateway(t *testing.T) {
|
||||
res.Body.Close()
|
||||
body = string(b)
|
||||
}
|
||||
if err != nil || !strings.Contains(body, "hello from backend, path: /hello/world?x=2") {
|
||||
t.Fatalf("Proxy wildcard failed, got: %s", body)
|
||||
if err != nil || body != "hello from backend, path: /hello/world" {
|
||||
t.Fatalf("Proxy /v2/* failed, got: %s", body)
|
||||
}
|
||||
|
||||
// ============================================
|
||||
@ -162,7 +167,7 @@ func TestGateway(t *testing.T) {
|
||||
body = string(b)
|
||||
}
|
||||
if err != nil || body != "hello from backend, path: /hello" {
|
||||
t.Fatalf("Rewrite+Proxy failed, got: %s", body)
|
||||
t.Fatalf("Rewrite + Proxy failed, got: %s", body)
|
||||
}
|
||||
|
||||
// ============================================
|
||||
@ -190,7 +195,7 @@ func TestGateway(t *testing.T) {
|
||||
}
|
||||
newProxyJson, _ := json.Marshal(newProxyRules)
|
||||
rd.Do("HSET", "gateway:host:gw.test", "proxies", string(newProxyJson))
|
||||
|
||||
|
||||
// 发布更新消息
|
||||
rd.Do("PUBLISH", "gateway:channel", `"gw.test"`)
|
||||
|
||||
@ -219,7 +224,6 @@ func TestGateway(t *testing.T) {
|
||||
t.Fatalf("Request failed: %v", err)
|
||||
}
|
||||
if res.StatusCode != 404 {
|
||||
t.Fatalf("Old proxy rule should be deleted (404), got status: %v", res.StatusCode)
|
||||
t.Fatalf("Atomic update failed, old route still exists")
|
||||
}
|
||||
res.Body.Close()
|
||||
}
|
||||
|
||||
34
go.mod
34
go.mod
@ -3,28 +3,28 @@ module apigo.cc/go/gateway
|
||||
go 1.25.0
|
||||
|
||||
require (
|
||||
apigo.cc/go/cast v1.5.0
|
||||
apigo.cc/go/config v1.5.1
|
||||
apigo.cc/go/log v1.5.5
|
||||
apigo.cc/go/redis v1.5.0
|
||||
apigo.cc/go/service v1.5.12
|
||||
apigo.cc/go/starter v1.5.3
|
||||
apigo.cc/go/cast v1.5.2
|
||||
apigo.cc/go/config v1.5.2
|
||||
apigo.cc/go/log v1.5.6
|
||||
apigo.cc/go/redis v1.5.4
|
||||
apigo.cc/go/service v1.5.14
|
||||
apigo.cc/go/starter v1.5.4
|
||||
)
|
||||
|
||||
require apigo.cc/go/jsmod v1.5.0 // indirect
|
||||
require apigo.cc/go/jsmod v1.5.2 // indirect
|
||||
|
||||
require (
|
||||
apigo.cc/go/crypto v1.5.0 // indirect
|
||||
apigo.cc/go/discover v1.5.0 // indirect
|
||||
apigo.cc/go/encoding v1.5.0 // indirect
|
||||
apigo.cc/go/file v1.5.0 // indirect
|
||||
apigo.cc/go/http v1.5.0 // indirect
|
||||
apigo.cc/go/id v1.5.0 // indirect
|
||||
apigo.cc/go/rand v1.5.0 // indirect
|
||||
apigo.cc/go/safe v1.5.0 // indirect
|
||||
apigo.cc/go/shell v1.5.0 // indirect
|
||||
apigo.cc/go/crypto v1.5.2 // indirect
|
||||
apigo.cc/go/discover v1.5.2 // indirect
|
||||
apigo.cc/go/encoding v1.5.3 // indirect
|
||||
apigo.cc/go/file v1.5.4 // indirect
|
||||
apigo.cc/go/http v1.5.2 // indirect
|
||||
apigo.cc/go/id v1.5.3 // indirect
|
||||
apigo.cc/go/rand v1.5.2 // indirect
|
||||
apigo.cc/go/safe v1.5.1 // indirect
|
||||
apigo.cc/go/shell v1.5.2 // indirect
|
||||
apigo.cc/go/timer v1.5.0 // indirect
|
||||
apigo.cc/go/watch v1.5.0 // indirect
|
||||
apigo.cc/go/watch v1.5.1 // indirect
|
||||
github.com/fsnotify/fsnotify v1.10.1 // indirect
|
||||
github.com/gobwas/glob v0.2.3 // indirect
|
||||
github.com/gomodule/redigo v2.0.0+incompatible // indirect
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user