当前位置：首页 > 文章列表 > Golang > Go教程 > 如何使用Golang实现Websocket通信

如何使用Golang实现Websocket通信

2023-06-28 13:05:41 0浏览收藏

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《如何使用Golang实现Websocket通信》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

随着互联网的发展，Websocket成为了一种非常流行的通信协议，可以帮助我们实现实时通信等应用。在这篇文章中，我们将着重介绍如何使用Golang实现Websocket通信。Golang在处理并发方面非常出色，因此使用它来处理Websocket通信非常方便。

一、了解Websocket

Websocket是一种协议，它允许在两个端点之间建立持久连接，以便进行双向通信。它是在HTTP协议的基础上构建的，在建立连接后，客户端和服务端可以通过发送消息进行通信。使用Websocket可以实现实时通信，比如在线聊天室、在线游戏等应用。

二、安装Golang

在开始使用Golang实现Websocket通信之前，我们需要先安装Golang。可以在官网https://golang.org/下载安装包，安装后添加环境变量。

三、创建Websocket服务器

接下来，我们需要创建一个Websocket服务器。首先，创建一个名为websocket.go的文件，并添加以下代码：

package main

import (
    "fmt"
    "net/http"

    "github.com/gorilla/websocket"
)

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

func websocketHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        fmt.Println(err)
        return
    }
    for {
        messageType, p, err := conn.ReadMessage()
        if err != nil {
            fmt.Println(err)
            return
        }
        fmt.Printf("Received message type: %d, message: %s
", messageType, string(p))
        err = conn.WriteMessage(messageType, p)
        if err != nil {
            fmt.Println(err)
            return
        }
    }
}

func main() {
    http.HandleFunc("/websocket", websocketHandler)
    fmt.Println("Websocket server started on port 8080")
    http.ListenAndServe(":8080", nil)
}

在这段代码中，我们创建了一个名为upgrader的变量，它会升级HTTP连接为Websocket连接。websocketHandler函数则是真正的Websocket服务器实现，它首先升级HTTP连接为Websocket连接，然后通过ReadMessage和WriteMessage分别读写消息。最后，在main函数中，我们使用http.HandleFunc函数将Websocket路由到websocketHandler函数，并使用http.ListenAndServe来启动服务器。

四、创建Websocket客户端

现在我们已经成功创建了一个Websocket服务器，接下来我们需要创建一个Websocket客户端来与服务器进行通信。在创建客户端之前，我们需要安装一个Golang的Websocket库，这里我们使用Gorilla Websocket库。

go get github.com/gorilla/websocket

安装完成后，我们可以创建一个名为websocket_client.go的文件，并添加以下代码：

package main

import (
    "fmt"
    "log"
    "os"
    "os/signal"

    "github.com/gorilla/websocket"
)

func main() {
    interrupt := make(chan os.Signal, 1)
    signal.Notify(interrupt, os.Interrupt)

    url := "ws://localhost:8080/websocket"
    conn, _, err := websocket.DefaultDialer.Dial(url, nil)
    if err != nil {
        log.Fatal("dial:", err)
    }
    defer conn.Close()

    done := make(chan struct{})
    go func() {
        defer close(done)
        for {
            messageType, message, err := conn.ReadMessage()
            if err != nil {
                log.Println("read:", err)
                return
            }
            log.Printf("Received message type: %d, message: %s", messageType, string(message))
        }
    }()

    for {
        select {
        case <-done:
            return
        case <-interrupt:
            fmt.Println("interrupt")
            err := conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
            if err != nil {
                log.Println("write close:", err)
                return
            }
            select {
            case <-done:
            }
            return
        }
    }
}

在这段代码中，我们首先使用websocket.DefaultDialer.Dial函数连接到Websocket服务器。接着，我们启动一个goroutine来等待接收从服务器发送过来的消息。最后，我们使用一个select语句来监听从服务器发送过来的消息和关闭客户端程序的信号。

五、运行程序

现在，我们已经完成了Websocket服务器和客户端的创建，接下来我们需要运行程序。首先，在命令行中进入到包含websocket.go和websocket_client.go的目录，并使用以下命令运行Websocket服务器：