当前位置：首页 > 文章列表 > Golang > Go教程 > Go语言实战：使用Websocket构建协作应用

Go语言实战：使用Websocket构建协作应用

2023-06-21 09:27:00 0浏览收藏

本篇文章向大家介绍《Go语言实战：使用Websocket构建协作应用》，主要包括，具有一定的参考价值，需要的朋友可以参考一下。

在现代Web应用程序中，实时协作是具有重要意义的。由于HTTP通常是单向的，以点对点方式推送数据不容易实现。这是Websocket涌现出的背景，它允许以全双工的方式使用TCP套接字进行通信，使得Web应用程序可以实现实时通信。

在本文中，我们将探讨如何使用Go语言编写一个简单的Websocket应用程序来实现实时协作。我们将建立一个包含聊天功能的简单Web应用程序，用户可以在其中通过Websocket进行实时通信。

什么是Websocket?

在Websocket出现之前，HTTP流和轮询等技术都曾被用于实现实时通信。然而，这些技术存在着各种限制，例如低效、不可靠、容易中断等。Websocket之所以能够重新定义实时通信是因为它提供了以下功能：

完全双向通信：服务器和客户端可以同时发送和接收消息。
高效性：Websocket建立在TCP套接字之上，因此可以通过单个连接进行双向通信，避免了HTTP中的频繁请求/响应周期。
可过滤性：Websocket协议使用HTTP端口（通常是80或443），因此可以穿过已有的防火墙和代理服务器。
可扩展性：通过自定义消息类型，可以方便地扩展Websocket应用程序以满足不同的需求。

实现聊天应用程序

在本文中，我们将构建一个简单的聊天应用程序。用户可以在其中通过Websocket进行实时通信。

项目结构

我们将使用Gorilla Websocket库来实现Websocket服务器。以下是该项目的结构：

.
├── main.go
├── public
│   ├── css
│   │   └── style.css
│   └── js
│       └── main.js
└── templates
    └── index.html

其中，public文件夹包含了我们Web应用程序的前端代码，包括CSS和Javascript。templates文件夹包含我们的HTML模板。

前端代码

在public/js/main.js中，我们实现了以下功能：

连接到Websocket服务器
发送消息
接收消息

以下是具体代码实现：

let socket = new WebSocket("ws://localhost:8000/ws");

// 连接成功
socket.onopen = function (event) {
  console.log("WebSocket connected");
};

// 发送消息
document.getElementById("send-button").onclick = function () {
  let message = document.getElementById("message-input").value;
  socket.send(message);
};

// 接收消息
socket.onmessage = function (event) {
  let message = event.data;
  let messageBox = document.getElementById("message-box");
  messageBox.innerHTML += "<p>" + message + "</p>";
};

后端代码

在main.go中，我们实现了以下功能：

创建一个Websocket服务器
处理Websocket连接
接收消息
广播消息

以下是具体代码实现：

package main

import (
    "fmt"
    "net/http"

    "github.com/gorilla/websocket"
)

var (
    upgrader = websocket.Upgrader{
        ReadBufferSize:  1024,
        WriteBufferSize: 1024,

        // 允许跨域请求
        CheckOrigin: func(r *http.Request) bool {
            return true
        },
    }

    // 维护所有Websocket连接的全局变量
    connections = make(map[*websocket.Conn]bool)
)

func main() {
    http.HandleFunc("/", indexHandler)
    http.HandleFunc("/ws", wsHandler)

    fmt.Println("Starting server...")
    http.ListenAndServe(":8000", nil)
}

func indexHandler(w http.ResponseWriter, r *http.Request) {
    http.ServeFile(w, r, "templates/index.html")
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
    // 升级HTTP连接为Websocket连接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        fmt.Println("Failed to upgrade connection:", err)
        return
    }

    // 将连接添加到connections中
    connections[conn] = true

    // 关闭连接时从connections中删除
    defer func() {
        delete(connections, conn)
        conn.Close()
    }()

    for {
        // 读取消息
        _, message, err := conn.ReadMessage()
        if err != nil {
            fmt.Println("Failed to read message:", err)
            break
        }

        // 打印消息并广播
        fmt.Printf("Received message: %s
", string(message))
        broadcast(string(message))
    }
}

func broadcast(message string) {
    for conn := range connections {
        err := conn.WriteMessage(websocket.TextMessage, []byte(message))
        if err != nil {
            fmt.Println("Failed to send message:", err)
            continue
        }
    }
}

HTML模板

在templates/index.html中，我们实现了以下功能：

显示聊天消息
允许用户发送消息

以下是具体代码实现：

<!DOCTYPE html>
<html>

<head>
  <meta charset="UTF-8">
  <title>Chat App</title>
  <link rel="stylesheet" href="/public/css/style.css">
</head>

<body>
  <div id="message-box">
    <p>Welcome to the chat!</p>
  </div>
  <div id="input-box">
    <input type="text" id="message-input" placeholder="Type a message...">
    <button id="send-button">Send</button>
  </div>
  <script src="/public/js/main.js"></script>
</body>

</html>

运行应用程序

现在，我们可以在终端中进入项目目录并运行以下命令来启动Web应用程序：

go run main.go

现在，通过在浏览器中访问http://localhost:8000，我们就可以打开Web应用程序并开始实时通信了！

总结

在本文中，我们探讨了如何使用Go语言编写一个简单的Websocket应用程序。我们使用Gorilla Websocket库来实现Websocket服务器，实现了一个包含聊天功能的简单Web应用程序，用户可以在其中通过Websocket进行实时通信。通过本文的学习，读者将获得构建实时协作Web应用程序的入门知识。

理论要掌握，实操不能落！以上关于《Go语言实战：使用Websocket构建协作应用》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！

Go语言 websocket 协作应用