如何为Micronaut gRPC指定服务器执行器（如VIRTUAL线程执行器）

问题描述 #

Micronaut gRPC文档的「gRPC Server」章节提到，默认情况下服务器会使用Micronaut的I/O执行器服务。这表明应该可以将服务器配置为使用非默认执行器服务，我尤其希望让它使用VIRTUAL执行器服务。

Micronaut gRPC的配置参考指出grpc.server.executor是GrpcServerConfiguration的可配置属性，类型为java.util.concurrent.Executor。但查看GrpcServerConfiguration的源码后，我不确定是否真的可以更改服务器执行器：其构造函数接受一个执行器，但明确指定了TaskExecutors.IO，且我未找到覆盖该设置的钩子；GrpcServerConfiguration仅在构造函数中调用NettyServerBuilder#executor，且没有任何公共（或私有）方法可在构造后修改它。

我已尝试将grpc.server.executor设置为BLOCKING/VIRTUAL，但gRPC调用仍会发送到IO执行器。想知道是否有其他方法可以配置Micronaut gRPC，使其为gRPC服务器执行器使用虚拟线程（或TaskExecutors.IO以外的任何执行器）？

解决方案 #

1. 自定义Bean替换默认GrpcServerConfiguration #

通过创建自定义的GrpcServerConfiguration Bean，手动传入目标执行器，覆盖默认实现：

import io.micronaut.context.annotation.Bean;
import io.micronaut.context.annotation.Factory;
import io.micronaut.context.annotation.Replaces;
import io.micronaut.grpc.server.GrpcServerConfiguration;
import io.micronaut.scheduling.TaskExecutors;
import jakarta.inject.Named;
import java.util.concurrent.Executor;

@Factory
public class GrpcServerExecutorConfig {

    @Bean
    @Replaces(GrpcServerConfiguration.class)
    public GrpcServerConfiguration customGrpcServerConfiguration(
            @Named(TaskExecutors.VIRTUAL) Executor virtualExecutor,
            GrpcServerConfiguration.DefaultGrpcServerConfiguration defaultConfig) {
        return new GrpcServerConfiguration(defaultConfig.getConfig(), virtualExecutor);
    }
}

2. 用GrpcServerBuilderCustomizer修改Netty构建器 #

实现GrpcServerBuilderCustomizer接口，直接修改Netty服务器构建器的执行器设置：

import io.grpc.netty.NettyServerBuilder;
import io.micronaut.grpc.server.GrpcServerBuilderCustomizer;
import io.micronaut.scheduling.TaskExecutors;
import jakarta.inject.Named;
import jakarta.inject.Singleton;
import java.util.concurrent.Executor;

@Singleton
public class VirtualThreadServerBuilderCustomizer implements GrpcServerBuilderCustomizer {

    private final Executor virtualExecutor;

    public VirtualThreadServerBuilderCustomizer(@Named(TaskExecutors.VIRTUAL) Executor virtualExecutor) {
        this.virtualExecutor = virtualExecutor;
    }

    @Override
    public void customize(NettyServerBuilder builder) {
        builder.executor(virtualExecutor);
    }
}

3. 验证执行器生效状态 #

添加日志拦截器，确认请求是否使用目标执行器：

import io.grpc.*;
import jakarta.inject.Singleton;

@Singleton
public class ExecutorLoggingInterceptor implements ServerInterceptor {

    @Override
    public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(
            ServerCall<ReqT, RespT> call,
            Metadata headers,
            ServerCallHandler<ReqT, RespT> next) {
        Thread currentThread = Thread.currentThread();
        System.out.printf("gRPC call handled by thread: %s, is virtual: %b%n",
                currentThread.getName(),
                currentThread.isVirtual());
        return next.startCall(call, headers);
    }
}

在配置文件中注册拦截器：

grpc:
  server:
    interceptors:
      - executorLoggingInterceptor

内容的提问来源于stack exchange，提问作者Jon Chambers