Support node query gRPC service in lifecycle service (#956)

Author: mrproliu

banyand/backup/lifecycle/current_node.go

@@ -0,0 +1,30 @@
+// Licensed to Apache Software Foundation (ASF) under one or more contributor
+// license agreements. See the NOTICE file distributed with
+// this work for additional information regarding copyright
+// ownership. Apache Software Foundation (ASF) licenses this file to you under
+// the Apache License, Version 2.0 (the "License"); you may
+// not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+package lifecycle
+
+import (
+	"context"
+
+	databasev1 "github.com/apache/skywalking-banyandb/api/proto/banyandb/database/v1"
+)
+
+func (l *lifecycleService) GetCurrentNode(_ context.Context, _ *databasev1.GetCurrentNodeRequest) (*databasev1.GetCurrentNodeResponse, error) {
+	return &databasev1.GetCurrentNodeResponse{
+		Node: l.currentNode,
+	}, nil
+}

banyand/backup/lifecycle/service.go

@@ -41,6 +41,7 @@ import (
 	"google.golang.org/grpc/credentials/insecure"
 	"google.golang.org/grpc/health"
 	"google.golang.org/grpc/health/grpc_health_v1"
+	"google.golang.org/protobuf/types/known/timestamppb"
 
 	"github.com/apache/skywalking-banyandb/api/common"
 	commonv1 "github.com/apache/skywalking-banyandb/api/proto/banyandb/common/v1"
@@ -70,6 +71,7 @@ var _ service = (*lifecycleService)(nil)
 
 type lifecycleService struct {
 	databasev1.UnimplementedClusterStateServiceServer
+	databasev1.UnimplementedNodeQueryServiceServer
 	metadata          metadata.Repo
 	omr               observability.MetricsRegistry
 	pm                protector.Memory
@@ -79,6 +81,7 @@ type lifecycleService struct {
 	grpcServer        *grpclib.Server
 	httpSrv           *http.Server
 	tlsReloader       *pkgtls.Reloader
+	currentNode       *databasev1.Node
 	clientCloser      context.CancelFunc
 	stopCh            chan struct{}
 	measureRoot       string
@@ -165,6 +168,16 @@ func (l *lifecycleService) Validate() error {
 				return errors.New("missing key file when TLS is enabled")
 			}
 		}
+		l.currentNode = &databasev1.Node{
+			Metadata: &commonv1.Metadata{
+				Name: l.lifecycleGRPCAddr,
+			},
+			GrpcAddress: l.lifecycleGRPCAddr,
+			HttpAddress: l.lifecycleHTTPAddr,
+			Roles:       make([]databasev1.Role, 0),
+			Labels:      common.ParseNodeFlags(),
+			CreatedAt:   timestamppb.Now(),
+		}
 	}
 	return nil
 }
@@ -309,6 +322,7 @@ func (l *lifecycleService) startServers() {
 
 	l.grpcServer = grpclib.NewServer(opts...)
 	databasev1.RegisterClusterStateServiceServer(l.grpcServer, l)
+	databasev1.RegisterNodeQueryServiceServer(l.grpcServer, l)
 	grpc_health_v1.RegisterHealthServer(l.grpcServer, health.NewServer())
 
 	// Setup HTTP server