能使用golang帮我写一个简单的ws聊天服务吗
当然可以,以下是一个简单的WebSocket聊天服务的示例:
package main
import (
"fmt"
"log"
"net/http"
"github.com/gorilla/websocket"
)
var (
upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
clients = make(map[*websocket.Conn]bool)
)
func main() {
http.HandleFunc("/ws", handleWebSocket)
log.Fatal(http.ListenAndServe(":8080", nil))
}
func handleWebSocket(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}
clients[conn] = true
for {
messageType, p, err := conn.ReadMessage()
if err != nil {
delete(clients, conn)
log.Println(err)
return
}
message := string(p)
fmt.Printf("Received message: %s\n", message)
for client := range clients {
if err := client.WriteMessage(messageType, p); err != nil {
log.Println(err)
delete(clients, client)
}
}
}
}
这个示例使用了Gorilla WebSocket库,它提供了更方便的WebSocket处理功能。在这个示例中,我们定义了一个全局变量clients,用于存储连接到服务器的所有客户端连接。在handleWebSocket函数中,我们将新连接添加到clients中,并使用一个无限循环来处理从客户端发送的消息。每当接收到一个消息时,我们将其广播到所有其他客户端。
你可以使用以下命令在本地运行这个示例:
go run main.go
然后在浏览器中打开http://localhost:8080,你就可以测试这个WebSocket聊天服务了。
原文地址: https://www.cveoy.top/t/topic/gT3 著作权归作者所有。请勿转载和采集!