add interface check in handler

couchbase · Jan 3, 2025 · 2e354d1 · 2e354d1
1 parent b989e99
commit 2e354d1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/rest/handler.go b/rest/handler.go
@@ -671,6 +671,11 @@ func (h *handler) validateAndWriteHeaders(method handlerMethod, accessPermission
 		}
 	}
 	h.updateResponseWriter()
+	// ensure wrapped ResponseWriter implements http.Flusher
+	_, ok := h.response.(http.Flusher)
+	if !ok {
+		return fmt.Errorf("ResponseWriter does not implement Flusher interface")
+	}
 	return nil
 }