OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 package memory | 5 package memory |
6 | 6 |
7 import ( | 7 import ( |
8 "infra/gae/libs/wrapper" | 8 "infra/gae/libs/wrapper" |
9 | 9 |
10 "golang.org/x/net/context" | 10 "golang.org/x/net/context" |
11 ) | 11 ) |
12 | 12 |
13 type giContextKeyType int | 13 type giContextKeyType int |
14 | 14 |
15 var giContextKey giContextKeyType | 15 var giContextKey giContextKeyType |
16 | 16 |
17 func curGID(c context.Context) *globalInfoData { | 17 func curGID(c context.Context) *globalInfoData { |
18 return c.Value(giContextKey).(*globalInfoData) | 18 return c.Value(giContextKey).(*globalInfoData) |
19 } | 19 } |
20 | 20 |
21 // UseGI adds a wrapper.GlobalInfo context, accessible | 21 // useGI adds a wrapper.GlobalInfo context, accessible |
22 // by wrapper.GetGI(c) | 22 // by wrapper.GetGI(c) |
23 func UseGI(c context.Context) context.Context { | 23 func useGI(c context.Context) context.Context { |
24 return wrapper.SetGIFactory(c, func(ic context.Context) wrapper.GlobalIn
fo { | 24 return wrapper.SetGIFactory(c, func(ic context.Context) wrapper.GlobalIn
fo { |
25 return &giImpl{wrapper.DummyGI(), curGID(ic), ic} | 25 return &giImpl{wrapper.DummyGI(), curGID(ic), ic} |
26 }) | 26 }) |
27 } | 27 } |
28 | 28 |
29 type globalInfoData struct{ namespace string } | 29 type globalInfoData struct{ namespace string } |
30 | 30 |
31 type giImpl struct { | 31 type giImpl struct { |
32 wrapper.GlobalInfo | 32 wrapper.GlobalInfo |
33 data *globalInfoData | 33 data *globalInfoData |
34 c context.Context | 34 c context.Context |
35 } | 35 } |
36 | 36 |
37 var _ = wrapper.GlobalInfo((*giImpl)(nil)) | 37 var _ = wrapper.GlobalInfo((*giImpl)(nil)) |
38 | 38 |
39 func (gi *giImpl) Namespace(ns string) (context.Context, error) { | 39 func (gi *giImpl) Namespace(ns string) (context.Context, error) { |
40 return context.WithValue(gi.c, giContextKey, &globalInfoData{ns}), nil | 40 return context.WithValue(gi.c, giContextKey, &globalInfoData{ns}), nil |
41 } | 41 } |
OLD | NEW |