如何在自定义线程池上执行返回ValueTask的函数及任务续体

• 临时切换同步上下文的操作完全多余：TaskScheduler.FromCurrentSynchronizationContext()本身只是读取当前上下文实例生成对应调度器，你完全可以直接实现TaskScheduler子类绑定自定义线程池，完全避免切换上下文的冗余操作。
• 隐式依赖同步上下文调度续体的逻辑存在漏洞：如果DoStuff内部主动调用了ConfigureAwait(false)，异步续体就会跳回默认线程池执行，无法满足你所有执行片段都在自定义线程池的要求。
• 流程笨重的核心问题是没有对入队ValueTask的逻辑做封装，而非AsTask()本身的开销：同步完成的ValueTask转Task几乎没有额外分配，问题出在你每次入队都重复走了上下文切换、调度器生成的流程。

核心思路是直接实现绑定自定义线程池的TaskScheduler，省略同步上下文的桥接层，再封装轻量的入队扩展方法适配ValueTask。

步骤1：实现自定义TaskScheduler #

public class MyThreadPoolTaskScheduler : TaskScheduler
{
    // 这里假设你的自定义线程池已经实现了Queue方法用于入队执行Action
    private readonly MyThreadPool _threadPool;

    public MyThreadPoolTaskScheduler(MyThreadPool threadPool) => _threadPool = threadPool;

    protected override IEnumerable<Task> GetScheduledTasks() => Enumerable.Empty<Task>();

    protected override void QueueTask(Task task)
    {
        // 将任务执行逻辑投递到自定义线程池
        _threadPool.Queue(() => TryExecuteTask(task));
    }

    protected override bool TryExecuteTaskInline(Task task, bool taskWasPreviouslyQueued)
    {
        // 若当前线程属于自定义线程池，可以直接执行任务，否则返回false禁止内联执行
        return false;
    }
}

你可以将该调度器实例和对应的TaskFactory直接缓存在MyThreadPool类中，避免重复生成。

步骤2：封装ValueTask入队扩展 #

public static class MyThreadPoolExtensions
{
    public static void Queue(this MyThreadPool threadPool, Func<ValueTask> valueTaskFunc)
    {
        // 直接用提前缓存的自定义调度器启动异步任务
        threadPool.TaskFactory.StartNew(async () =>
        {
            await valueTaskFunc();
        }, CancellationToken.None, TaskCreationOptions.None, threadPool.Scheduler);
    }
}

步骤3：调用简化 #

class Example
{
    async ValueTask DoStuff(int something)
    {
        // 业务逻辑，若要保证所有续体都在自定义线程池，不要主动调用ConfigureAwait(false)
    }

    private void Test()
    {
        // 你的自定义线程池实例，提前初始化完成，内部已绑定自定义调度器和TaskFactory
        MyThreadPool myThreadPool = new MyThreadPool();
        
        for (int i = 0; i < 1_000; i++)
        {
            int capture = i; // 规避闭包捕获循环变量的问题
            myThreadPool.Queue(() => DoStuff(capture));
        }
    }
}

如果你需要彻底杜绝ConfigureAwait(false)导致的续体跳走问题，可以自定义ValueTask的awaiter，强制所有续体投递到你的自定义线程池执行即可。

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