Havoc412
609e02dfd2
- 新增猫脸识别失败的错误码和错误信息 - 重构猫脸识别结果处理逻辑，增加对结果为空的处理 - 优化猫脸识别结果展示，包括品种翻译和动物信息展示 - 新增测试用例，验证猫脸识别功能正常工作
99 lines
3.5 KiB
Go
99 lines
3.5 KiB
Go
// //后端代码
|
||
|
||
// //注意 **我注释的代码，是不使用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")
|
||
// }
|