| 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 |