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