NestJS gRPC未实现的流式服务器方法

Question

我正在尝试用NestJS和gRPC构建微服务。它有两个服务，第一个是gRPC服务，第二个是调用gRPC服务的REST服务。

首先，它可以很好地处理一元调用，即findRepoById。但它不适用于服务器流式调用findAllRepos。当我试图调用findAllRepos时，它抛出如下错误

UnhandledPromiseRejection警告：错误：12 UNIMPLEMENTED：服务器没有实现findAllRepos方法

我写的文件如下所示

// main.proto

syntax = "proto3";

import "google/protobuf/timestamp.proto";

package main;

enum Visibility {
  PUBLIC = 0;
  PRIVATE = 1;
}

message Repository {
  int32 id = 1;
  string title = 2;
  Visibility visibility = 3;
  google.protobuf.Timestamp lastSeen = 4;
}

service MainService {
  rpc findRepoById (RepoById) returns (Repository) {}
  rpc findAllRepos (NoParam) returns (stream Repository) {}
}

message RepoById {
  int32 id = 1;
}

message NoParam {}

// server.controller.ts

@Controller('repo')
export class RepoController {
  constructor(
    @Inject('RepoService') private readonly repoService: RepoService
  ) {}

  @GrpcMethod('MainService', 'findRepoById')
  findRepoById(request: RepoById, metadata: Metadata): Repository {
    const targetId = request.id;
    return this.repoService.findRepoById(targetId);
  }

  @GrpcStreamMethod('MainService', 'findAllRepos')
  findAllRepos(request: NoParam, metadata: Metadata): Observable<Repository> {
    const subject = new Subject<Repository>();

    const repositories = this.repoService.findAllRepos();
    repositories.map((repo) => {
      subject.next(repo);
    });
    subject.complete();

    return subject.asObservable();
  }
}

// client.service.ts

export class RepoGrpcService implements OnModuleInit {
  private mainService: MainServiceClient;

  constructor(@Inject('main_package') private client: ClientGrpc) {}

  onModuleInit() {
    this.mainService = this.client.getService<MainServiceClient>('MainService');
  }

  findRepoById(id: number): Observable<Repository> {
    return this.mainService.findRepoById({ id });
  }

  @GrpcStreamCall('MainService')
  findAllRepos(): Observable<Repository[]> {
    const results: Repository[] = [];

    const repoStream = this.mainService.findAllRepos({});
    repoStream.forEach((value) => console.log(value));
    repoStream.subscribe({
      next: (repo) => {
        results.push(repo);
      }
    });

    const subject = new Subject<Repository[]>();
    subject.next(results);
    subject.complete();

    return subject.asObservable();
  }
}

我想我已经遵循了NestJS gRPC文档中的所有代码，但不知怎么的，它仍然不起作用。我做错什么了吗？

Answer 1

根据我的经验，如果是服务器端流，@GrpcMethod注释可以工作。文档似乎没有足够清楚地描述这一点，但对我来说，似乎只有双向流服务需要GrpcStreamMethod？

Answer 2

我知道这个问题已经问了很久了，但这个答案可能会帮助其他人。请注意文档，由于前面章节的原因，代码的某些部分没有提及，因此NestJS也不例外。您的问题似乎来自控制器模块关系再次查看文档，确保您的控制器在正确的模块中定义，并且您的模块关系（导入）良好。

NestJs文档模块

官方示例