Promote LogDog aliases for BuildBot JSON data.

milo/appengine/buildbot/grpc.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 buildbot
 
 import (
         "bytes"
         "compress/gzip"
         "encoding/json"
 
+        "golang.org/x/net/context"
         "google.golang.org/grpc"
         "google.golang.org/grpc/codes"
 
         "github.com/golang/protobuf/ptypes/timestamp"
         ds "github.com/luci/gae/service/datastore"
         "github.com/luci/luci-go/common/iotools"
         "github.com/luci/luci-go/common/logging"
         milo "github.com/luci/luci-go/milo/api/proto"
         "github.com/luci/luci-go/server/auth"
-        "golang.org/x/net/context"
 )
 
 // Service is a service implementation that displays BuildBot builds.
 type Service struct{}
 
 var errNotFoundGRPC = grpc.Errorf(codes.NotFound, "Master Not Found")
 
 // GetBuildbotBuildJSON implements milo.BuildbotServer.
-func (s *Service) GetBuildbotBuildJSON(
-        c context.Context, req *milo.BuildbotBuildRequest) (
+func (s *Service) GetBuildbotBuildJSON(c context.Context, req *milo.BuildbotBuildRequest) (
         *milo.BuildbotBuildJSON, error) {
 
         if req.Master == "" {
                 return nil, grpc.Errorf(codes.InvalidArgument, "No master specified")
         }
         if req.Builder == "" {
                 return nil, grpc.Errorf(codes.InvalidArgument, "No builder specified")
         }
 
         cu := auth.CurrentUser(c)
         logging.Debugf(c, "%s is requesting %s/%s/%d",
                 cu.Identity, req.Master, req.Builder, req.BuildNum)
 
         b, err := getBuild(c, req.Master, req.Builder, int(req.BuildNum))
         switch {
         case err == errBuildNotFound:
                 return nil, grpc.Errorf(codes.NotFound, "Build not found")
         case err == errNotAuth:
                 return nil, grpc.Errorf(codes.Unauthenticated, "Unauthenticated request")
         case err != nil:
                 return nil, err
         }
 
+        updatePostProcessBuild(b)
         bs, err := json.Marshal(b)
         if err != nil {
                 return nil, err
         }
 
         // Marshal the build back into JSON format.
         return &milo.BuildbotBuildJSON{Data: bs}, nil
 }
 
 // GetBuildbotBuildsJSON implements milo.BuildbotServer.
-func (s *Service) GetBuildbotBuildsJSON(
-        c context.Context, req *milo.BuildbotBuildsRequest) (
+func (s *Service) GetBuildbotBuildsJSON(c context.Context, req *milo.BuildbotBuildsRequest) (
         *milo.BuildbotBuildsJSON, error) {
 
         if req.Master == "" {
                 return nil, grpc.Errorf(codes.InvalidArgument, "No master specified")
         }
         if req.Builder == "" {
                 return nil, grpc.Errorf(codes.InvalidArgument, "No builder specified")
         }
 
         limit := req.Limit
@@ skipping 25 matching lines @@
                 q = q.Eq("finished", true)
         }
         builds := []*buildbotBuild{}
         err = getBuildQueryBatcher(c).GetAll(c, q, &builds)
         if err != nil {
                 return nil, err
         }
 
         results := make([]*milo.BuildbotBuildJSON, len(builds))
         for i, b := range builds {
+                updatePostProcessBuild(b)
+
                 // In theory we could do this in parallel, but it doesn't actually go faster
                 // since AppEngine is single-cored.
                 bs, err := json.Marshal(b)
                 if err != nil {
                         return nil, err
                 }
                 results[i] = &milo.BuildbotBuildJSON{Data: bs}
         }
         return &milo.BuildbotBuildsJSON{
                 Builds: results,
         }, nil
 }
 
 // GetCompressedMasterJSON assembles a CompressedMasterJSON object from the
 // provided MasterRequest.
-func (s *Service) GetCompressedMasterJSON(
-        c context.Context, req *milo.MasterRequest) (*milo.CompressedMasterJSON, error) {
+func (s *Service) GetCompressedMasterJSON(c context.Context, req *milo.MasterRequest) (
+        *milo.CompressedMasterJSON, error) {
 
         if req.Name == "" {
                 return nil, grpc.Errorf(codes.InvalidArgument, "No master specified")
         }
 
         cu := auth.CurrentUser(c)
         logging.Debugf(c, "%s is making a master request for %s", cu.Identity, req.Name)
 
         entry, err := getMasterEntry(c, req.Name)
         switch {
@@ skipping 23 matching lines @@
                                         Number:      buildNum,
                                 })
                         }
                 }
                 if err := ds.Get(c, slave.Runningbuilds); err != nil {
                         logging.WithError(err).Errorf(c,
                                 "Encountered error while trying to fetch running builds for %s: %v",
                                 master.Name, slave.Runningbuilds)
                         return nil, err
                 }
+
+                for _, b := range slave.Runningbuilds {
+                        updatePostProcessBuild(b)
+                }
         }
 
         // Also inject cached builds information.
         for builderName, builder := range master.Builders {
                 // Get the most recent 50 buildNums on the builder to simulate what the
                 // cachedBuilds field looks like from the real buildbot master json.
                 q := ds.NewQuery("buildbotBuild").
                         Eq("finished", true).
                         Eq("master", req.Name).
                         Eq("builder", builderName).
@@ skipping 26 matching lines @@
 
         return &milo.CompressedMasterJSON{
                 Internal: entry.Internal,
                 Modified: &timestamp.Timestamp{
                         Seconds: entry.Modified.Unix(),
                         Nanos:   int32(entry.Modified.Nanosecond()),
                 },
                 Data: gzbs.Bytes(),
         }, nil
 }