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

本篇文章向大家介绍《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中，我们实现了以下功能：

1. 连接到Websocket服务器
2. 发送消息
3. 接收消息

以下是具体代码实现：

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 += "
" + message + "
";
};

后端代码

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

1. 创建一个Websocket服务器
2. 处理Websocket连接
3. 接收消息
4. 广播消息

以下是具体代码实现：

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中，我们实现了以下功能：

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

以下是具体代码实现：

  
  
  



  

    
Welcome to the chat!
  
  

    
    Send
  
  

运行应用程序

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

go run main.go

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

总结

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

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