From 40f37e85ef5d62ea8c5945e7a593cd7207fca3d6 Mon Sep 17 00:00:00 2001 From: Marvin Zhang Date: Wed, 3 Jul 2024 14:57:33 +0800 Subject: [PATCH] fix: missing name and max runners when registering nodes --- core/grpc/server/node_server_v2.go | 12 +++++++----- core/node/config/config_service.go | 3 --- 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/core/grpc/server/node_server_v2.go b/core/grpc/server/node_server_v2.go index 7df8d997..a45e222e 100644 --- a/core/grpc/server/node_server_v2.go +++ b/core/grpc/server/node_server_v2.go @@ -58,11 +58,13 @@ func (svr NodeServerV2) Register(ctx context.Context, req *grpc.NodeServiceRegis } else if errors2.Is(err, mongo.ErrNoDocuments) { // register new node = &models.NodeV2{ - Key: req.Key, - Status: constants.NodeStatusRegistered, - Active: true, - ActiveAt: time.Now(), - Enabled: true, + Key: req.Key, + Name: req.Name, + Status: constants.NodeStatusRegistered, + Active: true, + ActiveAt: time.Now(), + Enabled: true, + MaxRunners: int(req.MaxRunners), } node.SetCreated(primitive.NilObjectID) node.SetUpdated(primitive.NilObjectID) diff --git a/core/node/config/config_service.go b/core/node/config/config_service.go index 9e772c6a..9a4cee4f 100644 --- a/core/node/config/config_service.go +++ b/core/node/config/config_service.go @@ -119,9 +119,6 @@ var _service interfaces.NodeConfigService var _serviceOnce = new(sync.Once) func GetNodeConfigService() interfaces.NodeConfigService { - if _service != nil { - return _service - } _serviceOnce.Do(func() { var err error _service, err = newNodeConfigService()