GO-WebSockets客户端实现方法

最新推荐文章于 2024-02-27 15:00:24 发布

区块爱好者88

最新推荐文章于 2024-02-27 15:00:24 发布

阅读量880

点赞数

分类专栏： golang 文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/xuankucom/article/details/121597406

版权

golang 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

使用到的包：github.com/gorilla/websocket

package main

import (
    "fmt"
    "github.com/gorilla/websocket"
    "log"
    "net/url"
    "sync"
    "time"
)

type websocketClientManager struct {
    conn        *websocket.Conn
    addr        *string
    path        string
    sendMsgChan chan string
    recvMsgChan chan string
    isAlive     bool
    timeout     int
}

// 构造函数
func NewWsClientManager(addrIp, addrPort, path string, timeout int) *websocketClientManager {
    addrString := addrIp + ":" + addrPort
    var sendChan = make(chan string, 10)
    var recvChan = make(chan string, 10)
    var conn *websocket.Conn
    return &websocketClientManager{
        addr:        &addrString,
        path:        path,
        conn:        conn,
        sendMsgChan: sendChan,
        recvMsgChan: recvChan,
        isAlive:     false,
        timeout:     timeout,
    }
}

// 链接服务端
func (wsc *websocketClientManager) dail() {
    var err error
    u := url.URL{Scheme: "ws", Host: *wsc.addr, Path: wsc.path}
    log.Printf("connecting to %s", u.String())
    wsc.conn, _, err = websocket.DefaultDialer.Dial(u.String(), nil)
    if err != nil {
        fmt.Println(err)
        return

    }
    wsc.isAlive = true
    log.Printf("connecting to %s 链接成功！！！", u.String())

}

// 发送消息
func (wsc *websocketClientManager) sendMsgThread() {
    go func() {
        for {
            msg := <-wsc.sendMsgChan
            err := wsc.conn.WriteMessage(websocket.TextMessage, []byte(msg))
            if err != nil {
                log.Println("write:", err)
                continue
            }
        }
    }()
}

// 读取消息
func (wsc *websocketClientManager) readMsgThread() {
    go func() {
        for {
            if wsc.conn != nil {
                _, message, err := wsc.conn.ReadMessage()
                if err != nil {
                    log.Println("read:", err)
                    wsc.isAlive = false
                    // 出现错误，退出读取，尝试重连
                    break
                }
                log.Printf("recv: %s", message)
                // 需要读取数据，不然会阻塞
                wsc.recvMsgChan <- string(message)
            }

        }
    }()
}

// 开启服务并重连
func (wsc *websocketClientManager) start() {
    for {
        if wsc.isAlive == false {
            wsc.dail()
            wsc.sendMsgThread()
            wsc.readMsgThread()
        }
        time.Sleep(time.Second * time.Duration(wsc.timeout))
    }
}

func main() {
    wsc := NewWsClientManager("192.168.12.15", "10086", "/v1", 10)
    wsc.start()
    var w1 sync.WaitGroup
    w1.Add(1)
    w1.Wait()
}