69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
package proxy
|
||
|
||
import (
|
||
"strings"
|
||
"sync"
|
||
|
||
"github.com/gofiber/fiber/v3"
|
||
|
||
"github.com/any-hub/any-hub/internal/server"
|
||
)
|
||
|
||
// Forwarder 根据 HubRoute 的 module_key 选择对应的 ProxyHandler，默认回退到构造时注入的 handler。
|
||
type Forwarder struct {
|
||
defaultHandler server.ProxyHandler
|
||
}
|
||
|
||
// NewForwarder 创建 Forwarder，defaultHandler 不能为空。
|
||
func NewForwarder(defaultHandler server.ProxyHandler) *Forwarder {
|
||
return &Forwarder{defaultHandler: defaultHandler}
|
||
}
|
||
|
||
var (
|
||
moduleHandlers sync.Map
|
||
)
|
||
|
||
// RegisterModuleHandler 将特定 module_key 映射到 ProxyHandler，重复注册会覆盖旧值。
|
||
func RegisterModuleHandler(key string, handler server.ProxyHandler) {
|
||
normalized := normalizeModuleKey(key)
|
||
if normalized == "" || handler == nil {
|
||
return
|
||
}
|
||
moduleHandlers.Store(normalized, handler)
|
||
}
|
||
|
||
// Handle 实现 server.ProxyHandler，根据 route.ModuleKey 选择 handler。
|
||
func (f *Forwarder) Handle(c fiber.Ctx, route *server.HubRoute) error {
|
||
handler := f.lookup(route)
|
||
if handler == nil {
|
||
return fiber.NewError(fiber.StatusInternalServerError, "proxy handler unavailable")
|
||
}
|
||
return handler.Handle(c, route)
|
||
}
|
||
|
||
func (f *Forwarder) lookup(route *server.HubRoute) server.ProxyHandler {
|
||
if route != nil {
|
||
if handler := lookupModuleHandler(route.ModuleKey); handler != nil {
|
||
return handler
|
||
}
|
||
}
|
||
return f.defaultHandler
|
||
}
|
||
|
||
func lookupModuleHandler(key string) server.ProxyHandler {
|
||
normalized := normalizeModuleKey(key)
|
||
if normalized == "" {
|
||
return nil
|
||
}
|
||
if value, ok := moduleHandlers.Load(normalized); ok {
|
||
if handler, ok := value.(server.ProxyHandler); ok {
|
||
return handler
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func normalizeModuleKey(key string) string {
|
||
return strings.ToLower(strings.TrimSpace(key))
|
||
}
|