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 "fmt" | 9 "fmt" |
10 | 10 |
11 "golang.org/x/net/context" | 11 "golang.org/x/net/context" |
12 | 12 |
13 ds "github.com/luci/gae/service/datastore" | 13 ds "github.com/luci/gae/service/datastore" |
14 "github.com/luci/gae/service/info" | |
14 ) | 15 ) |
15 | 16 |
16 //////////////////////////////////// public //////////////////////////////////// | 17 //////////////////////////////////// public //////////////////////////////////// |
17 | 18 |
18 // useRDS adds a gae.Datastore implementation to context, accessible | 19 // useRDS adds a gae.Datastore implementation to context, accessible |
19 // by gae.GetDS(c) | 20 // by gae.GetDS(c) |
20 func useRDS(c context.Context) context.Context { | 21 func useRDS(c context.Context) context.Context { |
21 return ds.SetRawFactory(c, func(ic context.Context) ds.RawInterface { | 22 return ds.SetRawFactory(c, func(ic context.Context) ds.RawInterface { |
22 dsd := cur(ic).Get(memContextDSIdx) | 23 dsd := cur(ic).Get(memContextDSIdx) |
23 | 24 |
24 ns := curGID(ic).namespace | 25 ns := curGID(ic).namespace |
25 if x, ok := dsd.(*dataStoreData); ok { | 26 if x, ok := dsd.(*dataStoreData); ok { |
26 return &dsImpl{x, ns, ic} | 27 return &dsImpl{x, ns, ic} |
27 } | 28 } |
28 return &txnDsImpl{dsd.(*txnDataStoreData), ns} | 29 return &txnDsImpl{dsd.(*txnDataStoreData), ns} |
29 }) | 30 }) |
30 } | 31 } |
31 | 32 |
33 // NewDatastore creates a new standalone memory implementation of the datastore, | |
34 // suitable for embedding for doing in-memory data organization. | |
35 // | |
36 // It's configured by default with the following settings: | |
37 // * AutoIndex(true) | |
38 // * Consistent(true) | |
39 // * DisableSpecialEntities(true) | |
40 // | |
41 // These settings can of course be changed by using the Testable() interface. | |
42 func NewDatastore(aid, ns string) (ds.Interface, error) { | |
43 ctx := UseWithAppID(context.Background(), aid) | |
44 ctx, err := info.Get(ctx).Namespace(ns) | |
45 if err != nil { | |
46 return nil, err | |
47 } | |
48 ret := ds.Get(ctx) | |
49 t := ret.Testable() | |
Vadim Sh.
2015/09/29 20:08:36
name 'Testable' is unfortunate, considering it is
iannucci
2015/09/29 20:09:45
Yeah that occurred to me. AlterSemantics?
Vadim Sh.
2015/09/29 20:12:03
"Tweaks"? "Hacks"? :)
| |
50 t.AutoIndex(true) | |
51 t.Consistent(true) | |
52 t.DisableSpecialEntities(true) | |
53 return ret, nil | |
54 } | |
55 | |
32 //////////////////////////////////// dsImpl //////////////////////////////////// | 56 //////////////////////////////////// dsImpl //////////////////////////////////// |
33 | 57 |
34 // dsImpl exists solely to bind the current c to the datastore data. | 58 // dsImpl exists solely to bind the current c to the datastore data. |
35 type dsImpl struct { | 59 type dsImpl struct { |
36 data *dataStoreData | 60 data *dataStoreData |
37 ns string | 61 ns string |
38 c context.Context | 62 c context.Context |
39 } | 63 } |
40 | 64 |
41 var _ ds.RawInterface = (*dsImpl)(nil) | 65 var _ ds.RawInterface = (*dsImpl)(nil) |
(...skipping 139 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
181 return countQuery(fq, d.data.parent.aid, d.ns, true, d.data.snap, d.data .snap) | 205 return countQuery(fq, d.data.parent.aid, d.ns, true, d.data.snap, d.data .snap) |
182 } | 206 } |
183 | 207 |
184 func (*txnDsImpl) RunInTransaction(func(c context.Context) error, *ds.Transactio nOptions) error { | 208 func (*txnDsImpl) RunInTransaction(func(c context.Context) error, *ds.Transactio nOptions) error { |
185 return errors.New("datastore: nested transactions are not supported") | 209 return errors.New("datastore: nested transactions are not supported") |
186 } | 210 } |
187 | 211 |
188 func (*txnDsImpl) Testable() ds.Testable { | 212 func (*txnDsImpl) Testable() ds.Testable { |
189 return nil | 213 return nil |
190 } | 214 } |
OLD | NEW |