58 lines
2.3 KiB
C#
58 lines
2.3 KiB
C#
using Microsoft.AspNetCore.Builder;
|
||
using Microsoft.AspNetCore.Routing;
|
||
using Microsoft.Extensions.DependencyInjection;
|
||
using System.Reflection;
|
||
|
||
namespace IM.InitCommon
|
||
{
|
||
public static class GrpcExtension
|
||
{
|
||
public static IServiceCollection AddAllGrpcServer(this IServiceCollection services)
|
||
{
|
||
services.AddGrpc(options =>
|
||
{
|
||
// 开启详细错误（开发环境很有用，生产环境可结合配置读取）
|
||
options.EnableDetailedErrors = true;
|
||
|
||
// 限制最大接收和发送的消息大小 (例如 10MB，防止大包攻击)
|
||
options.MaxReceiveMessageSize = 10 * 1024 * 1024;
|
||
options.MaxSendMessageSize = 10 * 1024 * 1024;
|
||
|
||
// TODO: 未来你可以在这里添加全局异常拦截器 (Interceptor)
|
||
// options.Interceptors.Add<GlobalGrpcExceptionInterceptor>();
|
||
});
|
||
|
||
return services;
|
||
}
|
||
|
||
public static IEndpointRouteBuilder MapAllGrpcServer(this IEndpointRouteBuilder endpoints)
|
||
{
|
||
// 获取调用此方法的程序集（即具体的微服务项目，如 MessageService）
|
||
var assembly = Assembly.GetCallingAssembly();
|
||
|
||
// 获取 MapGrpcService<T> 的方法反射信息
|
||
var mapGrpcServiceMethod = typeof(GrpcEndpointRouteBuilderExtensions)
|
||
.GetMethods(BindingFlags.Static | BindingFlags.Public)
|
||
.First(m => m.Name == "MapGrpcService" && m.GetGenericArguments().Length == 1);
|
||
|
||
// 查找当前项目中所有继承了 gRPC 生成的 Base 类的具体实现类
|
||
// gRPC 生成的基类通常以 "Base" 结尾，例如 ConversationInternalBase
|
||
var grpcTypes = assembly.GetTypes()
|
||
.Where(t => t.IsClass
|
||
&& !t.IsAbstract
|
||
&& t.BaseType != null
|
||
&& t.BaseType.Name.EndsWith("Base"))
|
||
.ToList();
|
||
|
||
// 循环并动态调用 MapGrpcService<T>
|
||
foreach (var type in grpcTypes)
|
||
{
|
||
var genericMethod = mapGrpcServiceMethod.MakeGenericMethod(type);
|
||
genericMethod.Invoke(null, new object[] { endpoints });
|
||
}
|
||
|
||
return endpoints;
|
||
}
|
||
}
|
||
}
|