add X-Session-ID header

webtor-io · Jul 23, 2021 · 6448c3b · 6448c3b
1 parent d5516b3
commit 6448c3b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/services/job_location.go b/services/job_location.go
@@ -518,7 +518,6 @@ func (s *JobLocation) invoke() (*Location, error) {
 			Value: s.cfg.UseSnapshot,
 		},
 		{
-
 			Name:  "SNAPSHOT_START_THRESHOLD",
 			Value: fmt.Sprintf("%f", s.cfg.SnapshotStartThreshold),
 		},

diff --git a/services/web.go b/services/web.go
@@ -186,6 +186,11 @@ func (s *Web) proxyHTTP(w http.ResponseWriter, r *http.Request, src *Source, log
 		domain = d
 	}
 
+	sessionID := ""
+	if sid, ok := claims["sessionID"].(string); ok {
+		sessionID = sid
+	}
+
 	promHTTPProxyRequestCurrent.WithLabelValues(string(source), src.GetEdgeName()).Inc()
 	defer func() {
 		promHTTPProxyRequestDuration.WithLabelValues(string(source), src.GetEdgeName(), strconv.Itoa(wi.GroupedStatusCode())).Observe(time.Since(wi.start).Seconds())
@@ -235,6 +240,7 @@ func (s *Web) proxyHTTP(w http.ResponseWriter, r *http.Request, src *Source, log
 		"X-Token":       src.Token,
 		"X-Api-Key":     apiKey,
 		"X-Client":      clientName,
+		"X-Session-ID":  sessionID,
 	}
 	rate, ok := claims["rate"].(string)
 	if ok {