概述: 用Go在whisper中监听/订阅消息的教程。

监听/订阅Whisper消息

在本节中，我们将订阅websockets上的Whisper消息。 我们首先需要的是一个通道，它将从whisper/whisperv6包中的Message类型接收Whispe消息。

  1. messages := make(chan *whisperv6.Message)

在我们调用订阅之前，我们首先需要确定消息的过滤标准。 从whisperv6包中初始化一个新的Criteria对象。 由于我们只对定位到我们的消息感兴趣，因此我们将条件对象上的PrivateKeyID属性设置为我们用于加密消息的相同密钥ID。

  1. criteria := whisperv6.Criteria{
  2.   PrivateKeyID: keyID,
  3. }

接下来，我们调用客户端的SubscribeMessages方法，该方法订阅符合给定条件的消息。 HTTP不支持此方法; 仅支持双向连接，例如websockets和IPC。 最后一个参数是我们之前创建的消息通道。

  1. sub, err := client.SubscribeMessages(context.Background(), criteria, messages)
  2. if err != nil {
  3.   log.Fatal(err)
  4. }

现在我们已经订阅了，我们可以使用select语句来读取消息，并处理订阅中的错误。 如果您从上一节回忆起来，消息内容在Payload属性中作为字节切片，我们可以将其转换回人类可读的字符串。

  1. for {
  2.   select {
  3.   case err := <-sub.Err():
  4.     log.Fatal(err)
  5.   case message := <-messages:
  6.     fmt.Printf(string(message.Payload)) // "Hello"
  7.   }
  8. }

查看下面的完整代码，获取完整的栗子。 这就是消息订阅的所有内容。

完整代码

Commands

  1. geth --shh --rpc --ws

whisper_subscribe.go

  1. package main
  3. import (
  4.     "context"
  5.     "fmt"
  6.     "log"
  7.     "os"
  8.     "runtime"
  10.     "github.com/ethereum/go-ethereum/common/hexutil"
  11.     "github.com/ethereum/go-ethereum/whisper/shhclient"
  12.     "github.com/ethereum/go-ethereum/whisper/whisperv6"
  13. )
  15. func main() {
  16.     client, err := shhclient.Dial("ws://127.0.0.1:8546")
  17.     if err != nil {
  18.         log.Fatal(err)
  19.     }
  21.     keyID, err := client.NewKeyPair(context.Background())
  22.     if err != nil {
  23.         log.Fatal(err)
  24.     }
  25.     fmt.Println(keyID) // 0ec5cfe4e215239756054992dbc2e10f011db1cdfc88b9ba6301e2f9ea1b58d2
  27.     messages := make(chan *whisperv6.Message)
  28.     criteria := whisperv6.Criteria{
  29.         PrivateKeyID: keyID,
  30.     }
  31.     sub, err := client.SubscribeMessages(context.Background(), criteria, messages)
  32.     if err != nil {
  33.         log.Fatal(err)
  34.     }
  36.     go func() {
  37.         for {
  38.             select {
  39.             case err := <-sub.Err():
  40.                 log.Fatal(err)
  41.             case message := <-messages:
  42.                 fmt.Printf(string(message.Payload)) // "Hello"
  43.                 os.Exit(0)
  44.             }
  45.         }
  46.     }()
  48.     publicKey, err := client.PublicKey(context.Background(), keyID)
  49.     if err != nil {
  50.         log.Print(err)
  51.     }
  52.     fmt.Println(hexutil.Encode(publicKey)) // 0x04f17356fd52b0d13e5ede84f998d26276f1fc9d08d9e73dcac6ded5f3553405db38c2f257c956f32a0c1fca4c3ff6a38a2c277c1751e59a574aecae26d3bf5d1d
  54.     message := whisperv6.NewMessage{
  55.         Payload:   []byte("Hello"),
  56.         PublicKey: publicKey,
  57.         TTL:       60,
  58.         PowTime:   2,
  59.         PowTarget: 2.5,
  60.     }
  62.     messageHash, err := client.Post(context.Background(), message)
  63.     if err != nil {
  64.         log.Fatal(err)
  65.     }
  66.     fmt.Println(messageHash) // 0xdbfc815d3d122a90d7fb44d1fc6a46f3d76ec752f3f3d04230fe5f1b97d2209a
  68.     runtime.Goexit() // wait for goroutines to finish
  69. }