Milo: pRPC endpoint for getting Buildbot master data

milo/appengine/frontend/milo.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 frontend
 
 import (
         "net/http"
 
+        "golang.org/x/net/context"
+
+        "github.com/golang/protobuf/proto"
         "github.com/luci/luci-go/appengine/gaemiddleware"
+        "github.com/luci/luci-go/grpc/discovery"
+        "github.com/luci/luci-go/grpc/prpc"
+        milo "github.com/luci/luci-go/milo/api/proto"
         "github.com/luci/luci-go/milo/appengine/buildbot"
         "github.com/luci/luci-go/milo/appengine/buildbucket"
         "github.com/luci/luci-go/milo/appengine/console"
         "github.com/luci/luci-go/milo/appengine/logdog"
         "github.com/luci/luci-go/milo/appengine/settings"
         "github.com/luci/luci-go/milo/appengine/swarming"
         "github.com/luci/luci-go/server/router"
 )
 
+func emptyPrelude(c context.Context, methodName string, req proto.Message) (context.Context, error) {
+        return c, nil
+}
+
 // Where it all begins!!!
 func init() {
         // Register plain ol' http handlers.
         r := router.New()
         basemw := settings.Base()
         gaemiddleware.InstallHandlers(r, basemw)
         r.GET("/", basemw, settings.Wrap(frontpage{}))
 
         // Admin and cron endpoints.
         r.GET("/admin/update", basemw.Extend(gaemiddleware.RequireCron),
@@ skipping 21 matching lines @@
         // LogDog Milo Annotation Streams.
         r.GET("/logdog/build/:project/*path", basemw, settings.Wrap(&logdog.AnnotationStream{}))
 
         // User settings
         r.GET("/settings", basemw, settings.Wrap(settings.Settings{}))
         r.POST("/settings", basemw, settings.ChangeSettings)
 
         // PubSub subscription endpoints.
         r.POST("/pubsub/buildbot", basemw, buildbot.PubSubHandler)
 
-        http.Handle("/", r)
+        // pRPC style endpoints.
+        var api prpc.Server
+        milo.RegisterBuildbotServer(&api, &milo.DecoratedBuildbot{
+                Service: &buildbot.BuildbotService{},
+                Prelude: emptyPrelude,
+        })
+        discovery.Enable(&api)
+        api.InstallHandlers(r, gaemiddleware.BaseProd())
+
+        http.DefaultServeMux.Handle("/", r)
 }