Use responsewriter.UserProvidedDecorator instead of auto flush response (#6171)

Signed-off-by: hongming <coder.scala@gmail.com>
This commit is contained in:
hongming
2024-09-13 16:40:36 +08:00
committed by GitHub
parent 499e10d913
commit bbc476d47c
3 changed files with 21 additions and 9 deletions

View File

@@ -25,6 +25,7 @@ import (
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/types"
"k8s.io/apiserver/pkg/apis/audit"
"k8s.io/apiserver/pkg/endpoints/responsewriter"
"k8s.io/klog/v2"
clusterv1alpha1 "kubesphere.io/api/cluster/v1alpha1"
"kubesphere.io/api/iam/v1beta1"
@@ -398,6 +399,9 @@ func (a *auditing) eventToBytes(events []*Event) [][]byte {
return res
}
var _ http.ResponseWriter = &ResponseCapture{}
var _ responsewriter.UserProvidedDecorator = &ResponseCapture{}
type ResponseCapture struct {
http.ResponseWriter
wroteHeader bool
@@ -413,6 +417,10 @@ func NewResponseCapture(w http.ResponseWriter) *ResponseCapture {
}
}
func (c *ResponseCapture) Unwrap() http.ResponseWriter {
return c.ResponseWriter
}
func (c *ResponseCapture) Header() http.Header {
return c.ResponseWriter.Header()
}
@@ -425,9 +433,6 @@ func (c *ResponseCapture) Write(data []byte) (int, error) {
if err != nil {
return n, err
}
if flusher, ok := c.ResponseWriter.(http.Flusher); ok {
flusher.Flush()
}
return n, nil
}