Milo: Buildbucket PubSub ingestion outline

milo/frontend/config.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"
 
         "cloud.google.com/go/datastore"
+        "google.golang.org/appengine"
 
         "github.com/luci/luci-go/common/logging"
         "github.com/luci/luci-go/milo/common"
         "github.com/luci/luci-go/server/router"
         "github.com/luci/luci-go/server/templates"
 )
 
 // ConfigsHandler renders the page showing the currently loaded set of luci-configs.
 func ConfigsHandler(c *router.Context) {
         projects, err := common.GetAllProjects(c.Context)
@@ skipping 13 matching lines @@
 
         templates.MustRender(c.Context, c.Writer, "pages/configs.html", templates.Args{
                 "Projects":      projects,
                 "ServiceConfig": sc,
         })
 }
 
 // UpdateHandler is an HTTP handler that handles configuration update requests.
 func UpdateConfigHandler(ctx *router.Context) {
         c, h := ctx.Context, ctx.Writer
+        // Needed to access the PubSub API
+        c = appengine.WithContext(c, ctx.Request)
         projErr := common.UpdateProjectConfigs(c)
         if projErr != nil {
                 logging.WithError(projErr).Errorf(c, "project update handler encountered error")
         }
-        servErr := common.UpdateServiceConfig(c)
+        settings, servErr := common.UpdateServiceConfig(c)
         if servErr != nil {
                 logging.WithError(servErr).Errorf(c, "service update handler encountered error")
+        } else {
+                servErr = common.EnsurePubSubSubscribed(c, settings)
+                if servErr != nil {
+                        logging.WithError(servErr).Errorf(
+                                c, "pubsub subscriber handler encountered error")
+                }
         }
         if projErr != nil || servErr != nil {
                 h.WriteHeader(http.StatusInternalServerError)
                 return
         }
         h.WriteHeader(http.StatusOK)
 }