git-sync/crawlab
1
0
Fork 0
mirror of https://github.com/crawlab-team/crawlab.git synced 2026-01-31 18:10:50 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: support notification for node

Browse Source
This commit is contained in:
Marvin Zhang
2024-07-24 17:00:35 +08:00
parent 923921c17a
commit 7b1fa48fd9
5 changed files with 92 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
core/grpc/server/node_server_v2.go
View File

@@ -10,6 +10,8 @@ import (
"github.com/crawlab-team/crawlab/core/models/models/v2"
"github.com/crawlab-team/crawlab/core/models/service"
nodeconfig "github.com/crawlab-team/crawlab/core/node/config"
"github.com/crawlab-team/crawlab/core/notification"
"github.com/crawlab-team/crawlab/core/utils"
"github.com/crawlab-team/crawlab/grpc"
errors2 "github.com/pkg/errors"
"go.mongodb.org/mongo-driver/bson"
@@ -93,6 +95,7 @@ func (svr NodeServerV2) SendHeartbeat(_ context.Context, req *grpc.NodeServiceSe
}
return HandleError(err)
}
oldStatus := node.Status
// validate status
if node.Status == constants.NodeStatusUnregistered {
@@ -107,6 +110,14 @@ func (svr NodeServerV2) SendHeartbeat(_ context.Context, req *grpc.NodeServiceSe
if err != nil {
return HandleError(err)
}
newStatus := node.Status
// send notification if status changed
if utils.IsPro() {
if oldStatus != newStatus {
go notification.GetNotificationServiceV2().SendNodeNotification(node)
}
}
return HandleSuccessWithData(node)
}