xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-19 00:46:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: try to fix panic #369

Browse Source
This commit is contained in:
CalciumIon 2024-07-17 16:43:55 +08:00
parent 4d0d18931d
commit e3b83f886f
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
service/relay.go
View File

@ -5,6 +5,7 @@ import (
"errors" "errors"
"fmt" "fmt"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"net/http"
"one-api/common" "one-api/common"
) )
@ -20,10 +21,10 @@ func StringData(c *gin.Context, str string) error {
//str = strings.TrimPrefix(str, "data: ") //str = strings.TrimPrefix(str, "data: ")
//str = strings.TrimSuffix(str, "\r") //str = strings.TrimSuffix(str, "\r")
c.Render(-1, common.CustomEvent{Data: "data: " + str}) c.Render(-1, common.CustomEvent{Data: "data: " + str})
if c.Writer != nil { if flusher, ok := c.Writer.(http.Flusher); ok {
c.Writer.Flush() flusher.Flush()
} else { } else {
return errors.New("writer is nil") return errors.New("streaming error: flusher not found")
} }
return nil return nil
} }