fix bug of resume flag

Jing-ze · Dec 26, 2024 · e98bf3c · e98bf3c
1 parent 10767df
commit e98bf3c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pkg/middleware/stored_session.go b/pkg/middleware/stored_session.go
@@ -104,6 +104,7 @@ func (s *StoredSessionLoader) loadSession(next http.Handler) http.Handler {
 			updateKeysCallback := func(args ...interface{}) {
 				resumeFlag := args[0].(bool)
 				validateSessionCallback := func(args ...interface{}) {
+					resumeFlag := args[0].(bool)
 					scope.Session = session
 					next.ServeHTTP(rw, req)
 					if resumeFlag {
@@ -118,10 +119,10 @@ func (s *StoredSessionLoader) loadSession(next http.Handler) http.Handler {
 						session = nil
 					}
 					if !isAsync {
-						validateSessionCallback()
+						validateSessionCallback(resumeFlag)
 					}
 				} else {
-					validateSessionCallback()
+					validateSessionCallback(resumeFlag)
 				}
 			}
 			keysNeedsUpdate := (session != nil) && (s.NeedsVerifier)

diff --git a/providers/internal_util.go b/providers/internal_util.go
@@ -70,7 +70,7 @@ func validateToken(ctx context.Context, p Provider, accessToken string, header h
 	client.Get(endpoint, headerArray, func(statusCode int, responseHeaders http.Header, responseBody []byte) {
 		util.Logger.Debugf("%d GET %s %s", statusCode, stripToken(endpoint), responseBody)
 		if statusCode == 200 {
-			callback()
+			callback(true)
 		} else {
 			util.SendError(fmt.Sprintf("token validation request failed: status %d - %s", statusCode, responseBody), nil, http.StatusInternalServerError)
 		}