public/PandaX
1
0
Fork 0
mirror of https://gitee.com/XM-GO/PandaX.git synced 2026-04-23 02:48:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

【fix】设备数据处理并发控制修复

Browse Source
This commit is contained in:
PandaX
2023-10-09 11:15:23 +08:00
parent fa6797141a
commit eb83923849
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
iothub/hook_message_work/hook_message_work.go
View File

@@ -34,9 +34,15 @@ func (s *HookService) handleOne(msg *netbase.DeviceEventInfo) {
if s.Ch != nil { // 用于并发限制 if s.Ch != nil { // 用于并发限制
s.Ch <- struct{}{} s.Ch <- struct{}{}
} }
// 用于做优雅关闭， 主要作用是，程序关闭，将队列中的消息处理完成后在关闭，
s.Wg.Add(1) s.Wg.Add(1)
go func() { go func() {
defer s.Wg.Done() defer func() {
s.Wg.Done()
if s.Ch != nil { // 用于并发限制
<-s.Ch
}
}()
switch msg.Type { switch msg.Type {
case message.RowMes, message.AttributesMes, message.TelemetryMes, message.RpcRequestMes: case message.RowMes, message.AttributesMes, message.TelemetryMes, message.RpcRequestMes:
msgVals := make(map[string]interface{}) msgVals := make(map[string]interface{})