Typescript GRPC客户端拦截器

基础概念

gRPC（Google Remote Procedure Call）是一种高性能、开源和通用的RPC框架，用于在不同服务之间进行通信。TypeScript是JavaScript的一个超集，提供了静态类型检查和其他高级特性。gRPC客户端拦截器允许你在客户端调用gRPC服务之前或之后执行自定义逻辑。

类型

gRPC客户端拦截器主要有两种类型：

UnaryInterceptor：用于处理单个请求和响应的拦截器。
StreamInterceptor：用于处理流式请求和响应的拦截器。

应用场景

日志记录：记录每个gRPC调用的详细信息，便于调试和监控。
认证和授权：在调用gRPC服务前验证用户的身份和权限。
错误处理：在调用失败后进行自定义的错误处理和重试逻辑。
请求修改：在发送请求前对请求内容进行修改，例如添加额外的请求头。
响应处理：在接收响应后对响应内容进行处理，例如解密或格式化。

示例代码

以下是一个使用TypeScript编写的gRPC客户端拦截器的示例：

import { Client, ClientUnaryCall, ClientReadableStream } from '@grpc/grpc-js';
import { Request, Response } from './your-grpc-service'; // 替换为你的gRPC服务定义

// UnaryInterceptor 示例
function unaryInterceptor(
  error: grpc.ServiceError | null,
  req: Request,
  metadata: grpc.Metadata,
  client: Client,
  method: grpc.UnaryMethodDefinition<Request, Response>
): Response {
  console.log('UnaryInterceptor - Before request');
  if (error) {
    console.error('UnaryInterceptor - Error:', error);
  }
  return method(req, metadata, client);
}

// StreamInterceptor 示例
function streamInterceptor(
  err: grpc.ServiceError | null,
  reqs: Request[],
  metadata: grpc.Metadata,
  client: Client,
  method: grpc.ClientStreamingMethodDefinition<Request, Response>
): ClientReadableStream<Response> {
  console.log('StreamInterceptor - Before request');
  if (err) {
    console.error('StreamInterceptor - Error:', err);
  }
  return method(reqs, metadata, client);
}

// 创建gRPC客户端并应用拦截器
const client = new YourGrpcServiceClient('localhost:50051', grpc.credentials.createInsecure());
client.yourMethod = unaryInterceptor.bind(null, client.yourMethod);
client.yourStreamingMethod = streamInterceptor.bind(null, client.yourStreamingMethod);