实现监控节点在线状态

This commit is contained in:
GoEdgeLab
2020-10-25 18:26:46 +08:00
parent 89b7ee07d4
commit 473ad47a7f
7 changed files with 163 additions and 0 deletions

View File

@@ -5,6 +5,7 @@ import (
"encoding/json"
"fmt"
"github.com/TeaOSLab/EdgeAPI/internal/configs"
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
"github.com/TeaOSLab/EdgeAPI/internal/errors"
rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
"github.com/TeaOSLab/EdgeCommon/pkg/messageconfigs"
@@ -135,6 +136,12 @@ func (this *NodeService) NodeStream(server pb.NodeService_NodeStreamServer) erro
for {
req, err := server.Recv()
if err != nil {
// 修改节点状态
err1 := models.SharedNodeDAO.UpdateNodeIsActive(nodeId, false)
if err1 != nil {
logs.Println(err1.Error())
}
return err
}