idlewatcher fixed idlewatcher incorrect respond haviour, keep url path

yusing · Oct 7, 2024 · d89155a · d89155a
1 parent 921ce23
commit d89155a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/internal/docker/idlewatcher/waker.go b/internal/docker/idlewatcher/waker.go
@@ -45,7 +45,8 @@ func (w *Waker) wake(next http.HandlerFunc, rw http.ResponseWriter, r *http.Requ
 	ctx, cancel := context.WithTimeout(r.Context(), w.WakeTimeout)
 	defer cancel()
 
-	if r.Header.Get(headerCheckRedirect) == "" {
+	isCheckRedirect := r.Header.Get(headerCheckRedirect) != ""
+	if !isCheckRedirect {
 		// Send a loading response to the client
 		rw.Header().Set("Content-Type", "text/html; charset=utf-8")
 		rw.Write(w.makeRespBody("%s waking up...", w.ContainerName))
@@ -63,7 +64,11 @@ func (w *Waker) wake(next http.HandlerFunc, rw http.ResponseWriter, r *http.Requ
 
 	// maybe another request came in while we were waiting for the wake
 	if w.ready.Load() {
-		next(rw, r)
+		if isCheckRedirect {
+			rw.WriteHeader(http.StatusOK)
+		} else {
+			next(rw, r)
+		}
 		return
 	}