Milo: Grpc endpoint for multiple builds on a builder

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"
@@ skipping 35 matching lines @@
 
         bs, err := json.Marshal(b)
         if err != nil {
                 return nil, err
         }
 
         // Marshal the build back into JSON format.
         return &milo.BuildbotBuildJSON{Data: bs}, nil
 }
 
+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")
+        }
+
+        // Perform an ACL check by fetching the master.
+        _, err := getMasterEntry(c, req.Master)
+        switch {
+        case err == errMasterNotFound:
+                return nil, grpc.Errorf(codes.NotFound, "Master not found")
+        case err != nil:
+                return nil, err
+        }
+
+        limit := req.Limit
+        if limit == 0 {
+                limit = 20
+        }
+
+        q := ds.NewQuery("buildbotBuild")
+        q = q.Eq("master", req.Master).
+                Eq("builder", req.Builder).
+                Limit(limit).
+                Order("-number")
+        if req.IncludeCurrent == false {
+                q = q.Eq("finished", true)
+        }
+        builds := []*buildbotBuild{}
+        err = ds.GetAll(c, q, &builds)
+        if err != nil {
+                return nil, err
+        }
+
+        results := make([]*milo.BuildbotBuildJSON, len(builds))
+        for i, b := range builds {
+                // 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) {
 
         if req.Name == "" {
                 return nil, grpc.Errorf(codes.InvalidArgument, "No master specified")
         }
 
         entry, err := getMasterEntry(c, req.Name)
@@ skipping 69 matching lines @@
 
         return &milo.CompressedMasterJSON{
                 Internal: entry.Internal,
                 Modified: &google.Timestamp{
                         Seconds: entry.Modified.Unix(),
                         Nanos:   int32(entry.Modified.Nanosecond()),
                 },
                 Data: gzbs.Bytes(),
         }, nil
 }