Merge pull request #25 from buildkite/add-queue-to-lambda

Support the queue parameter and logs in lambda func

README.md

@@ -59,6 +59,19 @@ Buildkite > (Pipeline) > ScheduledJobsCount
 Buildkite > (Pipeline) > UnfinishedJobsCount
 ```
 
+## AWS Lambda
+
+An AWS Lambda bundle is created and published as part of the build process.
+
+It's entrypoint is `handler.handle`, it requires a `python2.7` environment and makes use of the following env vars:
+
+ - BUILDKITE_ORG
+ - BUILDKITE_TOKEN
+ - BUILDKITE_BACKEND
+ - BUILDKITE_QUEUE
+
+Checkout https://github.com/buildkite/elastic-ci-stack-for-aws/blob/master/templates/metrics.yml for examples of usage.
+
 ## License
 
 See [LICENSE.md](LICENSE.md) (MIT)

lambda.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"bytes"
 	"encoding/json"
 	"log"
 	"os"
@@ -15,12 +14,10 @@ import (
 )
 
 func handle(evt json.RawMessage, ctx *runtime.Context) (interface{}, error) {
-	output := &bytes.Buffer{}
-	log.SetOutput(output)
-
 	org := os.Getenv("BUILDKITE_ORG")
 	token := os.Getenv("BUILDKITE_TOKEN")
 	backendOpt := os.Getenv("BUILDKITE_BACKEND")
+	queue := os.Getenv("BUILDKITE_QUEUE")
 
 	config, err := buildkite.NewTokenConfig(token, false)
 	if err != nil {
@@ -35,6 +32,10 @@ func handle(evt json.RawMessage, ctx *runtime.Context) (interface{}, error) {
 		Historical: time.Hour * 24,
 	})
 
+	if queue != "" {
+		col.Queue = queue
+	}
+
 	var bk backend.Backend
 	if backendOpt == "statsd" {
 		bk, err = backend.NewStatsDBackend(os.Getenv("STATSD_HOST"), strings.ToLower(os.Getenv("STATSD_TAGS")) == "true")
@@ -58,7 +59,7 @@ func handle(evt json.RawMessage, ctx *runtime.Context) (interface{}, error) {
 	}
 
 	log.Printf("Finished in %s", time.Now().Sub(t))
-	return output.String(), nil
+	return "", nil
 }
 
 func init() {