Initial distributor implementation

appengine/cmd/dm/model/quest.go

 // Copyright 2016 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 model
 
 import (
-        "crypto/sha256"
-        "encoding/base64"
-        "fmt"
         "time"
 
-        "github.com/golang/protobuf/proto"
         "golang.org/x/net/context"
 
         "github.com/luci/gae/service/datastore"
         "github.com/luci/luci-go/common/clock"
         google_pb "github.com/luci/luci-go/common/proto/google"
 
-        "github.com/luci/luci-go/common/api/dm/service/v1"
-        "github.com/luci/luci-go/common/api/template"
-)
-
-var (
-        // QuestIDLength is the number of encoded bytes to use. It removes the
-        // single padding character.
-        QuestIDLength = base64.URLEncoding.EncodedLen(sha256.Size) - 1
-)
-
-const (
-        // payloadMaxLength is the maximum size of the Quest Desc
-        payloadMaxLength = 256 * 1024
+        dm "github.com/luci/luci-go/common/api/dm/service/v1"
 )
 
 // NewQuest builds a new Quest object with a correct ID given the current
 // contents of the Quest_Desc. It returns an error if the Desc is invalid.
 //
-// This will also compactify the inner json Desc as a side effect.
-func NewQuest(c context.Context, desc *dm.Quest_Desc) (ret *Quest, err error) {
-        if len(desc.JsonPayload) > payloadMaxLength {
-                return nil, fmt.Errorf("quest payload is too large: %d > %d",
-                        len(desc.JsonPayload), payloadMaxLength)
-        }
-        desc.JsonPayload, err = template.NormalizeJSON(desc.JsonPayload, true)
-        if err != nil {
-                return
-        }
-
-        data, err := proto.Marshal(desc)
-        if err != nil {
-                panic(err)
-        }
-        h := sha256.Sum256(data)
-
-        ret = &Quest{
-                ID:      base64.URLEncoding.EncodeToString(h[:])[:QuestIDLength],
-                Desc:    *desc,
-                Created: clock.Now(c).UTC(),
-        }
-        return
+// Desc must already be Normalize()'d
+func NewQuest(c context.Context, desc *dm.Quest_Desc) *Quest {
+        return &Quest{ID: desc.QuestID(), Desc: *desc, Created: clock.Now(c).UTC()}
 }
 
 // Quest is the model for a job-to-run. Its questPayload should fully
 // describe the job to be done.
 type Quest struct {
         // ID is the base64 sha256 of questPayload
         ID string `gae:"$id"`
 
         Desc    dm.Quest_Desc `gae:",noindex"`
         BuiltBy TemplateInfo  `gae:",noindex"`
@@ skipping 23 matching lines @@
         for i := range q.BuiltBy {
                 spec[i] = &q.BuiltBy[i]
         }
 
         return &dm.Quest_Data{
                 Created: google_pb.NewTimestamp(q.Created),
                 Desc:    &q.Desc,
                 BuiltBy: spec,
         }
 }