在Go语言中,你可以使用`net/http`和`gorilla/websocket`包来实现一个WebSocket服务器。下面是一个简单的示例代码:
```go
package main
import (
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
func echo(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}
defer conn.Close()
for {
// 读取客户端发送的消息
_, message, err := conn.ReadMessage()
if err != nil {
log.Println(err)
break
}
// 将消息原封不动地返回给客户端
err = conn.WriteMessage(websocket.TextMessage, message)
if err != nil {
log.Println(err)
break
}
}
}
func main() {
http.HandleFunc("/echo", echo)
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal(err)
}
}
```
这段代码创建了一个基本的WebSocket服务器。它使用`gorilla/websocket`包来处理WebSocket连接。当有客户端连接到`/echo`路径时,服务器会升级该连接,并将消息原样返回给客户端。
请注意,这只是一个简单的示例,你可能需要根据自己的需求进行修改和扩展。
本网转载内容版权归原作者和授权发表网站所有,仅供学习交流之用,如有涉及版权问题,请通知我们尽快处理。