TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-04 07:50:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
GoEdgeLab
2022-11-17 10:01:07 +08:00
parent 0fa7d43cba
commit 1e1f81f2a0
1 changed files with 8 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
internal/rpc/services/service_node_stream.go
View File

@@ -112,24 +112,24 @@ func (this *NodeService) NodeStream(server pb.NodeService_NodeStreamServer) erro
var tx = this.NullTx() var tx = this.NullTx()
// 标记为活跃状态 // 是否发送恢复通知
oldIsActive, err := models.SharedNodeDAO.FindNodeActive(tx, nodeId) oldIsActive, err := models.SharedNodeDAO.FindNodeActive(tx, nodeId)
if err != nil { if err != nil {
return err return err
} }
if !oldIsActive { if !oldIsActive {
inactiveNotifiedAt, err := models.SharedNodeDAO.FindNodeInactiveNotifiedAt(tx, nodeId) inactiveNotifiedAt, err := models.SharedNodeDAO.FindNodeInactiveNotifiedAt(tx, nodeId)
if err != nil { if err != nil {
return err return err
} }
if inactiveNotifiedAt > 0 {
// 设置为活跃
err = models.SharedNodeDAO.UpdateNodeActive(tx, nodeId, true)
if err != nil {
return err
}
// 设置为活跃
err = models.SharedNodeDAO.UpdateNodeActive(tx, nodeId, true)
if err != nil {
return err
}
if inactiveNotifiedAt > 0 {
// 发送恢复消息 // 发送恢复消息
clusterId, err := models.SharedNodeDAO.FindNodeClusterId(tx, nodeId) clusterId, err := models.SharedNodeDAO.FindNodeClusterId(tx, nodeId)
if err != nil { if err != nil {
@@ -145,12 +145,6 @@ func (this *NodeService) NodeStream(server pb.NodeService_NodeStreamServer) erro
if err != nil { if err != nil {
return err return err
} }
} else {
// 设置为活跃
err = models.SharedNodeDAO.UpdateNodeActive(tx, nodeId, true)
if err != nil {
return err
}
} }
} }