如何使用Go语言开发Websocket聊天室-uusu优素-乐高,模型,3d打印,编程

如何使用Go语言开发Websocket聊天室

Websocket是一种实时通信协议，通过建立一次连接，可以在服务器和客户端之间进行双向通信。在开发聊天室时，Websocket是一个非常好的选择，因为它可以实现实时消息交流，并且能够提供高效的性能。本文将介绍如何使用Go语言开发一个简单的Websocket聊天室，并提供一些具体的代码示例。

一、准备工作

1.安装Go语言环境

在开始之前，首先需要安装Go语言环境。Go语言的官方网站（https://golang.org/）上提供了各个平台的安装包，可以根据自己的操作系统下载并安装。

2.安装websocket库

在Go语言中，使用“github.com/gorilla/websocket”包提供了WebSocket相关的功能。可以通过以下命令来安装该包：

go get github.com/gorilla/websocket

二、创建聊天室服务器

首先，我们需要创建一个简单的服务器，用于处理Websocket连接的建立和消息的收发。

package main
import (
"fmt"
"log"
"net/http"
"github.com/gorilla/websocket"
)
var clients = make(map[*websocket.Conn]bool) // 存储连接的客户端
var broadcast = make(chan Message)           // 消息广播通道
// 定义消息结构体
type Message struct {
Username string `json:"username"`
Message  string `json:"message"`
}
func main() {
http.HandleFunc("/ws", handleConnections)
go handleMessages()
// 启动服务器
log.Println("服务器启动，监听端口：8080")
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal("ListenAndServe: ", err)
}
}
func handleConnections(w http.ResponseWriter, r *http.Request) {
// 升级HTTP连接为Websocket连接
ws, err := websocket.Upgrade(w, r, nil, 1024, 1024)
if err != nil {
log.Fatal(err)
}
// 关闭连接
defer ws.Close()
// 将客户端连接添加到clients映射中
clients[ws] = true
for {
var msg Message
// 读取客户端发送的消息
err := ws.ReadJSON(&msg)
if err != nil {
log.Printf("读取消息错误：%v", err)
delete(clients, ws)
break
}
// 将接收到的消息放入广播通道中
broadcast <- msg
}
}
func handleMessages() {
for {
// 从广播通道中读取消息
msg := <-broadcast
// 遍历所有连接的客户端，将消息发送给每个客户端
for client := range clients {
err := client.WriteJSON(msg)
if err != nil {
log.Printf("发送消息错误：%v", err)
client.Close()
delete(clients, client)
}
}
}
}

以上代码创建了一个简单的Websocket聊天室服务器。在handleConnections函数中，我们将接收到的消息放入广播通道中。handleMessages函数从广播通道中读取消息，并将消息发送给所有连接的客户端。

三、创建前端界面

接下来，我们需要创建一个前端界面，用于用户输入用户名和消息，并实时显示聊天内容。

<!doctype html>
<html>
<head>
<title>Websocket 聊天室</title>
<style>
#message-box {
height: 500px;
overflow-y: scroll;
}
</style>
</head>
<body>
<h1>Websocket 聊天室</h1>
<div id="message-box"></div>
<form id="message-form" onsubmit="sendMessage(event)">
<input type="text" id="username" placeholder="请输入用户名" required>
<br />
<input type="text" id="message" placeholder="请输入消息" required>
<br />
<button type="submit">发送</button>
</form>
<script>
const socket = new WebSocket("ws://localhost:8080/ws");
socket.onmessage = function (event) {
const data = JSON.parse(event.data);
const messageBox = document.getElementById("message-box");
const messageElement = document.createElement("p");
messageElement.innerText = data.username + ": " + data.message;
messageBox.appendChild(messageElement);
};
function sendMessage(event) {
event.preventDefault();
const username = document.getElementById("username").value;
const message = document.getElementById("message").value;
const data = {
username: username,
message: message
};
socket.send(JSON.stringify(data));
document.getElementById("message").value = "";
}
</script>
</body>
</html>

以上代码创建了一个简单的HTML界面，包含一个用户名输入框、一个消息输入框和一个发送按钮。通过socket.onmessage处理接收到的消息，并将消息显示在消息盒子中。通过socket.send发送输入的消息。

四、运行程序

在终端中进入项目目录，执行以下命令启动服务器：