在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: %sn", 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: %sn", 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: %sn", 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: %sn", 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: %sn", 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: %sn", 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应用的稳定性和可靠性。
评论（已关闭）
评论已关闭