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来触发中断处理逻辑。

相关推荐
默默前行的虫虫5 小时前
解决EMQX WebSocket连接不稳定及优化WS配置提升稳定性?
websocket
Tony Bai6 小时前
再见,丑陋的 container/heap!Go 泛型堆 heap/v2 提案解析
开发语言·后端·golang
念何架构之路8 小时前
Go进阶之panic
开发语言·后端·golang
先跑起来再说8 小时前
Git 入门到实战:一篇搞懂安装、命令、远程仓库与 IDEA 集成
ide·git·后端·elasticsearch·golang·intellij-idea
闲人编程13 小时前
使用FastAPI和WebSocket构建高性能实时聊天系统
websocket·网络协议·网络编程·fastapi·持久化·实时聊天·codecapsule
Tony Bai17 小时前
“Go 2,请不要发生!”:如果 Go 变成了“缝合怪”,你还会爱它吗?
开发语言·后端·golang
灰子学技术1 天前
go response.Body.close()导致连接异常处理
开发语言·后端·golang
源代码•宸1 天前
大厂技术岗面试之谈薪资
经验分享·后端·面试·职场和发展·golang·大厂·职级水平的薪资
青春给了代码1 天前
基于WebSocket实现在线语音(实时+保存)+文字双向传输完整实现
网络·websocket·网络协议
我爱加班、、2 天前
Websocket能携带token过去后端吗
前端·后端·websocket