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 meta | 5 package meta |
6 | 6 |
7 import ( | 7 import ( |
8 "golang.org/x/net/context" | 8 "golang.org/x/net/context" |
9 | 9 |
10 "appengine/datastore" | |
11 | |
12 "github.com/luci/luci-go/common/errors" | 10 "github.com/luci/luci-go/common/errors" |
13 | 11 |
14 » "infra/gae/libs/wrapper" | 12 » "infra/gae/libs/gae" |
15 ) | 13 ) |
16 | 14 |
17 var mark = errors.MakeMarkFn("eg") | 15 var mark = errors.MakeMarkFn("eg") |
18 | 16 |
19 // EntityGroupMeta is the model corresponding to the __entity_group__ model in | 17 // EntityGroupMeta is the model corresponding to the __entity_group__ model in |
20 // appengine. You shouldn't need to use this struct directly, but instead should | 18 // appengine. You shouldn't need to use this struct directly, but instead should |
21 // use GetEntityGroupVersion. | 19 // use GetEntityGroupVersion. |
22 type EntityGroupMeta struct { | 20 type EntityGroupMeta struct { |
23 » _kind string `datastore:"-" goon:"kind,__entity_group__"` | 21 » Version int64 `gae:"__version__"` |
24 | |
25 » ID int64 `datastore:"-" goon:"id"` | |
26 » Parent *datastore.Key `datastore:"-" goon:"parent"` | |
27 | |
28 » Version int64 `datastore:"__version__"` | |
29 } | 22 } |
30 | 23 |
31 // GetEntityGroupVersion returns the entity group version for the entity group | 24 // GetEntityGroupVersion returns the entity group version for the entity group |
32 // containing root. If the entity group doesn't exist, this function will return | 25 // containing root. If the entity group doesn't exist, this function will return |
33 // zero and a nil error. | 26 // zero and a nil error. |
34 func GetEntityGroupVersion(c context.Context, root *datastore.Key) (int64, error
) { | 27 func GetEntityGroupVersion(c context.Context, root gae.DSKey) (int64, error) { |
35 for root.Parent() != nil { | 28 for root.Parent() != nil { |
36 root = root.Parent() | 29 root = root.Parent() |
37 } | 30 } |
38 » egm := &EntityGroupMeta{ID: 1, Parent: root} | 31 » egm := &EntityGroupMeta{} |
39 » err := wrapper.GetDS(c).Get(egm) | 32 » rds := gae.GetRDS(c) |
40 » if err != datastore.ErrNoSuchEntity { | 33 » err := rds.Get(rds.NewKey("__entity_group__", "", 1, root), egm) |
| 34 » if err != gae.ErrDSNoSuchEntity { |
41 err = mark(err) | 35 err = mark(err) |
42 } else { | 36 } else { |
43 // this is OK for callers. The version of the entity group is ef
fectively 0 | 37 // this is OK for callers. The version of the entity group is ef
fectively 0 |
44 // in this case. | 38 // in this case. |
45 err = nil | 39 err = nil |
46 } | 40 } |
47 return egm.Version, err | 41 return egm.Version, err |
48 } | 42 } |
OLD | NEW |