diff --git a/internal/rpc/services/service_node_stream.go b/internal/rpc/services/service_node_stream.go index 9133b91b..189b281c 100644 --- a/internal/rpc/services/service_node_stream.go +++ b/internal/rpc/services/service_node_stream.go @@ -124,6 +124,7 @@ func (this *NodeService) NodeStream(server pb.NodeService_NodeStreamServer) erro return err } if inactiveNotifiedAt > 0 { + // 设置为活跃 err = models.SharedNodeDAO.UpdateNodeActive(tx, nodeId, true) if err != nil { return err @@ -144,6 +145,12 @@ func (this *NodeService) NodeStream(server pb.NodeService_NodeStreamServer) erro if err != nil { return err } + } else { + // 设置为活跃 + err = models.SharedNodeDAO.UpdateNodeActive(tx, nodeId, true) + if err != nil { + return err + } } }