客户端连接中断通过监听r.context().done()实现,当通道关闭时停止发送数据并释放资源。代码中使用goroutine监听该通道,一旦客户端断开连接,即输出日志并退出循环,从而避免无效的数据发送和资源浪费。
使用go语言构建服务器发送事件(SSE)服务器,关键在于利用Go的并发特性和标准库中的net/http包,实现高效、实时的事件推送。这不仅能让你快速构建原型,还能深入理解SSE协议的底层运作机制。
package main import ( "fmt" "log" "net/http" "time" ) func sseHandler(w http.ResponseWriter, r *http.Request) { // 设置响应头,关键在于Content-Type w.Header().Set("Content-Type", "text/event-stream") w.Header().Set("Cache-Control", "no-cache") w.Header().Set("Connection", "keep-alive") // 创建一个通道,用于发送数据 messageChan := make(chan string) // 启动一个goroutine,模拟数据源 go func() { defer close(messageChan) ticker := time.NewTicker(2 * time.Second) defer ticker.Stop() for range ticker.C { messageChan <- fmt.Sprintf("当前时间:%s", time.Now().Format(time.RFC3339)) } }() // 监听客户端断开连接 notify := r.Context().Done() go func() { <-notify log.Println("客户端已断开连接") return }() // 循环发送数据 for msg := range messageChan { fmt.Fprintf(w, "data: %snn", msg) // 强制刷新缓冲区,确保数据立即发送 flusher, ok := w.(http.Flusher) if ok { flusher.Flush() } } log.Println("SSE 完成") } func main() { http.HandleFunc("/events", sseHandler) log.Println("SSE服务器已启动,监听端口:8080") log.Fatal(http.ListenAndServe(":8080", nil)) }
如何处理客户端连接中断?
服务器端需要监听客户端的连接状态,一旦检测到连接中断,就应该停止发送数据,释放资源。上面的代码示例中,r.Context().Done() 就是用来监听客户端连接中断的。当客户端断开连接时,该通道会被关闭,goroutine 就会退出。如果没有这个机制,服务器可能会持续尝试向已经断开的连接发送数据,导致资源浪费甚至错误。
立即学习“go语言免费学习笔记(深入)”;
SSE与websocket相比,有哪些优缺点?
SSE是单向通信协议,服务器向客户端推送数据。WebSocket是双向通信协议,客户端和服务器可以互相发送数据。
- SSE优点: 简单易用,基于HTTP协议,无需额外的握手协议,服务器实现起来更简单。对只需要服务器向客户端推送数据的场景,效率更高。
- SSE缺点: 只能服务器向客户端推送数据,不支持客户端向服务器发送数据。浏览器支持度不如WebSocket。
- WebSocket优点: 双向通信,实时性更好,浏览器支持度高。
- WebSocket缺点: 实现相对复杂,需要处理连接的建立、维护和关闭。
选择哪种协议取决于具体的应用场景。如果只需要服务器向客户端推送数据,SSE是一个不错的选择。如果需要双向通信,WebSocket更合适。
如何优化SSE服务器的性能?
- 使用连接池: 避免频繁创建和销毁连接,使用连接池可以提高性能。
- 使用缓冲区: 将多个事件合并成一个消息发送,减少网络传输的次数。
- 启用Gzip压缩: 减少数据传输量,提高传输速度。
- 使用CDN: 将SSE服务器部署到CDN上,可以提高访问速度,降低服务器负载。
- 监控服务器性能: 定期监控服务器的CPU、内存、网络等指标,及时发现和解决性能问题。
SSE在哪些场景下比较适用?
- 实时更新: 股票行情、体育赛事直播、新闻推送等。
- 服务器监控: 实时显示服务器的CPU、内存、网络等指标。
- 消息通知: 向客户端推送消息通知,例如:邮件提醒、订单状态更新等。
- 日志流: 将服务器的日志实时推送到客户端,方便开发人员调试和监控。
- 在线游戏: 实时更新游戏状态,例如:玩家位置、血量等。
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END