56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
|
package http
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"forge.cadoles.com/arcad/edge/pkg/bus"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
type contextKey string
|
||
|
|
|
||
|
|
var (
|
||
|
|
contextKeyBus contextKey = "bus"
|
||
|
|
contextKeyHTTPRequest contextKey = "httpRequest"
|
||
|
|
contextKeyHTTPClient contextKey = "httpClient"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (h *Handler) contextMiddleware(next http.Handler) http.Handler {
|
||
|
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
ctx := r.Context()
|
||
|
|
|
||
|
|
ctx = context.WithValue(ctx, contextKeyBus, h.bus)
|
||
|
|
ctx = context.WithValue(ctx, contextKeyHTTPRequest, r)
|
||
|
|
ctx = context.WithValue(ctx, contextKeyHTTPClient, h.httpClient)
|
||
|
|
|
||
|
|
r = r.WithContext(ctx)
|
||
|
|
|
||
|
|
next.ServeHTTP(w, r)
|
||
|
|
}
|
||
|
|
|
||
|
|
return http.HandlerFunc(fn)
|
||
|
|
}
|
||
|
|
|
||
|
|
func ContextBus(ctx context.Context) bus.Bus {
|
||
|
|
return contextValue[bus.Bus](ctx, contextKeyBus)
|
||
|
|
}
|
||
|
|
|
||
|
|
func ContextHTTPRequest(ctx context.Context) *http.Request {
|
||
|
|
return contextValue[*http.Request](ctx, contextKeyHTTPRequest)
|
||
|
|
}
|
||
|
|
|
||
|
|
func ContextHTTPClient(ctx context.Context) *http.Client {
|
||
|
|
return contextValue[*http.Client](ctx, contextKeyHTTPClient)
|
||
|
|
}
|
||
|
|
|
||
|
|
func contextValue[T any](ctx context.Context, key any) T {
|
||
|
|
value, ok := ctx.Value(key).(T)
|
||
|
|
if !ok {
|
||
|
|
panic(errors.Errorf("could not find key '%v' on context", key))
|
||
|
|
}
|
||
|
|
|
||
|
|
return value
|
||
|
|
}
|