refactor/idiomatic-handlers-middlewares #25
|
@ -3,14 +3,8 @@ package api
|
|||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// custom context key type
|
||||
type ctxKey int
|
||||
|
||||
const (
|
||||
ctxRequest ctxKey = iota
|
||||
ctxAuth
|
||||
"git.xdrm.io/go/aicra/internal/ctx"
|
||||
)
|
||||
|
||||
// Context is a simple wrapper around context.Context that adds helper methods
|
||||
|
@ -20,7 +14,7 @@ type Context struct{ context.Context }
|
|||
// Request current request
|
||||
func (c Context) Request() *http.Request {
|
||||
var (
|
||||
raw = c.Value(ctxRequest)
|
||||
raw = c.Value(ctx.Request)
|
||||
cast, ok = raw.(*http.Request)
|
||||
)
|
||||
if !ok {
|
||||
|
@ -29,10 +23,22 @@ func (c Context) Request() *http.Request {
|
|||
return cast
|
||||
}
|
||||
|
||||
// ResponseWriter for this request
|
||||
func (c Context) ResponseWriter() http.ResponseWriter {
|
||||
var (
|
||||
raw = c.Value(ctx.Response)
|
||||
cast, ok = raw.(http.ResponseWriter)
|
||||
)
|
||||
if !ok {
|
||||
return nil
|
||||
}
|
||||
return cast
|
||||
}
|
||||
|
||||
// Auth associated with this request
|
||||
func (c Context) Auth() *Auth {
|
||||
var (
|
||||
raw = c.Value(ctxAuth)
|
||||
raw = c.Value(ctx.Auth)
|
||||
cast, ok = raw.(*Auth)
|
||||
)
|
||||
if !ok {
|
||||
|
|
Loading…
Reference in New Issue