Initial distributor implementation

appengine/cmd/dm/mutate/add_finished_deps.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 10 matching lines @@
         // here must be a subset of the quests mentioned in FinishedAttempts.
         MergeQuests []*model.Quest
 
         // FinishedAttempts are a list of attempts that we already know are in the
         // Finished state.
         FinishedAttempts *dm.AttemptList
 }
 
 // Root implements tumble.Mutation
 func (f *AddFinishedDeps) Root(c context.Context) *datastore.Key {
-        return datastore.Get(c).KeyForObj(&model.Attempt{ID: *f.Auth.Id.AttemptID()})
+        return model.AttemptKeyFromID(c, f.Auth.Id.AttemptID())
 }
 
 // RollForward implements tumble.Mutation
 func (f *AddFinishedDeps) RollForward(c context.Context) (muts []tumble.Mutation, err error) {
         atmpt, _, err := model.AuthenticateExecution(c, f.Auth)
         if err != nil {
                 return
         }
 
         fwdDeps, err := filterExisting(c, model.FwdDepsFromList(c, f.Auth.Id.AttemptID(), f.FinishedAttempts))
         if err != nil || len(fwdDeps) == 0 {
                 return
         }
 
         muts = make([]tumble.Mutation, 0, len(fwdDeps)+len(f.MergeQuests))
         for _, d := range fwdDeps {
                 d.ForExecution = atmpt.CurExecution
                 muts = append(muts, &AddBackDep{Dep: d.Edge()})
         }
         for _, q := range f.MergeQuests {
-                muts = append(muts, &MergeQuest{q})
+                muts = append(muts, &MergeQuest{Quest: q})
         }
 
         return muts, datastore.Get(c).PutMulti(fwdDeps)
 }
 
 func init() {
         tumble.Register((*AddFinishedDeps)(nil))
 }