refactor: add back trim & flush

2024-06-30 18:30:46 +08:00 · 2024-06-30 18:30:46 +08:00 · d05344c543
commit d05344c543
parent 616759933a
1 changed files with 4 additions and 0 deletions
--- a/common/render/render.go
+++ b/common/render/render.go
@ -5,10 +5,14 @@ import (
 	"fmt"
 	"github.com/gin-gonic/gin"
 	"github.com/songquanpeng/one-api/common"
+	"strings"
 )

 func StringData(c *gin.Context, str string) {
+	str = strings.TrimPrefix(str, "data: ")
+	str = strings.TrimSuffix(str, "\r")
 	c.Render(-1, common.CustomEvent{Data: "data: " + str})
+	c.Writer.Flush()
 }

 func ObjectData(c *gin.Context, object interface{}) error {