灵隙(synapse) v1.0.0 版本

internal/api/grpcapi/server.go

@@ -96,15 +96,24 @@ type VoteStatus struct {
 // NewGRPCServer creates a new gRPC server
 func NewGRPCServer(taskExec TaskExecutor, msgRouter MessageRouter, capsProvider CapabilitiesProvider, groupManager GroupManager) *GRPCServer {
 	return &GRPCServer{
-		taskExecutor:        taskExec,
-		msgRouter:           msgRouter,
-		capsProvider:        capsProvider,
-		groupManager:        groupManager,
-		taskService:         &TaskServiceImpl{},
-		messageService:      &MessageServiceImpl{},
-		capabilitiesService: &CapabilitiesServiceImpl{},
-		groupService:        &GroupServiceImpl{},
-		healthService:       &HealthServiceImpl{},
+		taskExecutor: taskExec,
+		msgRouter:    msgRouter,
+		capsProvider: capsProvider,
+		groupManager: groupManager,
+		taskService: &TaskServiceImpl{
+			taskExecutor:  taskExec,
+			streamClients: make(map[string]chan *TaskStatusUpdate),
+		},
+		messageService: &MessageServiceImpl{
+			msgRouter: msgRouter,
+		},
+		capabilitiesService: &CapabilitiesServiceImpl{
+			provider: capsProvider,
+		},
+		groupService: &GroupServiceImpl{
+			groupManager: groupManager,
+		},
+		healthService: &HealthServiceImpl{},
 	}
 }
 
@@ -124,11 +133,12 @@ func (s *GRPCServer) Start(grpcPort, gatewayPort int) error {
 		}
 		s.grpcServer = grpc.NewServer(opts...)
 
-		// Register services
-		// Note: These would be generated from proto
-		// RegisterTaskServiceServer(s.grpcServer, s.taskService)
-		// RegisterA2AMessageServiceServer(s.grpcServer, s.messageService)
-		// ... etc
+		// Register services using ServiceDesc
+		s.grpcServer.RegisterService(&TaskService_ServiceDesc, s.taskService)
+		s.grpcServer.RegisterService(&A2AMessageService_ServiceDesc, s.messageService)
+		s.grpcServer.RegisterService(&CapabilitiesService_ServiceDesc, s.capabilitiesService)
+		s.grpcServer.RegisterService(&GroupService_ServiceDesc, s.groupService)
+		s.grpcServer.RegisterService(&HealthService_ServiceDesc, s.healthService)
 
 		log.Printf("[gRPC] Server started on port %d", grpcPort)
 		if err := s.grpcServer.Serve(lis); err != nil {