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