mirror of
https://github.com/hanzoai/gateway.git
synced 2026-11-02 08:45:10 +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
2.2 KiB
Go
81 lines
2.2 KiB
Go
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package plugin
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
|
|
"github.com/hanzoai/gateway/v2/internal/lura/config"
|
|
"github.com/hanzoai/gateway/v2/internal/lura/logging"
|
|
"github.com/hanzoai/gateway/v2/internal/lura/transport/http/client"
|
|
)
|
|
|
|
const Namespace = "github.com/devopsfaith/krakend/transport/http/client/executor"
|
|
|
|
func HTTPRequestExecutor(
|
|
logger logging.Logger,
|
|
next func(*config.Backend) client.HTTPRequestExecutor,
|
|
) func(*config.Backend) client.HTTPRequestExecutor {
|
|
return HTTPRequestExecutorWithContext(context.Background(), logger, next)
|
|
}
|
|
|
|
func HTTPRequestExecutorWithContext(
|
|
ctx context.Context,
|
|
logger logging.Logger,
|
|
next func(*config.Backend) client.HTTPRequestExecutor,
|
|
) func(*config.Backend) client.HTTPRequestExecutor {
|
|
return func(cfg *config.Backend) client.HTTPRequestExecutor {
|
|
logPrefix := fmt.Sprintf("[BACKEND: %s %s -> %s]", cfg.ParentEndpointMethod, cfg.ParentEndpoint, cfg.URLPattern)
|
|
v, ok := cfg.ExtraConfig[Namespace]
|
|
if !ok {
|
|
return next(cfg)
|
|
}
|
|
extra, ok := v.(map[string]any)
|
|
if !ok {
|
|
logger.Debug(logPrefix, "["+Namespace+"]", "Wrong extra config type for backend")
|
|
return next(cfg)
|
|
}
|
|
|
|
// load plugin
|
|
r, ok := clientRegister.Get(Namespace)
|
|
if !ok {
|
|
logger.Debug(logPrefix, "No plugins registered for the module")
|
|
return next(cfg)
|
|
}
|
|
|
|
name, ok := extra["name"].(string)
|
|
if !ok {
|
|
logger.Debug(logPrefix, "No name defined in the extra config for", cfg.URLPattern)
|
|
return next(cfg)
|
|
}
|
|
|
|
rawHf, ok := r.Get(name)
|
|
if !ok {
|
|
logger.Debug(logPrefix, "No plugin registered as", name)
|
|
return next(cfg)
|
|
}
|
|
|
|
hf, ok := rawHf.(func(context.Context, map[string]any) (http.Handler, error))
|
|
if !ok {
|
|
logger.Warning(logPrefix, "Wrong plugin handler type:", name)
|
|
return next(cfg)
|
|
}
|
|
|
|
handler, err := hf(ctx, extra)
|
|
if err != nil {
|
|
logger.Warning(logPrefix, "Error getting the plugin handler:", err.Error())
|
|
return next(cfg)
|
|
}
|
|
|
|
logger.Debug(logPrefix, "Injecting plugin", name)
|
|
return func(ctx context.Context, req *http.Request) (*http.Response, error) {
|
|
w := httptest.NewRecorder()
|
|
handler.ServeHTTP(w, req.WithContext(ctx))
|
|
return w.Result(), nil
|
|
}
|
|
}
|
|
}
|