Go WebSocket EOF错误处理与连接管理-Golang-PHP中文网

Go WebSocket EOF错误处理与连接管理

心靈之曲

发布： 2025-08-26 17:30:35

原创

270人浏览过

go websocket eof错误处理与连接管理

在Go语言中使用WebSocket时，经常会遇到EOF（End Of File）错误，这通常是由于WebSocket连接意外断开导致的。原始问题中，服务端在处理完第一个请求后，会循环出现EOF错误，需要重新连接才能继续工作。根本原因是处理WebSocket连接的goroutine在完成首次请求后就结束了，导致连接被关闭。为了解决这个问题，我们需要确保每个WebSocket连接都有一个独立的goroutine来处理，并且这个goroutine要持续运行，直到连接主动关闭或者发生错误。

WebSocket连接处理

核心思路是在接受WebSocket连接时，为每个连接创建一个新的goroutine。这个goroutine负责接收和发送消息，以及处理连接过程中可能发生的错误。

func WSHandler(ws *websocket.Conn) {
    defer ws.Close()
    fmt.Println("Client Connected")
    for {
        var message Message
        err := websocket.JSON.Receive(ws, &message)
        if err != nil {
            fmt.Printf("Error: %s\n", err.Error())
            return
        }
        fmt.Println(message)

        // do something useful here...

        response := new(Message)
        response.RequestID = message.RequestID
        response.Success = true
        response.SomeOtherThing = "The hot dog left the castle as requested."
        err = websocket.JSON.Send(ws, response)
        if err != nil {
            fmt.Printf("Send failed: %s\n", err.Error())
            return
        }
    }
}

登录后复制

在上面的代码中，WSHandler 函数是WebSocket连接的处理函数。defer ws.Close() 确保在函数退出时关闭WebSocket连接。for 循环用于持续接收和发送消息。websocket.JSON.Receive 用于接收客户端发送的JSON消息，如果发生错误（例如连接断开），则会返回错误，导致循环退出，goroutine结束，连接关闭。websocket.JSON.Send 用于向客户端发送JSON消息。

服务器端实现

服务器端需要监听指定的地址和端口，并为每个新的WebSocket连接启动一个goroutine。

func RunServer() {
    http.Handle("/", http.FileServer(http.Dir("www")))
    http.Handle("/server", websocket.Handler(WSHandler))
    fmt.Println("Starting Server")
    err := http.ListenAndServe(*address, nil)
    if err != nil {
        fmt.Printf("HTTP failed: %s\n", err.Error())
        os.Exit(1)
    }
}

登录后复制

http.Handle("/server", websocket.Handler(WSHandler)) 将 /server 路径注册为WebSocket处理程序，当客户端连接到 /server 时，WSHandler 函数会被调用。http.ListenAndServe 启动HTTP服务器，监听指定的地址和端口。

客户端实现

客户端需要连接到服务器，并持续发送和接收消息。

钉钉 AI 助理

钉钉AI助理汇集了钉钉AI产品能力，帮助企业迈入智能新时代。

查看详情

func RunClient() {
    fmt.Println("Starting Client")
    ws, err := websocket.Dial(fmt.Sprintf("ws://%s/server", *address), "", fmt.Sprintf("http://%s/", *address))
    if err != nil {
        fmt.Printf("Dial failed: %s\n", err.Error())
        os.Exit(1)
    }
    incomingMessages := make(chan Message)
    go readClientMessages(ws, incomingMessages)
    i := 0
    for {
        select {
        case <-time.After(time.Duration(2e9)):
            i++
            response := new(Message)
            response.RequestID = i
            response.Command = "Eject the hot dog."
            err = websocket.JSON.Send(ws, response)
            if err != nil {
                fmt.Printf("Send failed: %s\n", err.Error())
                os.Exit(1)
            }
        case message := <-incomingMessages:
            fmt.Println(message)
        }
    }
}

func readClientMessages(ws *websocket.Conn, incomingMessages chan Message) {
    for {
        var message Message
        err := websocket.JSON.Receive(ws, &message)
        if err != nil {
            fmt.Printf("Error: %s\n", err.Error())
            return
        }
        incomingMessages <- message
    }
}

登录后复制

websocket.Dial 用于连接到WebSocket服务器。readClientMessages 函数在一个独立的goroutine中运行，用于接收服务器发送的消息，并将消息发送到 incomingMessages 通道。主循环使用 select 语句来定时发送消息和处理接收到的消息。

消息结构体

为了方便消息的序列化和反序列化，我们可以定义一个消息结构体。

type Message struct {
    RequestID      int
    Command        string
    SomeOtherThing string
    Success        bool
}

登录后复制

这个结构体包含了请求ID、命令、其他信息和成功标志。

注意事项

错误处理： 在WebSocket连接过程中，可能会发生各种错误，例如连接断开、消息格式错误等。需要对这些错误进行适当的处理，例如重新连接、记录日志等。
连接管理： 需要合理管理WebSocket连接，避免连接过多导致服务器资源耗尽。可以使用连接池等技术来管理连接。
并发安全： 如果多个goroutine同时访问WebSocket连接，需要注意并发安全问题。可以使用互斥锁等机制来保护共享资源。
心跳机制： 为了检测WebSocket连接是否仍然有效，可以实现心跳机制。客户端定期向服务器发送心跳消息，服务器收到心跳消息后回复确认消息。如果客户端在一定时间内没有收到服务器的确认消息，则认为连接已经断开，需要重新连接。