go语言SSE客户端

package main

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/r3labs/sse/v2"
	"net/http"
	"net/url"
	"strings"
)

func main() {
	r := gin.Default()
	r.Use(func(c *gin.Context) {
		c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
		c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
		c.Writer.Header().Set("Access-Control-Allow-Headers", "Accept, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization")
	})

	r.POST("/stream", ChatGPTSSE)
	r.Run(":8080")
}

func ChatGPTSSE(c *gin.Context) {
	fmt.Println("开始调用")
	c.Status(http.StatusOK)
	c.Header("Content-Type", "text/event-stream")
	c.Writer.Flush()
	client := &http.Client{
		// 设置Transport字段为自定义Transport，包含代理设置
		Transport: &http.Transport{
			Proxy: func(req *http.Request) (*url.URL, error) {
				// 设置代理
				proxyURL, err := url.Parse("http://127.0.0.1:7890")
				if err != nil {
					return nil, err
				}
				return proxyURL, nil
			},
		},
	}
	// 创建 SSE 客户端
	sseClient := sse.NewClient("https://api.openai.com/v1/chat/completions")
	sseClient.Connection = client
	// 设置请求体
	reqBody := strings.NewReader(`{"model": "gpt-3.5-turbo","messages": [{"role": "user", "content": "你的版本是多少"}],"stream": true}`)
	sseClient.Body = reqBody
	sseClient.Method = "POST"
	headers := make(map[string]string)
	headers["Authorization"] = "Bearer YOUR_TOKEN"
	headers["Content-Type"] = "application/json"
	sseClient.Headers = headers

	// 监听 SSE 事件
	err := sseClient.SubscribeRaw(func(msg *sse.Event) {
		fmt.Printf("Received event: %v\n", string(msg.Data))
		fmt.Fprintf(c.Writer, "data: %s\n\n", string(msg.Data))
		c.Writer.Flush()
	})
	if err != nil {
		return
	}
	fmt.Println("结束调用")
}