From a054f480adf26e90ddcb66408688486704ab7094 Mon Sep 17 00:00:00 2001 From: Alexander Neonxp Kiryukhin Date: Tue, 17 Sep 2024 01:19:25 +0300 Subject: simple handler --- ctxlib/context.go | 44 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 ctxlib/context.go (limited to 'ctxlib') diff --git a/ctxlib/context.go b/ctxlib/context.go new file mode 100644 index 0000000..79851ce --- /dev/null +++ b/ctxlib/context.go @@ -0,0 +1,44 @@ +package ctxlib + +import ( + "context" + "net/http" +) + +type ctxKey int + +const ( + Method ctxKey = iota + Headers + Request + RequestID + Response +) + +func ResponseFromContext(ctx context.Context) http.ResponseWriter { + return ctx.Value(Response).(http.ResponseWriter) +} + +func RequestFromContext(ctx context.Context) *http.Request { + c := ctx.Value(Request) + if c == nil { + return &http.Request{} + } + return c.(*http.Request) +} + +func HeadersFromContext(ctx context.Context) http.Header { + c := ctx.Value(Headers) + if c == nil { + return http.Header{} + } + return c.(http.Header) +} + +func MethodFromContext(ctx context.Context) string { + c := ctx.Value(Method) + if c == nil { + return http.MethodGet + } + return c.(string) +} -- cgit v1.2.3