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 prod | 5 package prod |
6 | 6 |
7 import ( | 7 import ( |
8 ds "github.com/luci/gae/service/datastore" | 8 ds "github.com/luci/gae/service/datastore" |
| 9 "github.com/luci/gae/service/datastore/dskey" |
9 "google.golang.org/appengine/datastore" | 10 "google.golang.org/appengine/datastore" |
10 ) | 11 ) |
11 | 12 |
12 type dsKeyImpl struct { | 13 type dsKeyImpl struct { |
13 *datastore.Key | 14 *datastore.Key |
14 } | 15 } |
15 | 16 |
16 var _ ds.Key = dsKeyImpl{} | 17 var _ ds.Key = dsKeyImpl{} |
17 | 18 |
18 func (k dsKeyImpl) Parent() ds.Key { return dsR2F(k.Key.Parent()) } | 19 func (k dsKeyImpl) Parent() ds.Key { return dsR2F(k.Key.Parent()) } |
| 20 func (k dsKeyImpl) Incomplete() bool { return k.Key.Incomplete() } |
| 21 func (k dsKeyImpl) Valid(allowSpecial bool, aid, ns string) bool { |
| 22 » return dskey.Valid(k, allowSpecial, aid, ns) |
| 23 } |
| 24 func (k dsKeyImpl) PartialValid(aid, ns string) bool { |
| 25 » return dskey.PartialValid(k, aid, ns) |
| 26 } |
19 | 27 |
20 // dsR2F (DS real-to-fake) converts an SDK Key to a ds.Key | 28 // dsR2F (DS real-to-fake) converts an SDK Key to a ds.Key |
21 func dsR2F(k *datastore.Key) ds.Key { | 29 func dsR2F(k *datastore.Key) ds.Key { |
22 return dsKeyImpl{k} | 30 return dsKeyImpl{k} |
23 } | 31 } |
24 | 32 |
25 // dsF2R (DS fake-to-real) converts a DSKey back to an SDK *Key. | 33 // dsF2R (DS fake-to-real) converts a DSKey back to an SDK *Key. |
26 func dsF2R(k ds.Key) *datastore.Key { | 34 func dsF2R(k ds.Key) *datastore.Key { |
27 if rkey, ok := k.(dsKeyImpl); ok { | 35 if rkey, ok := k.(dsKeyImpl); ok { |
28 return rkey.Key | 36 return rkey.Key |
29 } | 37 } |
30 // we should always hit the fast case above, but just in case, safely ro
und | 38 // we should always hit the fast case above, but just in case, safely ro
und |
31 // trip through the proto encoding. | 39 // trip through the proto encoding. |
32 » rkey, err := datastore.DecodeKey(ds.KeyEncode(k)) | 40 » rkey, err := datastore.DecodeKey(dskey.Encode(k)) |
33 if err != nil { | 41 if err != nil { |
34 // should never happen in a good program, but it's not ignorable
, and | 42 // should never happen in a good program, but it's not ignorable
, and |
35 // passing an error back makes this function too cumbersome (and
it causes | 43 // passing an error back makes this function too cumbersome (and
it causes |
36 // this `if err != nil { panic(err) }` logic to show up in a bun
ch of | 44 // this `if err != nil { panic(err) }` logic to show up in a bun
ch of |
37 // places. Realistically, everything should hit the early exit c
lause above. | 45 // places. Realistically, everything should hit the early exit c
lause above. |
38 panic(err) | 46 panic(err) |
39 } | 47 } |
40 return rkey | 48 return rkey |
41 } | 49 } |
42 | 50 |
43 // dsMF2R (DS multi-fake-to-fake) converts a slice of wrapped keys to SDK keys. | 51 // dsMF2R (DS multi-fake-to-fake) converts a slice of wrapped keys to SDK keys. |
44 func dsMF2R(ks []ds.Key) []*datastore.Key { | 52 func dsMF2R(ks []ds.Key) []*datastore.Key { |
45 ret := make([]*datastore.Key, len(ks)) | 53 ret := make([]*datastore.Key, len(ks)) |
46 for i, k := range ks { | 54 for i, k := range ks { |
47 ret[i] = dsF2R(k) | 55 ret[i] = dsF2R(k) |
48 } | 56 } |
49 return ret | 57 return ret |
50 } | 58 } |
OLD | NEW |