OLD | NEW |
1 // Copyright 2015 The LUCI Authors. All rights reserved. | 1 // Copyright 2015 The LUCI Authors. All rights reserved. |
2 // Use of this source code is governed under the Apache License, Version 2.0 | 2 // Use of this source code is governed under the Apache License, Version 2.0 |
3 // that can be found in the LICENSE file. | 3 // that can be found in the LICENSE file. |
4 | 4 |
5 package memory | 5 package memory |
6 | 6 |
7 import ( | 7 import ( |
8 "fmt" | 8 "fmt" |
9 "regexp" | 9 "regexp" |
10 | 10 |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
56 } | 56 } |
57 | 57 |
58 clone := *cur | 58 clone := *cur |
59 f(&clone) | 59 f(&clone) |
60 return context.WithValue(c, giContextKey, &clone) | 60 return context.WithValue(c, giContextKey, &clone) |
61 } | 61 } |
62 | 62 |
63 // useGI adds a gae.GlobalInfo context, accessible | 63 // useGI adds a gae.GlobalInfo context, accessible |
64 // by gae.GetGI(c) | 64 // by gae.GetGI(c) |
65 func useGI(c context.Context) context.Context { | 65 func useGI(c context.Context) context.Context { |
66 » return info.SetFactory(c, func(ic context.Context) info.Interface { | 66 » return info.SetFactory(c, func(ic context.Context) info.RawInterface { |
67 return &giImpl{dummy.Info(), curGID(ic), ic} | 67 return &giImpl{dummy.Info(), curGID(ic), ic} |
68 }) | 68 }) |
69 } | 69 } |
70 | 70 |
71 type giImpl struct { | 71 type giImpl struct { |
72 » info.Interface | 72 » info.RawInterface |
73 *globalInfoData | 73 *globalInfoData |
74 c context.Context | 74 c context.Context |
75 } | 75 } |
76 | 76 |
77 var _ = info.Interface((*giImpl)(nil)) | 77 var _ = info.RawInterface((*giImpl)(nil)) |
78 | 78 |
79 func (gi *giImpl) GetNamespace() (string, bool) { | 79 func (gi *giImpl) GetNamespace() (string, bool) { |
80 return gi.getNamespace() | 80 return gi.getNamespace() |
81 } | 81 } |
82 | 82 |
83 func (gi *giImpl) Namespace(ns string) (ret context.Context, err error) { | 83 func (gi *giImpl) Namespace(ns string) (ret context.Context, err error) { |
84 if !validNamespace.MatchString(ns) { | 84 if !validNamespace.MatchString(ns) { |
85 return nil, fmt.Errorf("appengine: namespace %q does not match /
%s/", ns, validNamespace) | 85 return nil, fmt.Errorf("appengine: namespace %q does not match /
%s/", ns, validNamespace) |
86 } | 86 } |
87 | 87 |
88 return useGID(gi.c, func(mod *globalInfoData) { | 88 return useGID(gi.c, func(mod *globalInfoData) { |
89 mod.namespace = &ns | 89 mod.namespace = &ns |
90 }), nil | 90 }), nil |
91 } | 91 } |
92 | 92 |
93 func (gi *giImpl) MustNamespace(ns string) context.Context { | |
94 ret, err := gi.Namespace(ns) | |
95 if err != nil { | |
96 panic(err) | |
97 } | |
98 return ret | |
99 } | |
100 | |
101 func (gi *giImpl) AppID() string { | 93 func (gi *giImpl) AppID() string { |
102 return gi.appID | 94 return gi.appID |
103 } | 95 } |
104 | 96 |
105 func (gi *giImpl) FullyQualifiedAppID() string { | 97 func (gi *giImpl) FullyQualifiedAppID() string { |
106 return gi.fqAppID | 98 return gi.fqAppID |
107 } | 99 } |
108 | 100 |
109 func (gi *giImpl) IsDevAppServer() bool { | 101 func (gi *giImpl) IsDevAppServer() bool { |
110 return true | 102 return true |
(...skipping 15 matching lines...) Expand all Loading... |
126 return useGID(gi.c, func(mod *globalInfoData) { | 118 return useGID(gi.c, func(mod *globalInfoData) { |
127 mod.versionID = v | 119 mod.versionID = v |
128 }) | 120 }) |
129 } | 121 } |
130 | 122 |
131 func (gi *giImpl) SetRequestID(v string) context.Context { | 123 func (gi *giImpl) SetRequestID(v string) context.Context { |
132 return useGID(gi.c, func(mod *globalInfoData) { | 124 return useGID(gi.c, func(mod *globalInfoData) { |
133 mod.requestID = v | 125 mod.requestID = v |
134 }) | 126 }) |
135 } | 127 } |
OLD | NEW |