Buildbucket: Save buildbucket build info and summary on pubsub push

milo/buildsource/swarming/build.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 swarming
 
 import (
         "bytes"
         "fmt"
         "net/http"
@@ skipping 71 matching lines @@
         getSwarmingResult(c context.Context, taskID string) (*swarming.SwarmingRpcsTaskResult, error)
         getSwarmingRequest(c context.Context, taskID string) (*swarming.SwarmingRpcsTaskRequest, error)
         getTaskOutput(c context.Context, taskID string) (string, error)
 }
 
 type prodSwarmingService struct {
         host   string
         client *swarming.Service
 }
 
-func newProdService(c context.Context, host string) (*prodSwarmingService, error) {
+func NewProdService(c context.Context, host string) (*prodSwarmingService, error) {
         client, err := getSwarmingClient(c, host)
         if err != nil {
                 return nil, err
         }
         return &prodSwarmingService{
                 host:   host,
                 client: client,
         }, nil
 }
 
@@ skipping 375 matching lines @@
         }
         // If this ever has more than one stream then memoryClient needs to become
         // goroutine safe
         if err := p.RunStreams([]*annotee.Stream{&is}); err != nil {
                 return nil, err
         }
         p.Finish()
         return c.ToLogDogStreams()
 }
 
-// buildLoader represents the ability to load a Milo build from a Swarming task.
+// BuildLoader represents the ability to load a Milo build from a Swarming task.
 //
 // It exists so that the internal build loading functionality can be stubbed out
 // for testing.
-type buildLoader struct {
+type BuildLoader struct {
         // logdogClientFunc returns a coordinator Client instance for the supplied
         // parameters.
         //
         // If nil, a production client will be generated.
         logDogClientFunc func(c context.Context, host string) (*coordinator.Client, error)
 }
 
-func (bl *buildLoader) newEmptyAnnotationStream(c context.Context, addr *types.StreamAddr) (
+func (bl *BuildLoader) newEmptyAnnotationStream(c context.Context, addr *types.StreamAddr) (
         *rawpresentation.AnnotationStream, error) {
 
         fn := bl.logDogClientFunc
         if fn == nil {
                 fn = rawpresentation.NewClient
         }
         client, err := fn(c, addr.Host)
         if err != nil {
                 return nil, errors.Annotate(err, "failed to create LogDog client").Err()
         }
@@ skipping 28 matching lines @@
         infoComp := infoComponent(model.InfraFailure,
                 "LogDog stream not found", "Job likely failed to start.")
         infoComp.Started = build.Summary.Started
         infoComp.Finished = build.Summary.Finished
         infoComp.Duration = build.Summary.Duration
         infoComp.Verbosity = resp.Interesting
         build.Components = append(build.Components, infoComp)
         return addTaskToBuild(c, host, res, build)
 }
 
-func (bl *buildLoader) swarmingBuildImpl(c context.Context, svc swarmingService, linkBase, taskID string) (*resp.MiloBuild, error) {
+func (bl *BuildLoader) SwarmingBuildImpl(c context.Context, svc swarmingService, linkBase, taskID string) (*resp.MiloBuild, error) {
         // Fetch the data from Swarming
         var logDogStreamAddr *types.StreamAddr
 
         fetchParams := swarmingFetchParams{
                 fetchRes: true,
                 fetchLog: true,
 
                 // Cancel if LogDog annotation stream parameters are present in the tag set.
                 taskResCallback: func(res *swarming.SwarmingRpcsTaskResult) (cancelLogs bool) {
                         // If the build hasn't started yet, then there is no LogDog log stream to
@@ skipping 208 matching lines @@
         for _, tag := range v {
                 var value string
                 parts := strings.SplitN(tag, ":", 2)
                 if len(parts) == 2 {
                         value = parts[1]
                 }
                 res[parts[0]] = value
         }
         return res
 }