go语言使用websocket发送一条消息A,持续接收返回的消息

在Go语言中实现一个WebSocket客户端,可以使用`gorilla/websocket`这个非常流行的库来处理WebSocket连接。下面是一个简单的示例,展示了如何创建一个WebSocket客户端,向服务器发送消息"A",并持续接收来自服务器的响应。

首先,确保你已经安装了`gorilla/websocket`包。如果没有安装,可以通过以下命令安装:

```sh

go get -u github.com/gorilla/websocket

```

然后,你可以使用下面的代码作为你的WebSocket客户端脚本:

它只发送一次消息"A",然后持续接收并打印服务器返回的消息,直到接收到特定数量的消息或遇到错误为止。

```go

package main

import (

"fmt"

"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: "example.com:8080", Path: "/ws"} // 修改为你的WebSocket服务器地址

log.Printf("connecting to %s", u.String())

c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)

if err != nil {

log.Fatal("dial:", err)

}

defer c.Close()

// 发送一次消息"A"

err = c.WriteMessage(websocket.TextMessage, []byte("A"))

if err != nil {

log.Fatal("write:", err)

}

log.Println("sent: A")

// 设置接收消息的数量

n := 10 // 假设你想接收10条消息

// 开始接收消息

for i := 0; i < n; i++ {

_, message, err := c.ReadMessage()

if err != nil {

log.Println("read:", err)

break

}

log.Printf("recv: %s", message)

}

// 处理中断信号

select {

case <-interrupt:

log.Println("interrupt")

// 清洁关闭连接

err := c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))

if err != nil {

log.Println("write close:", err)

}

select {

case <-time.After(time.Second):

}

}

}

```

代码解释

  1. **连接到WebSocket服务器**:使用`websocket.DefaultDialer.Dial`方法连接到指定的WebSocket服务器。

  2. **发送一次消息**:在连接成功后,发送一次消息"A"。

  3. **接收多条消息**:设置一个循环,接收指定数量的消息(在这个例子中是10条)。每次接收到消息后,打印消息内容。

  4. **处理中断信号**:监听操作系统的中断信号(如Ctrl+C),以便优雅地关闭WebSocket连接。

运行说明

  • 将`example.com:8080`替换为你的WebSocket服务器的实际地址。

  • 调整变量`n`的值,以指定你希望接收的消息数量。

运行此程序后,它将发送一次消息"A",然后持续接收并打印服务器返回的消息,直到接收到指定数量的消息或遇到错误。如果需要停止程序,可以按Ctrl+C来触发中断处理逻辑。

相关推荐
加油,旭杏6 小时前
【go语言】接口
开发语言·后端·golang
weisian1518 小时前
消息队列篇--通信协议篇--STOMP(STOMP特点、格式及示例,WebSocket上使用STOMP,消息队列上使用STOMP等)
websocket·网络协议
清北_11 小时前
万科000002
golang
兮动人12 小时前
Golang 执行流程分析
开发语言·后端·golang·golang 执行流程分析
zhuyasen13 小时前
多维度详细比较 kratos、go-zero、goframe、sponge 框架
后端·http·微服务·rpc·golang
猫九森13 小时前
golang面试题
面试·golang
特立独行的猫a16 小时前
Golang 应用的 Docker 部署方式介绍及使用详解
开发语言·docker·golang
慕璃嫣18 小时前
Clojure语言的系统运维
开发语言·后端·golang
YGGP1 天前
【Go语言圣经】第五节:函数
golang
乔冠宇1 天前
Java中初步使用websocket(springBoot版本)
java·网络·websocket·网络协议