blob: 08267c33cafaacf83a641084c697d38369ff98f0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
// Package gorilla provides a go.net/context.Context implementation whose Value
// method returns the values associated with a specific HTTP request in the
// github.com/gorilla/context package.
package gorilla
import (
"net/http"
"code.google.com/p/go.net/context"
gcontext "github.com/gorilla/context"
)
// NewContext returns a Context whose Value method returns values associated
// with req using the Gorilla context package:
// http://www.gorillatoolkit.org/pkg/context
func NewContext(parent context.Context, req *http.Request) context.Context {
return &wrapper{parent, req}
}
type wrapper struct {
context.Context
req *http.Request
}
var reqKey struct{}
// Value returns Gorilla's context package's value for this Context's request
// and key. It delegates to the parent Context if there is no such value.
func (ctx *wrapper) Value(key interface{}) interface{} {
if key == &reqKey {
return ctx.req
}
if val, ok := gcontext.GetOk(ctx.req, key); ok {
return val
}
return ctx.Context.Value(key)
}
// HTTPRequest returns the *http.Request associated with ctx using NewContext,
// if any.
func HTTPRequest(ctx context.Context) (*http.Request, bool) {
// We cannot use ctx.(*wrapper).req to get the request because ctx may
// be a Context derived from a *wrapper. Instead, we use Value to
// access the request if it is anywhere up the Context tree.
req, ok := ctx.Value(&reqKey).(*http.Request)
return req, ok
}
|