appengine/internal: add cloud trace header to Managed VM apps
Change-Id: I6da52649325496d570726e2b14908798c66df8e3
diff --git a/internal/api.go b/internal/api.go
index e031cab..90c4838 100644
--- a/internal/api.go
+++ b/internal/api.go
@@ -39,6 +39,7 @@
// Incoming headers.
ticketHeader = http.CanonicalHeaderKey("X-AppEngine-API-Ticket")
dapperHeader = http.CanonicalHeaderKey("X-Google-DapperTraceInfo")
+ traceHeader = http.CanonicalHeaderKey("X-Cloud-Trace-Context")
curNamespaceHeader = http.CanonicalHeaderKey("X-AppEngine-Current-Namespace")
userIPHeader = http.CanonicalHeaderKey("X-AppEngine-User-IP")
remoteAddrHeader = http.CanonicalHeaderKey("X-AppEngine-Remote-Addr")
@@ -387,6 +388,9 @@
if info := c.req.Header.Get(dapperHeader); info != "" {
hreq.Header.Set(dapperHeader, info)
}
+ if info := c.req.Header.Get(traceHeader); info != "" {
+ hreq.Header.Set(traceHeader, info)
+ }
tr := apiHTTPClient.Transport.(*http.Transport)
