Buildbucket: Save buildbucket build info and summary on pubsub push

milo/buildsource/buildbucket/struct.go

+// Copyright 2017 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 buildbucket
+
+import (
+        "encoding/json"
+        "fmt"
+        "time"
+
+        bucketApi "github.com/luci/luci-go/common/api/buildbucket/buildbucket/v1"
+        "github.com/luci/luci-go/milo/api/resp"
+)
+
+// buildEntry is a full buildbucket build along with its full resp rendering
+// at the time of modification.  This is a parent of a BuildSummary.
+type buildEntry struct {
+        // key is formulated via <project ID>:<build ID>.  From PubSub, project ID
+        // is determined via the topic name.
+        key string `gae:$id`
+
+        // buildData is the json marshalled form of
+        // a bucketApi.ApiCommonBuildMessage message.
+        buildbucketData []byte `gae:,noindex`
+
+        // respBuild is the resp.MiloBuild representation of the build.
+        respBuild *resp.MiloBuild `gae:,noindex`
+
+        // project is the luci project name of the build.
+        project string
+
+        // created is the time when this build entry was first created.
+        created time.Time
+
+        // last is the time when this build entry was last modified.
+        modified time.Time
+}
+
+// buildEntryKey returns the key for a build entry given a hostname and build ID.
+func buildEntryKey(host string, buildID int64) string {
+        return fmt.Sprintf("%s:%d", host, buildID)
+}
+
+func (b *buildEntry) getBuild() (*bucketApi.ApiCommonBuildMessage, error) {
+        msg := bucketApi.ApiCommonBuildMessage{}
+        err := json.Unmarshal(b.buildbucketData, &msg)
+        return &msg, err
+}