//后端代码

//注意 **我注释的代码，是不使用gin框架封装的Stream方法，也就是C.Stream(func())和C.ssevent(),只是C.Stream要改成for循环持续的从通道里面进行读，直到通道关闭，结束for循环**

package main

import (
	"catface/app/service/nlp/glm"
	_ "catface/bootstrap"
	"fmt"
	"io"
	"testing"
	// "time"

	"github.com/gin-gonic/gin"

)

func SSE(c *gin.Context) {
	// 设置响应头，告诉前端适用event-stream事件流交互
	//c.Writer.Header().Set("Content-Type", "text/event-stream")
	//c.Writer.Header().Set("Cache-Control", "no-cache")
	//c.Writer.Header().Set("Connection", "keep-alive")

	// 判断是否支持sse
	//w := c.Writer
	//flusher, _ := w.(http.Flusher)
	query := c.Query("query")

	// 接收前端页面关闭连接通知
	closeNotify := c.Request.Context().Done()

	// 开启协程监听前端页面是否关闭了连接，关闭连接会触发此方法
	go func() {
		<-closeNotify
		fmt.Println("SSE关闭了")
		return
	}()

	//新建一个通道，用于数据接收和响应
	Chan := make(chan string)

	// 异步接收GPT响应，然后把响应的数据发送到通道Chan
	go func() {
		err := glm.ChatStream(query, Chan)
		if err != nil {
			fmt.Println("Error", err)
		}

		close(Chan)
	}()

	// gin框架封装的stream,会持续的调用这个func方法，记得返回true;返回false代表结束调用func方法
	c.Stream(func(w io.Writer) bool {
		select {
		case i, ok := <-Chan:
			if !ok {
				return false
			}
			c.SSEvent("chat", i) // c.SSEvent会自动修改响应头为事件流，并发送”test“事件流给前端监听”test“的回调方法
			//flusher.Flush() // 确保立即发送
			return true
		case <-closeNotify:
			fmt.Println("SSE关闭了")
			return false
		}
	})
}

func TestSSE(t *testing.T) {
	engine := gin.Default()
	// 设置跨域中间件
	engine.Use(func(context *gin.Context) {
		origin := context.GetHeader("Origin")
		// 允许 Origin 字段中的域发送请求
		context.Writer.Header().Add("Access-Control-Allow-Origin", origin) // 这边我的前端页面在63342，会涉及跨域，这个根据自己情况设置，或者直接设置为”*“，放行所有的
		// 设置预验请求有效期为 86400 秒
		context.Writer.Header().Set("Access-Control-Max-Age", "86400")
		// 设置允许请求的方法
		context.Writer.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, U`PDATE, PATCH")
		// 设置允许请求的 Header
		context.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Content-Length, Apitoken")
		// 设置拿到除基本字段外的其他字段，如上面的Apitoken, 这里通过引用Access-Control-Expose-Headers，进行配置，效果是一样的。
		context.Writer.Header().Set("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Headers")
		// 配置是否可以带认证信息
		context.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
		// OPTIONS请求返回200
		if context.Request.Method == "OPTIONS" {
			fmt.Println(context.Request.Header)
			context.AbortWithStatus(200)
		} else {
			context.Next()
		}
	})
	engine.GET("/admin/rag/default_talk", SSE) // TIP 记得适用get请求，我用post前端报404，资料说是SSE只支持get请求
	engine.Run(":20201")
}