OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 package helper | |
6 | |
7 import ( | |
8 "encoding/json" | |
9 "infra/gae/libs/gae" | |
10 ) | |
11 | |
12 // GenericDSKey is an implementation of gae.DSKey which doesn't rely on the | |
13 // SDK's implementation. It differs slightly in that it's not recursive (and | |
14 // thus cannot express some of the invalid Key's that the SDK implementation | |
15 // can). In particular, it's not possible to have a GenericDSKey in a namespace | |
16 // whose Parent() is in a different namespace. | |
17 // | |
18 // GenericDSKey also implements json.Marshaler and json.Unmarshaler so it's | |
19 // suitable for use in structs which need to serialize both to json and to | |
20 // datastore. | |
21 type GenericDSKey struct { | |
22 appID string | |
23 namespace string | |
24 toks []gae.DSKeyTok | |
25 } | |
26 | |
27 var _ interface { | |
28 gae.DSKey | |
29 json.Marshaler | |
30 json.Unmarshaler | |
31 } = (*GenericDSKey)(nil) | |
32 | |
33 // NewDSKeyToks creates a new GenericDSKey. It is the DSKey implementation | |
34 // returned from the various DSPropertyMap serialization routines, as well as | |
35 // the native key implementation for the in-memory implementation of gae. | |
36 func NewDSKeyToks(appID, ns string, toks []gae.DSKeyTok) *GenericDSKey { | |
37 newToks := make([]gae.DSKeyTok, len(toks)) | |
38 copy(newToks, toks) | |
39 return &GenericDSKey{appID, ns, newToks} | |
40 } | |
41 | |
42 // NewDSKey is a wrapper around NewDSKeyToks which has an interface similar | |
43 // to NewKey in the SDK. | |
44 func NewDSKey(appID, ns, kind, stringID string, intID int64, parent gae.DSKey) *
GenericDSKey { | |
45 _, _, toks := DSKeySplit(parent) | |
46 newToks := make([]gae.DSKeyTok, len(toks)) | |
47 copy(newToks, toks) | |
48 newToks = append(newToks, gae.DSKeyTok{Kind: kind, StringID: stringID, I
ntID: intID}) | |
49 return &GenericDSKey{appID, ns, newToks} | |
50 } | |
51 | |
52 func (k *GenericDSKey) lastTok() (ret gae.DSKeyTok) { | |
53 if len(k.toks) > 0 { | |
54 ret = k.toks[len(k.toks)-1] | |
55 } | |
56 return | |
57 } | |
58 | |
59 func (k *GenericDSKey) AppID() string { return k.appID } | |
60 func (k *GenericDSKey) Namespace() string { return k.namespace } | |
61 func (k *GenericDSKey) Kind() string { return k.lastTok().Kind } | |
62 func (k *GenericDSKey) StringID() string { return k.lastTok().StringID } | |
63 func (k *GenericDSKey) IntID() int64 { return k.lastTok().IntID } | |
64 func (k *GenericDSKey) String() string { return DSKeyString(k) } | |
65 | |
66 func (k *GenericDSKey) MarshalJSON() ([]byte, error) { | |
67 return DSKeyMarshalJSON(k) | |
68 } | |
69 | |
70 func (k *GenericDSKey) UnmarshalJSON(buf []byte) error { | |
71 appID, namespace, toks, err := DSKeyUnmarshalJSON(buf) | |
72 if err != nil { | |
73 return err | |
74 } | |
75 *k = *NewDSKeyToks(appID, namespace, toks) | |
76 return nil | |
77 } | |
78 | |
79 func (k *GenericDSKey) Parent() gae.DSKey { | |
80 if len(k.toks) <= 1 { | |
81 return nil | |
82 } | |
83 return &GenericDSKey{k.appID, k.namespace, k.toks[:len(k.toks)-1]} | |
84 } | |
OLD | NEW |