构建一个实时聊天应用是一个非常有趣的项目，尤其是在使用WebSocket协议时。WebSocket允许客户端和服务器之间进行全双工通信，使得消息可以即时传递，非常适合聊天应用等需要实时交互的场景。
在本文中，我们将探讨如何使用Go语言构建一个基于WebSocket的实时聊天应用。我们会从基础概念开始，逐步深入到代码实现，并讨论一些常见的优化和扩展方法。
WebSocket是一种网络通信协议，它提供了一个持久化的连接，使得客户端和服务器可以在连接期间随时发送数据。与传统的HTTP请求-响应模型不同，WebSocket允许服务器主动向客户端推送数据。
Go语言本身并不直接内置对WebSocket的支持，但通过第三方库如gorilla/websocket，我们可以轻松地处理WebSocket连接。
首先，我们需要安装gorilla/websocket库：
go get github.com/gorilla/websocket
我们的项目将包括以下几个部分：
首先，我们创建一个HTTP服务器来处理WebSocket升级请求，并维持客户端连接。
package main
import (
"fmt"
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
}
func handleConnections(w http.ResponseWriter, r *http.Request) {
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Fatal(err)
}
defer ws.Close()
for {
var msg string
err := ws.ReadMessage(&msgType, &msg)
if err != nil {
log.Printf("error: %v", err)
break
}
log.Printf("received: %v", msg)
// 广播消息逻辑待补充
}
}
func main() {
http.HandleFunc("/ws", handleConnections)
fmt.Println("Server started on localhost:8080")
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal("ListenAndServe: ", err)
}
}
为了让每个客户端都能接收到其他客户端发送的消息，我们需要实现一个广播系统。
type Client struct {
ws *websocket.Conn
}
type Message struct {
Message string `json:"message"`
}
var clients = make(map[*Client]bool)
var broadcast = make(chan Message)
func handleMessages() {
for {
msg := <-broadcast
for client := range clients {
err := client.ws.WriteJSON(msg)
if err != nil {
log.Printf("error: %v", err)
client.ws.Close()
delete(clients, client)
}
}
}
}
func handleConnections(w http.ResponseWriter, r *http.Request) {
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Fatal(err)
}
client := &Client{ws: ws}
clients[client] = true
go func() {
for {
var msg Message
err := ws.ReadJSON(&msg)
if err != nil {
log.Printf("error: %v", err)
delete(clients, client)
break
}
broadcast <- msg
}
}()
}
最后，我们需要一个简单的HTML页面来测试我们的聊天应用。
<!DOCTYPE html>
<html>
<head>
<title>WebSocket Chat</title>
</head>
<body>
<h1>WebSocket Chat</h1>
<ul id="messages"></ul>
<form id="form" action="">
<input id="message" autocomplete="off"/>
<button>Send</button>
</form>
<script>
var ws = new WebSocket("ws://localhost:8080/ws");
ws.onmessage = function(event) {
var messages = document.getElementById('messages')
var message = document.createElement('li')
var content = document.createTextNode(event.data)
message.appendChild(content)
messages.appendChild(message)
};
document.getElementById('form').onsubmit = function() {
var msg = document.getElementById('message').value;
ws.send(JSON.stringify({message: msg}));
return false;
};
</script>
</body>
</html>