From e3b83f886f71cd02ef97e53e2d38abed0d8254a3 Mon Sep 17 00:00:00 2001
From: CalciumIon <1808837298@qq.com>
Date: Wed, 17 Jul 2024 16:43:55 +0800
Subject: [PATCH] fix: try to fix panic #369

---
 service/relay.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

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