Initial distributor implementation

appengine/cmd/dm/mutate/ensure_quest_attempts.go

 // Copyright 2015 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package mutate
 
 import (
         "github.com/luci/gae/service/datastore"
         "github.com/luci/luci-go/appengine/cmd/dm/model"
         "github.com/luci/luci-go/appengine/tumble"
@@ skipping 13 matching lines @@
         Quest *model.Quest
         AIDs  []uint32
 
         // DoNotMergeQuest causes this mutation to not attempt to merge the BuiltBy of
         // Quest.
         DoNotMergeQuest bool
 }
 
 // Root implements tumble.Mutation.
 func (e *EnsureQuestAttempts) Root(c context.Context) *datastore.Key {
-        return datastore.Get(c).KeyForObj(&model.Quest{ID: e.Quest.ID})
+        return model.QuestKeyFromID(c, e.Quest.ID)
 }
 
 // RollForward implements tumble.Mutation.
 func (e *EnsureQuestAttempts) RollForward(c context.Context) (muts []tumble.Mutation, err error) {
         if !e.DoNotMergeQuest {
-                if _, err = (&MergeQuest{e.Quest}).RollForward(c); err != nil {
+                if _, err = (&MergeQuest{Quest: e.Quest}).RollForward(c); err != nil {
                         return
                 }
         }
 
         if len(e.AIDs) > 0 {
                 lim := len(e.AIDs)
                 if lim > MaxEnsureAttempts {
                         lim = MaxEnsureAttempts + 1
                 }
                 muts = make([]tumble.Mutation, 0, lim)
                 for i, aid := range e.AIDs {
                         if i > MaxEnsureAttempts {
                                 muts = append(muts, &EnsureQuestAttempts{e.Quest, e.AIDs[i:], true})
                                 break
                         }
                         muts = append(muts, &EnsureAttempt{dm.NewAttemptID(e.Quest.ID, aid)})
                 }
         }
 
         return
 }
 
 func init() {
         tumble.Register((*EnsureQuestAttempts)(nil))
 }