如何解决Golang 结束 (Binance) 网络服务流使用 goroutine
我正在将 Binance API 集成到现有系统中,虽然大多数部分是直截了当的,但数据流 API 触及了我对 go-routines 的有限理解。我不相信用于 Binance 的 golang SDK 有什么特别之处,但基本上我只需要两个函数,一个启动数据流并使用作为参数给出的事件处理程序处理事件,第二个结束数据流没有实际关闭客户端,因为它会关闭所有其他连接。在之前的一个项目中,有两种消息类型,但币安 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,errHandler)
}
我不是很清楚,主要是因为理解不完全,真的是我如何停止流。我认为返回的 stopC 通道可用于以某种方式发出类似于系统级别的 sigterm 的结束信号,然后流应该结束。
比如说,我有一个以符号作为参数的 stopDataStream 函数
func stopDataStream(symbol){
}
假设我为五个符号启动了 5 个数据流,现在我只想停止其中一个流。这就引出了一个问题:
-
我如何跟踪所有这些 stopC 频道?
-
我可以使用以符号为键的集合,拉出 stopC 通道,然后发出一个信号来结束该数据流吗?
-
我实际上如何从 stop 函数写入 stopC 通道?
再说一次,我不认为这特别难,只是我还没有从文档中弄清楚,所以任何帮助都将不胜感激。
谢谢
解决方法
(答案最初由@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,errHandler)
time.Sleep(3 * time.Second)
println("Stop stream")
stopDataStream(symbol)
time.Sleep(1 * time.Second)
}
就是这样。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。