在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服务器，监听指定的地址和端口。
客户端实现
客户端需要连接到服务器，并持续发送和接收消息。
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连接是否仍然有效，可以实现心跳机制。客户端定期向服务器发送心跳消息，服务器收到心跳消息后回复确认消息。如果客户端在一定时间内没有收到服务器的确认消息，则认为连接已经断开，需要重新连接。
总结
通过为每个WebSocket连接创建一个独立的goroutine，可以有效地解决EOF错误问题，并确保连接的持续性。同时，需要注意错误处理、连接管理和并发安全问题，以保证WebSocket应用的稳定性和可靠性。
