From 554e666e32ba7e7729dfc1cb605aab7f27fc8019 Mon Sep 17 00:00:00 2001 From: Ryan Gregg Date: Fri, 3 May 2019 13:02:34 -0700 Subject: [PATCH] Add validation token support (#1248) --- docs/serving/samples/autoscale-go/autoscale.go | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/docs/serving/samples/autoscale-go/autoscale.go b/docs/serving/samples/autoscale-go/autoscale.go index d241ce2e1..caaf1657b 100644 --- a/docs/serving/samples/autoscale-go/autoscale.go +++ b/docs/serving/samples/autoscale-go/autoscale.go @@ -22,6 +22,7 @@ import ( "strconv" "sync" "time" + "os" ) // Algorithm from https://stackoverflow.com/a/21854246 @@ -153,7 +154,22 @@ func handler(w http.ResponseWriter, r *http.Request) { } } +func replyWithToken(token string) func(http.ResponseWriter, *http.Request) { + return func(w http.ResponseWriter, r *http.Request) { + fmt.Fprintln(w, token) + } + } + func main() { + validateToken := os.Getenv("VALIDATION") + if validateToken != "" { + http.HandleFunc("/" + validateToken + "/", replyWithToken(validateToken)) + } + + listenPort := os.Getenv("PORT") + if listenPort == "" { + listenPort = "8080" + } http.HandleFunc("/", handler) - http.ListenAndServe(":8080", nil) + http.ListenAndServe(":" + listenPort, nil) }