mirror of
https://github.com/hanzoai/gateway.git
synced 2026-10-30 15:04:44 +00:00
zeekayandHanzo Dev
a5c5cc019e
Modules declare go 1.27.1 and the build images follow. go fix applied the modernizers; omitzero is not applied, since it changes what the JSON says. Co-authored-by: Hanzo Dev <[email protected]>
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package plugin
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/hanzoai/gateway/v2/internal/lura/config"
|
|
"github.com/hanzoai/gateway/v2/internal/lura/logging"
|
|
)
|
|
|
|
const Namespace = "github_com/devopsfaith/krakend/transport/http/server/handler"
|
|
const logPrefix = "[PLUGIN: Server]"
|
|
|
|
type RunServer func(context.Context, config.ServiceConfig, http.Handler) error
|
|
|
|
func New(logger logging.Logger, next RunServer) RunServer {
|
|
return func(ctx context.Context, cfg config.ServiceConfig, handler http.Handler) error {
|
|
v, ok := cfg.ExtraConfig[Namespace]
|
|
|
|
if !ok {
|
|
return next(ctx, cfg, handler)
|
|
}
|
|
extra, ok := v.(map[string]any)
|
|
if !ok {
|
|
logger.Debug(logPrefix, "Wrong extra_config type")
|
|
return next(ctx, cfg, handler)
|
|
}
|
|
|
|
// load plugin(s)
|
|
r, ok := serverRegister.Get(Namespace)
|
|
if !ok {
|
|
logger.Debug(logPrefix, "No plugins registered for the module")
|
|
return next(ctx, cfg, handler)
|
|
}
|
|
|
|
name, nameOk := extra["name"].(string)
|
|
fifoRaw, fifoOk := extra["name"].([]any)
|
|
if !nameOk && !fifoOk {
|
|
logger.Debug(logPrefix, "No plugins required in the extra config")
|
|
return next(ctx, cfg, handler)
|
|
}
|
|
var fifo []string
|
|
|
|
if !fifoOk {
|
|
fifo = []string{name}
|
|
} else {
|
|
for _, x := range fifoRaw {
|
|
if v, ok := x.(string); ok {
|
|
fifo = append(fifo, v)
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, name := range fifo {
|
|
rawHf, ok := r.Get(name)
|
|
if !ok {
|
|
logger.Error(logPrefix, "No plugin registered as", name)
|
|
continue
|
|
}
|
|
|
|
hf, ok := rawHf.(func(context.Context, map[string]any, http.Handler) (http.Handler, error))
|
|
if !ok {
|
|
logger.Error(logPrefix, "Wrong plugin handler type:", name)
|
|
continue
|
|
}
|
|
|
|
handlerWrapper, err := hf(ctx, extra, handler)
|
|
if err != nil {
|
|
logger.Error(logPrefix, "Error getting the plugin handler:", err.Error())
|
|
continue
|
|
}
|
|
|
|
logger.Info(logPrefix, "Injecting plugin", name)
|
|
handler = handlerWrapper
|
|
}
|
|
return next(ctx, cfg, handler)
|
|
}
|
|
}
|