From d05344c543d7ed2d69fff16a5a7cc35f43ee5235 Mon Sep 17 00:00:00 2001 From: JustSong Date: Sun, 30 Jun 2024 18:30:46 +0800 Subject: [PATCH] refactor: add back trim & flush --- common/render/render.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/common/render/render.go b/common/render/render.go index 0b988793..646b3777 100644 --- 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 {