修复添加服务器死锁

This commit is contained in:
奶爸 committed 2019-12-11 14:41:12 +08:00
1 parent 3ede8b18a9
commit 5a8750b065
2 files changed
+3 -3

No files matched your search

+1 -1
View File
@@ -36,10 +36,10 @@ func (s *NezhaHandler) Heartbeat(r *pb.Beat, stream pb.NezhaService_HeartbeatSer
return err
}
dao.ServerLock.RLock()
defer dao.ServerLock.RUnlock()
closeCh := make(chan error)
dao.ServerList[clientID].StreamClose = closeCh
dao.ServerList[clientID].Stream = stream
dao.ServerLock.RUnlock()
select {
case err = <-closeCh:
return err