fix: use ChunkedBodyWriter

hertz-contrib · May 4, 2023 · 9045eb7 · 9045eb7
1 parent 5b0bd5a
commit 9045eb7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 55 deletions.
diff --git a/sse.go b/sse.go
@@ -17,6 +17,7 @@
 package sse
 
 import (
+	"github.com/cloudwego/hertz/pkg/protocol/http1/resp"
 	"strings"
 
 	"github.com/cloudwego/hertz/pkg/app"
@@ -65,7 +66,7 @@ func NewStream(c *app.RequestContext) *Stream {
 		c.Response.Header.Set(cacheControl, noCache)
 	}
 
-	writer := NewStreamBodyWriter(&c.Response, c.GetWriter())
+	writer := resp.NewChunkedBodyWriter(&c.Response, c.GetWriter())
 	c.Response.HijackWriter(writer)
 	return &Stream{
 		writer,

diff --git a/writer.go b/writer.go