File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed
Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -19,7 +19,8 @@ import (
1919)
2020
2121const (
22- HeaderAccessKey = "X-Access-Key"
22+ HeaderAccessKey = "X-Access-Key"
23+ HeaderSessionType = "Session-Type"
2324)
2425
2526type AccessKeyFunc func (* http.Request ) string
Original file line number Diff line number Diff line change @@ -146,9 +146,9 @@ func Session(cfg Options) func(next http.Handler) http.Handler {
146146 if _ , ok := GetSessionType (ctx ); ok {
147147 // Track this as a SpecialKey session for now.
148148 // TODO: Remove once node-gateway SpecialKey support is gone.
149+ w .Header ().Set (HeaderSessionType , "SpecialKey" )
149150 httplog .SetAttrs (ctx , slog .String ("sessionType" , "SpecialKey" ))
150151 requestsCounter .Inc (sessionLabels {SessionType : "SpecialKey" , RateLimited : "false" })
151-
152152 next .ServeHTTP (w , r )
153153 return
154154 }
@@ -260,6 +260,7 @@ func Session(cfg Options) func(next http.Handler) http.Handler {
260260 }
261261
262262 ctx = WithSessionType (ctx , sessionType )
263+ w .Header ().Set (HeaderSessionType , sessionType .String ())
263264 httplog .SetAttrs (ctx , slog .String ("sessionType" , sessionType .String ()))
264265
265266 ww , ok := w .(middleware.WrapResponseWriter )
You can’t perform that action at this time.
0 commit comments