如何使用Go语言开发Websocket聊天室-编程学习网

如何使用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发送输入的消息。

四、运行程序

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