Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1476)

Unified Diff: appengine/cmd/dm/model/keys.go

Issue 1537883002: Initial distributor implementation (Closed) Base URL: https://chromium.googlesource.com/external/github.com/luci/luci-go@master
Patch Set: self review Created 4 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: appengine/cmd/dm/model/keys.go
diff --git a/appengine/cmd/dm/model/keys.go b/appengine/cmd/dm/model/keys.go
new file mode 100644
index 0000000000000000000000000000000000000000..8a7e2cb6653f655d6e2f39413adaf4fa0cca7f5f
--- /dev/null
+++ b/appengine/cmd/dm/model/keys.go
@@ -0,0 +1,86 @@
+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package model
+
+import (
+ "fmt"
+ "math"
+
+ "golang.org/x/net/context"
+
+ "github.com/luci/gae/service/datastore"
+ dm "github.com/luci/luci-go/common/api/dm/service/v1"
+)
+
+// QuestKeyFromID makes a datastore.Key given the QuestID.
+func QuestKeyFromID(c context.Context, qid string) *datastore.Key {
+ return datastore.Get(c).MakeKey("Quest", qid)
+}
+
+// QuestFromID produces an empty Quest model from the QuestID.
+func QuestFromID(qid string) *Quest {
+ return &Quest{ID: qid}
+}
+
+// QuestIDFromKey makes a QuestID from the given datastore.Key. It panics if the
+// Key does not point to a Quest.
+func QuestIDFromKey(k *datastore.Key) string {
+ if k.Kind() != "Quest" || k.Parent() != nil {
+ panic(fmt.Errorf("invalid Quest key: %s", k))
+ }
+ return k.StringID()
+}
+
+// AttemptKeyFromID makes a datastore.Key given the AttemptID.
+func AttemptKeyFromID(c context.Context, aid *dm.Attempt_ID) *datastore.Key {
+ return datastore.Get(c).MakeKey("Attempt", aid.DMEncoded())
+}
+
+// AttemptFromID produces an empty Attempt model from the AttemptID.
+func AttemptFromID(aid *dm.Attempt_ID) *Attempt {
+ ret := &Attempt{}
+ ret.ID = *aid
+ return ret
+}
+
+// AttemptIDFromKey makes a AttemptID from the given datastore.Key. It panics if the
+// Key does not point to a Attempt.
+func AttemptIDFromKey(k *datastore.Key) *dm.Attempt_ID {
+ if k.Kind() != "Attempt" || k.Parent() != nil {
+ panic(fmt.Errorf("invalid Attempt key: %s", k))
+ }
+ ret := &dm.Attempt_ID{}
+ if err := ret.SetDMEncoded(k.StringID()); err != nil {
+ panic(fmt.Errorf("invalid Attempt key: %s: %s", k, err))
+ }
+ return ret
+}
+
+// ExecutionKeyFromID makes a datastore.Key given the ExecutionID.
+func ExecutionKeyFromID(c context.Context, eid *dm.Execution_ID) *datastore.Key {
+ return datastore.Get(c).MakeKey("Attempt", eid.AttemptID().DMEncoded(), "Execution", eid.Id)
+}
+
+// ExecutionFromID produces an empty Execution model from the ExecutionID.
+func ExecutionFromID(c context.Context, eid *dm.Execution_ID) *Execution {
+ ret := &Execution{}
+ ret.ID = eid.Id
+ ret.Attempt = AttemptKeyFromID(c, eid.AttemptID())
+ return ret
+}
+
+// ExecutionIDFromKey makes a ExecutionID from the given datastore.Key. It panics if the
+// Key does not point to a Execution.
+func ExecutionIDFromKey(k *datastore.Key) *dm.Execution_ID {
+ if k.Kind() != "Execution" || k.Parent() == nil {
+ panic(fmt.Errorf("invalid Execution key: %s", k))
+ }
+ id := k.IntID()
+ if id <= 0 || id > math.MaxUint32 {
+ panic(fmt.Errorf("invalid Execution key: %s", k))
+ }
+ atmpt := AttemptIDFromKey(k.Parent())
+ return &dm.Execution_ID{Quest: atmpt.Quest, Attempt: atmpt.Id, Id: uint32(id)}
+}

Powered by Google App Engine
This is Rietveld 408576698