b21b3b5b46
* refactor: abusing goroutines * fix: trim data prefix * refactor: move functions to render package * refactor: add back trim & flush --------- Co-authored-by: JustSong <quanpengsong@gmail.com>
30 lines
613 B
Go
30 lines
613 B
Go
package render
|
|
|
|
import (
|
|
"encoding/json"
|
|
"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 {
|
|
jsonData, err := json.Marshal(object)
|
|
if err != nil {
|
|
return fmt.Errorf("error marshalling object: %w", err)
|
|
}
|
|
StringData(c, string(jsonData))
|
|
return nil
|
|
}
|
|
|
|
func Done(c *gin.Context) {
|
|
StringData(c, "[DONE]")
|
|
}
|