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 "errors" | 8 "errors" |
9 "sync" | 9 "sync" |
10 | 10 |
(...skipping 79 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
90 } | 90 } |
91 | 91 |
92 // Use calls UseWithAppID with the appid of "dev~app" | 92 // Use calls UseWithAppID with the appid of "dev~app" |
93 func Use(c context.Context) context.Context { | 93 func Use(c context.Context) context.Context { |
94 return UseWithAppID(c, "dev~app") | 94 return UseWithAppID(c, "dev~app") |
95 } | 95 } |
96 | 96 |
97 // UseWithAppID adds implementations for the following gae services to the | 97 // UseWithAppID adds implementations for the following gae services to the |
98 // context: | 98 // context: |
99 // * github.com/luci/gae/service/datastore | 99 // * github.com/luci/gae/service/datastore |
| 100 // * github.com/luci/gae/service/info |
| 101 // * github.com/luci/gae/service/mail |
| 102 // * github.com/luci/gae/service/memcache |
100 // * github.com/luci/gae/service/taskqueue | 103 // * github.com/luci/gae/service/taskqueue |
101 // * github.com/luci/gae/service/memcache | |
102 // * github.com/luci/gae/service/info | |
103 // * github.com/luci/gae/service/user | 104 // * github.com/luci/gae/service/user |
104 // * github.com/luci/luci-go/common/logger (using memlogger) | 105 // * github.com/luci/luci-go/common/logger (using memlogger) |
105 // | 106 // |
106 // The application id wil be set to 'aid', and will not be modifiable in this | 107 // The application id wil be set to 'aid', and will not be modifiable in this |
107 // context. | 108 // context. |
108 // | 109 // |
109 // These can be retrieved with the gae.Get functions. | 110 // These can be retrieved with the gae.Get functions. |
110 // | 111 // |
111 // The implementations are all backed by an in-memory implementation, and start | 112 // The implementations are all backed by an in-memory implementation, and start |
112 // with an empty state. | 113 // with an empty state. |
113 // | 114 // |
114 // Using this more than once per context.Context will cause a panic. | 115 // Using this more than once per context.Context will cause a panic. |
115 func UseWithAppID(c context.Context, aid string) context.Context { | 116 func UseWithAppID(c context.Context, aid string) context.Context { |
116 if c.Value(memContextKey) != nil { | 117 if c.Value(memContextKey) != nil { |
117 panic(errors.New("memory.Use: called twice on the same Context")
) | 118 panic(errors.New("memory.Use: called twice on the same Context")
) |
118 } | 119 } |
119 c = memlogger.Use(c) | 120 c = memlogger.Use(c) |
120 | 121 |
121 memctx := newMemContext(aid) | 122 memctx := newMemContext(aid) |
122 c = context.WithValue(c, memContextKey, memctx) | 123 c = context.WithValue(c, memContextKey, memctx) |
123 c = context.WithValue(c, memContextNoTxnKey, memctx) | 124 c = context.WithValue(c, memContextNoTxnKey, memctx) |
124 c = context.WithValue(c, giContextKey, &globalInfoData{appid: aid}) | 125 c = context.WithValue(c, giContextKey, &globalInfoData{appid: aid}) |
125 » return useUser(useTQ(useRDS(useMC(useGI(c, aid))))) | 126 » return useMail(useUser(useTQ(useRDS(useMC(useGI(c, aid)))))) |
126 } | 127 } |
127 | 128 |
128 func cur(c context.Context) (p *memContext) { | 129 func cur(c context.Context) (p *memContext) { |
129 p, _ = c.Value(memContextKey).(*memContext) | 130 p, _ = c.Value(memContextKey).(*memContext) |
130 return | 131 return |
131 } | 132 } |
132 | 133 |
133 func curNoTxn(c context.Context) (p *memContext) { | 134 func curNoTxn(c context.Context) (p *memContext) { |
134 p, _ = c.Value(memContextNoTxnKey).(*memContext) | 135 p, _ = c.Value(memContextNoTxnKey).(*memContext) |
135 return | 136 return |
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
194 if o != nil && o.Attempts != 0 { | 195 if o != nil && o.Attempts != 0 { |
195 attempts = o.Attempts | 196 attempts = o.Attempts |
196 } | 197 } |
197 for attempt := 0; attempt < attempts; attempt++ { | 198 for attempt := 0; attempt < attempts; attempt++ { |
198 if err := loopBody(attempt >= d.data.txnFakeRetry); err != ds.Er
rConcurrentTransaction { | 199 if err := loopBody(attempt >= d.data.txnFakeRetry); err != ds.Er
rConcurrentTransaction { |
199 return err | 200 return err |
200 } | 201 } |
201 } | 202 } |
202 return ds.ErrConcurrentTransaction | 203 return ds.ErrConcurrentTransaction |
203 } | 204 } |
OLD | NEW |