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