Go语言实现SSE中转demo
Go语言实现SSE中转demo
文章概要：本文主要通过一个demo来介绍如何使用Go语言实现SSE中转。
本文内容来自：谷流仓AI - ai.guliucang.com
前提
- 已安装Go语言环境（参考这篇文章:Macbook安装Go以及镜像设置）
创建项目
- 创建项目目录
mkdir go-app && cd go-app
- 初始化项目
# 后面的模块名自己定义
go mod init example/user/go-app
创建文件
先看一下完成之后的目录结构：
go-app/
├─ go.mod
├─ http/
│ ├─ requests.go
│ └─ responses.go
└─ main.go
然后每个文件的代码如下：
- main.go
package mainimport ("example/user/go-app/http""github.com/gin-gonic/gin""log"
)// 主函数：初始化并启动 Gin 框架的 HTTP 服务器，支持 /event-stream 的 POST 和 GET 请求。
func main() {// 创建一个用于传递事件的通道ch := make(chan string)// 初始化并配置默认的 Gin路由器router := gin.Default()// 设置 POST /event-stream 的处理函数，用于处理 POST 请求router.POST("/event-stream", func(c *gin.Context) {http.HandleEventStreamPost(c, ch) // 处理 POST 请求的逻辑})// 设置 GET /event-stream 的处理函数，用于处理 GET 请求router.GET("/event-stream", func(c *gin.Context) {http.HandleEventStreamGet(c, ch) // 处理 GET 请求的逻辑})// 启动 HTTP 服务器并监听端口 9990，记录启动失败的错误日志log.Fatalf("error running HTTP server: %s\n", router.Run(":9990"))
}- 创建http目录，并创建文件
- requests.go
package httpimport ("errors""fmt""github.com/gin-gonic/gin""io"
)// EventStreamRequest 结构体定义了事件流请求的数据模型
type EventStreamRequest struct {Message string `form:"message" json:"message" binding:"required,max=100"` // 请求中必须提供的消息内容，最大长度为100
}// HandleEventStreamPost 处理POST方法的事件流请求
// c: Gin框架的上下文对象，用于处理HTTP请求和响应
// ch: 用于事件流通信的通道，将请求消息发送到此通道
func HandleEventStreamPost(c *gin.Context, ch chan string) {var request EventStreamRequest// 尝试绑定请求数据到EventStreamRequest结构体if err := c.ShouldBind(&request); err != nil {// 如果绑定失败，生成错误响应并返回errorMessage := fmt.Sprintf("request validation error: %s", err.Error())BadRequestResponse(c, errors.New(errorMessage))return}// 将请求消息发送到通道ch <- request.Message// 创建成功响应并返回CreatedResponse(c, &request.Message)return
}// HandleEventStreamGet 处理获取事件流的请求。
// c: Gin框架的上下文对象，用于处理HTTP请求和响应。
// ch: 一个字符串类型的通道，用于向客户端发送事件消息。
func HandleEventStreamGet(c *gin.Context, ch chan string) {// 使用Stream方法来建立一个服务器端事件流，不断检查通道中是否有新消息。c.Stream(func(w io.Writer) bool {// 如果通道中有消息，通过SSEvent方法以"message"事件类型发送到客户端。if msg, ok := <-ch; ok {c.SSEvent("message", msg)return true // 表示继续发送下一个事件}return false // 表示没有更多事件，结束流})return
}- reponses.go
package httpimport ("github.com/gin-gonic/gin""net/http"
)// JSendFailResponse 定义了一个失败响应的结构体，包含状态和数据字段
type JSendFailResponse[T any] struct {Status string `json:"status"` // 响应状态Data T `json:"data"` // 响应数据，这里泛型T可以是任意类型
}// JSendSuccessResponse 定义了一个成功响应的结构体，包含状态和可选的数据字段
type JSendSuccessResponse[T any] struct {Status string `json:"status"` // 响应状态Data T `json:"data,omitempty"` // 响应数据，成功时可选
}// BadRequestResponse 用于处理Bad Request错误，返回400状态码和错误信息
func BadRequestResponse(c *gin.Context, error error) {c.JSON(http.StatusBadRequest,JSendFailResponse[string]{Status: "fail", // 设置响应状态为失败Data: error.Error(), // 将错误信息填入数据字段},)return
}// CreatedResponse 用于处理创建资源的成功响应，返回201状态码和创建的资源信息
func CreatedResponse[T interface{}](c *gin.Context, i *T) {c.JSON(http.StatusCreated,JSendSuccessResponse[T]{Status: "success", // 设置响应状态为成功Data: *i, // 填入创建的资源信息},)return
}运行程序
- 启动项目
go run main.go
- 开一个终端，运行以下命令， 监听数据：
curl http://localhost:9990/event-stream
- 打开另一个终端，运行以下命令，发送数据：
curl -d '{"message":"Hello, Event Stream!"}' -H "Content-Type: application/json" -X POST http://localhost:9990/event-stream
- 观察第一个终端，可以看到数据已经发送过来了：
event:message
data:Hello, Event Stream!