使用 Go 例程的 Golang 结局（币安）Web 服务流

Question

我正在将 Binance API 集成到现有系统中，虽然大部分部分都很简单，但数据流 API 触及了我对 go 例程的有限理解。我不认为 Binance 的 golang SDK 有什么特别之处，但本质上我只需要两个函数，一个函数启动数据流并使用作为参数给出的事件处理程序处理事件，第二个函数结束数据流无需实际关闭客户端，因为它会关闭所有其他连接。在之前的项目中，有两种消息类型，但 binance SDK 使用了一种返回两个 go 通道的实现，一个用于错误，另一个（我从名称上看）用于停止数据流。

我为启动数据流编写的代码如下所示：

func startDataStream(symbol, interval string, wsKlineHandler futures.WsKlineHandler, errHandler futures.ErrHandler) (err error){

    doneC, stopC, err := futures.WsKlineServe(symbol, interval, wsKlineHandler, errHandler)
    if err != nil {
        fmt.Println(err)
        return err
    }

    return nil
}

这按预期工作并传输数据。一个简单的测试就验证了这一点：

func runWSDataTest() {
    symbol := "BTCUSDT"
    interval := "15m"
    errHandler := func(err error) {fmt.Println(err)}

    wsKlineHandler := func(event *futures.WsKlineEvent) {fmt.Println(event)}

    _ = startDataStream(symbol, interval, wsKlineHandler, errHandler)
}

我不太清楚，主要是由于理解不完全，实际上是如何停止流。我认为返回的 stopC 通道可用于以某种方式发出类似于系统级别上的 sigterm 的结束信号，然后流应该结束。

比如说，我有一个 stopDataStream 函数，它接受一个符号作为参数

func stopDataStream(symbol){

}

假设我为 5 个符号启动 5 个数据流，现在我只想停止其中一个数据流。这就引出了一个问题：

1. 我如何跟踪所有这些 stopC 频道？

2. 我可以使用带有符号的集合，拉动 stopC 通道，然后发出信号来结束该数据流吗？

3. 如何从 stop 函数实际写入 stopC 通道？

再次，我认为这不是特别难，只是我还无法从文档中弄清楚，所以任何帮助将不胜感激。

谢谢你

Answer 1

（答案最初由@Marvin.Hansen撰写）

事实证明，只需保存并关闭频道即可解决所有问题。我真的很惊讶这是多么容易，但这里是更新函数的代码：

func startDataStream(symbol, interval string, wsKlineHandler futures.WsKlineHandler, errHandler futures.ErrHandler) (err error) {

    _, stopC, err := futures.WsKlineServe(symbol, interval, wsKlineHandler, errHandler)
    if err != nil {
        fmt.Println(err)
        return err
    }
    // just save the stop channel 
    chanMap[symbol] = stopC
    return nil
}

然后，停止功能真的变得微不足道：

func stopDataStream(symbol string) {
    stopC := chanMap[symbol]  // load the stop channel for the symbol
    close(stopC) // just close it. 
}

最后，全部测试一下：

var (
    chanMap map[string]chan struct{}
)

func runWSDataTest() {
    chanMap = make(map[string]chan struct{})

    symbol := "BTCUSDT"
    interval := "15m"
    errHandler := func(err error) { fmt.Println(err) }
    wsKlineHandler := getKLineHandler()

    println("Start stream")
    _ = startDataStream(symbol, interval, wsKlineHandler, errHandler)

    time.Sleep(3 * time.Second)

    println("Stop stream")
    stopDataStream(symbol)

    time.Sleep(1 * time.Second)
}

就是这样。

Answer 2

您可以在那里看到工作示例您可以在那里看到工作示例https://github.com/AndreyMashukov/go-crypto-bot/tree/main

这是在 Golang 上经过充分测试的币安加密 go 机器人实现