| 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 |
| 11 » rds "github.com/luci/gae/service/rawdatastore" | 11 » ds "github.com/luci/gae/service/datastore" |
| 12 "golang.org/x/net/context" | 12 "golang.org/x/net/context" |
| 13 ) | 13 ) |
| 14 | 14 |
| 15 type memContextObj interface { | 15 type memContextObj interface { |
| 16 sync.Locker | 16 sync.Locker |
| 17 canApplyTxn(m memContextObj) bool | 17 canApplyTxn(m memContextObj) bool |
| 18 applyTxn(c context.Context, m memContextObj) | 18 applyTxn(c context.Context, m memContextObj) |
| 19 | 19 |
| 20 endTxn() | 20 endTxn() |
| 21 » mkTxn(*rds.TransactionOptions) memContextObj | 21 » mkTxn(*ds.TransactionOptions) memContextObj |
| 22 } | 22 } |
| 23 | 23 |
| 24 type memContext []memContextObj | 24 type memContext []memContextObj |
| 25 | 25 |
| 26 var _ = memContextObj((memContext)(nil)) | 26 var _ = memContextObj((memContext)(nil)) |
| 27 | 27 |
| 28 func newMemContext() memContext { | 28 func newMemContext() memContext { |
| 29 return memContext{ | 29 return memContext{ |
| 30 newTaskQueueData(), | 30 newTaskQueueData(), |
| 31 newDataStoreData(), | 31 newDataStoreData(), |
| (...skipping 22 matching lines...) Expand all Loading... |
| 54 m[i].Unlock() | 54 m[i].Unlock() |
| 55 } | 55 } |
| 56 } | 56 } |
| 57 | 57 |
| 58 func (m memContext) endTxn() { | 58 func (m memContext) endTxn() { |
| 59 for _, itm := range m { | 59 for _, itm := range m { |
| 60 itm.endTxn() | 60 itm.endTxn() |
| 61 } | 61 } |
| 62 } | 62 } |
| 63 | 63 |
| 64 func (m memContext) mkTxn(o *rds.TransactionOptions) memContextObj { | 64 func (m memContext) mkTxn(o *ds.TransactionOptions) memContextObj { |
| 65 ret := make(memContext, len(m)) | 65 ret := make(memContext, len(m)) |
| 66 for i, itm := range m { | 66 for i, itm := range m { |
| 67 ret[i] = itm.mkTxn(o) | 67 ret[i] = itm.mkTxn(o) |
| 68 } | 68 } |
| 69 return ret | 69 return ret |
| 70 } | 70 } |
| 71 | 71 |
| 72 func (m memContext) canApplyTxn(txnCtxObj memContextObj) bool { | 72 func (m memContext) canApplyTxn(txnCtxObj memContextObj) bool { |
| 73 txnCtx := txnCtxObj.(memContext) | 73 txnCtx := txnCtxObj.(memContext) |
| 74 for i := range m { | 74 for i := range m { |
| (...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 125 // a couple ways, for example. | 125 // a couple ways, for example. |
| 126 // | 126 // |
| 127 // It really should have been appengine.Context.RunInTransaction(func(tc...)), | 127 // It really should have been appengine.Context.RunInTransaction(func(tc...)), |
| 128 // but because it's not, this method is on dsImpl instead to mirror the official | 128 // but because it's not, this method is on dsImpl instead to mirror the official |
| 129 // API. | 129 // API. |
| 130 // | 130 // |
| 131 // The fake implementation also differs from the real implementation because the | 131 // The fake implementation also differs from the real implementation because the |
| 132 // fake TaskQueue is NOT backed by the fake Datastore. This is done to make the | 132 // fake TaskQueue is NOT backed by the fake Datastore. This is done to make the |
| 133 // test-access API for TaskQueue better (instead of trying to reconstitute the | 133 // test-access API for TaskQueue better (instead of trying to reconstitute the |
| 134 // state of the task queue from a bunch of datastore accesses). | 134 // state of the task queue from a bunch of datastore accesses). |
| 135 func (d *dsImpl) RunInTransaction(f func(context.Context) error, o *rds.Transact
ionOptions) error { | 135 func (d *dsImpl) RunInTransaction(f func(context.Context) error, o *ds.Transacti
onOptions) error { |
| 136 curMC := cur(d.c) | 136 curMC := cur(d.c) |
| 137 | 137 |
| 138 txnMC := curMC.mkTxn(o) | 138 txnMC := curMC.mkTxn(o) |
| 139 | 139 |
| 140 defer func() { | 140 defer func() { |
| 141 txnMC.Lock() | 141 txnMC.Lock() |
| 142 defer txnMC.Unlock() | 142 defer txnMC.Unlock() |
| 143 | 143 |
| 144 txnMC.endTxn() | 144 txnMC.endTxn() |
| 145 }() | 145 }() |
| 146 | 146 |
| 147 if err := f(context.WithValue(d.c, memContextKey, txnMC)); err != nil { | 147 if err := f(context.WithValue(d.c, memContextKey, txnMC)); err != nil { |
| 148 return err | 148 return err |
| 149 } | 149 } |
| 150 | 150 |
| 151 txnMC.Lock() | 151 txnMC.Lock() |
| 152 defer txnMC.Unlock() | 152 defer txnMC.Unlock() |
| 153 | 153 |
| 154 if curMC.canApplyTxn(txnMC) { | 154 if curMC.canApplyTxn(txnMC) { |
| 155 curMC.applyTxn(d.c, txnMC) | 155 curMC.applyTxn(d.c, txnMC) |
| 156 } else { | 156 } else { |
| 157 » » return rds.ErrConcurrentTransaction | 157 » » return ds.ErrConcurrentTransaction |
| 158 } | 158 } |
| 159 return nil | 159 return nil |
| 160 } | 160 } |
| OLD | NEW |