当前位置：首页 > 文章列表 > Golang > Go教程 > Go语言实战：手把手教你用SSE实现网页服务器推送

Go语言实战：手把手教你用SSE实现网页服务器推送

2025-06-22 23:33:34 0浏览收藏

想掌握Go语言服务器推送技术？本文手把手教你使用Go语言和SSE（Server-Sent Events）协议，构建一个高效、实时的服务器推送应用。通过本文，你将学会如何利用Go的并发特性和标准库`net/http`包，快速实现服务器向客户端单向推送数据。文章重点讲解了如何处理客户端连接中断，通过监听`r.Context().Done()`通道，优雅地停止数据发送并释放资源，避免资源浪费。同时，对比SSE与WebSocket的优缺点，并分享优化SSE服务器性能的实用技巧，以及SSE适用的典型应用场景，助你深入理解SSE协议并应用于实际项目。立即学习，提升你的Go语言实战技能！

客户端连接中断通过监听r.Context().Done()实现，当通道关闭时停止发送数据并释放资源。代码中使用goroutine监听该通道，一旦客户端断开连接，即输出日志并退出循环，从而避免无效的数据发送和资源浪费。

快速上手：利用Go语言构建SSE服务器推送

使用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: %s\n\n", 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 就会退出。如果没有这个机制，服务器可能会持续尝试向已经断开的连接发送数据，导致资源浪费甚至错误。