feat: impl health checker and init data collector (#1077)

* feat: impl health checker and init data collector

* chore: change module name

---------

Co-authored-by: KAAANG <[email protected]>

Author: Kaikaikaifang

core/cmd/core/core.go

@@ -11,6 +11,8 @@ import (
 	"net"
 
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/health"
+	grpcHealth "google.golang.org/grpc/health/grpc_health_v1"
 
 	"github.com/SwanHubX/SwanLab/core/internal/service"
 	"github.com/SwanHubX/SwanLab/core/pkg/pb"
@@ -26,7 +28,13 @@ func main() {
 	}
 
 	s := grpc.NewServer()
-	pb.RegisterHealthServiceServer(s, &service.HealthService{})
+	// Health Check Service
+	healthCheck := health.NewServer()
+	grpcHealth.RegisterHealthServer(s, healthCheck)
+	// Collector Service
+	collector := &service.Collector{}
+	pb.RegisterCollectorServer(s, collector)
+	healthCheck.SetServingStatus("collector", grpcHealth.HealthCheckResponse_SERVING)
 	slog.Info("server listening at", "address", lis.Addr())
 
 	if err = s.Serve(lis); err != nil {

core/go.mod

@@ -3,12 +3,12 @@ module github.com/SwanHubX/SwanLab/core
 go 1.24.4
 
 require (
+	golang.org/x/net v0.38.0
 	google.golang.org/grpc v1.73.0
 	google.golang.org/protobuf v1.36.6
 )
 
 require (
-	golang.org/x/net v0.38.0 // indirect
 	golang.org/x/sys v0.31.0 // indirect
 	golang.org/x/text v0.23.0 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20250324211829-b45e905df463 // indirect

core/internal/service/collector.go

@@ -0,0 +1,21 @@
+package service
+
+import (
+	"log/slog"
+
+	"golang.org/x/net/context"
+
+	"github.com/SwanHubX/SwanLab/core/pkg/pb"
+)
+
+type Collector struct {
+	pb.UnimplementedCollectorServer
+}
+
+func (c *Collector) Upload(ctx context.Context, in *pb.CollectorUploadRequest) (*pb.CollectorUploadResponse, error) {
+	slog.InfoContext(ctx, "collector: Upload: started", "data", in.GetData())
+	return &pb.CollectorUploadResponse{
+		Success: true,
+		Message: "ok",
+	}, nil
+}