Kitchen: Generate LogDog prefix from Swarming.

go/src/infra/tools/kitchen/kitchen.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 main
 
 import (
         "bytes"
         "os"
         "os/signal"
 
         "github.com/maruel/subcommands"
         "golang.org/x/net/context"
 
         "github.com/luci/luci-go/common/cli"
+        "github.com/luci/luci-go/common/environ"
         "github.com/luci/luci-go/common/errors"
         log "github.com/luci/luci-go/common/logging"
         "github.com/luci/luci-go/common/logging/gologger"
 )
 
 var application = &cli.Application{
         Name:  "kitchen",
         Title: "Kitchen. It can run a recipe.",
         Context: func(ctx context.Context) context.Context {
                 ctx = gologger.StdConfig.Use(ctx)
@@ skipping 35 matching lines @@
 
         var buf bytes.Buffer
         st := errors.RenderStack(err)
         if _, derr := st.DumpTo(&buf); derr != nil {
                 // This can't really fail, since we're rendering to a Buffer.
                 panic(derr)
         }
 
         log.Errorf(ctx, "Annotated error stack:\n%s", buf.String())
 }
+
+// getSwarmingTaskParameters retrieves the Swarming task parameters from the
+// environment.
+func getSwarmingTaskParameters(env environ.Env) (host, taskID string, err error) {
+        if host, _ = env.Get("SWARMING_SERVER"); host == "" {
+                err = errors.Reason("missing or empty SWARMING_SERVER").Err()
+                return
+        }
+        if taskID, _ = env.Get("SWARMING_TASK_ID"); taskID == "" {
+                err = errors.Reason("missing SWARMING_TASK_ID").Err()
+                return
+        }
+
+        return
+}