当前位置：首页 > 文章列表 > Golang > Go教程 > Golang网络编程学习之Websocket实现

Golang网络编程学习之Websocket实现

2023-06-30 12:29:48 0浏览收藏

在IT行业这个发展更新速度很快的行业，只有不停止的学习，才不会被行业所淘汰。如果你是Golang学习者，那么本文《Golang网络编程学习之Websocket实现》就很适合你！本篇内容主要包括##content_title##，希望对大家的知识积累有所帮助，助力实战开发！

Websocket作为一种新兴的网络通信协议，已经被广泛应用于现代的Web应用中。在Golang中，通过使用官方标准库中的websocket包，可以轻松地实现Websocket协议的通信。

本文将基于Golang语言，介绍如何使用websocket包实现Websocket通信。

什么是Websocket

Websocket是一种双向通信协议，通过在客户端和服务器之间建立一个长连接，实现双方之间的实时通信。与HTTP协议不同的是，Websocket协议是基于TCP协议实现的。

Websocket存在于HTTP协议之上，因此Websocket握手的过程与HTTP协议有些相似，但是Websocket握手后，通信双方之间便可以发送任意格式的数据。Websocket协议还支持心跳包、断线重连等功能，使得Websocket可以应用于实时推送、在线聊天等场景。

Golang中的Websocket

在Golang官方标准库中，websocket包提供了用于实现Websocket协议通信的API。

Websocket服务端

以下代码展示了如何使用websocket包实现Websocket服务端：

package main

import (
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

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

func handleWS(conn *websocket.Conn) {
    defer conn.Close()

    for {
        _, msg, err := conn.ReadMessage()
        if err != nil {
            log.Println("read error:", err)
            break
        }
        log.Printf("received: %s
", msg)

        err = conn.WriteMessage(websocket.TextMessage, msg)
        if err != nil {
            log.Println("write error:", err)
            break
        }
    }
}

func main() {
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        log.Println("connection from:", r.RemoteAddr)

        conn, err := upgrader.Upgrade(w, r, nil)
        if err != nil {
            log.Println("upgrade error:", err)
            return
        }

        handleWS(conn)
    })

    log.Println("starting server...")
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("listen error:", err)
    }
}

以上代码实现了一个简单的Websocket服务端，它会接收客户端连接，并在客户端连接成功后进入一个死循环，监听客户端发送过来的信息，并将收到的信息原样发送回客户端。

在实现中，通过http.HandleFunc()函数来指定处理客户端连接的函数handleWS()。在handleWS()函数中，使用websocket包提供的Upgrader进行升级连接，将HTTP连接升级为Websocket连接，进而实现了双向通信。在连接建立后，使用ReadMessage()函数监听客户端发送的信息，使用WriteMessage()函数将信息原样发送回客户端。当连接关闭时，通过defer语句关闭连接。

Websocket客户端

以下代码展示了如何使用websocket包实现Websocket客户端：

package main

import (
    "log"
    "net/url"
    "os"
    "os/signal"
    "time"

    "github.com/gorilla/websocket"
)

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

    u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/"}
    log.Printf("connecting to %s", u.String())

    conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
    if err != nil {
        log.Fatal("dial error:", err)
    }
    defer conn.Close()

    done := make(chan struct{})
    go func() {
        defer close(done)
        for {
            _, msg, err := conn.ReadMessage()
            if err != nil {
                log.Println("read error:", err)
                return
            }
            log.Printf("received: %s", msg)
        }
    }()

    ticker := time.NewTicker(time.Second)
    defer ticker.Stop()

    for {
        select {
        case <-done:
            return
        case t := <-ticker.C:
            err := conn.WriteMessage(websocket.TextMessage, []byte(t.String()))
            if err != nil {
                log.Println("write error:", err)
                return
            }
        case <-interrupt:
            log.Println("interrupt")
            err := conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
            if err != nil {
                log.Println("write close error:", err)
                return
            }
            select {
            case <-done:
            case <-time.After(time.Second):
            }
            return
        }
    }
}

以上代码实现了一个简单的Websocket客户端，它会连接到服务端并定时发送当前时间字符串。在客户端运行过程中，可以收到服务端发送的原样信息。

在实现中，通过websocket.DefaultDialer.Dial()函数建立与服务端的连接，并在连接建立后使用ReadMessage()函数监听服务端发送的信息。同时，使用time.Ticker定时发送信息。客户端还支持中断操作，当收到操作系统中断信号时，首先使用websocket.CloseMessage发送关闭连接消息，其次等待服务端关闭连接或等待1秒后自行关闭连接。