add interface check in handler

gregns1 · gregns1 · commit 2e354d17ff1d · 2025-01-03T13:16:43.000Z
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
 }
 

Original file line number	Diff line number	Diff line change
`@@ -671,6 +671,11 @@ func (h *handler) validateAndWriteHeaders(method handlerMethod, accessPermission`
`671`	`671`	`}`
`672`	`672`	`}`
`673`	`673`	`h.updateResponseWriter()`
	`674`	`+ // ensure wrapped ResponseWriter implements http.Flusher`
	`675`	`+ _, ok := h.response.(http.Flusher)`
	`676`	`+ if !ok {`
	`677`	`+ return fmt.Errorf("ResponseWriter does not implement Flusher interface")`
	`678`	`+ }`
`674`	`679`	`return nil`
`675`	`680`	`}`
`676`	`681`