如何在C#中根据可用系统内存用ThreadPool创建多线程

1. 实现系统可用内存获取（对应Python psutil逻辑） #

你的Python代码通过psutil获取系统可用物理内存，转换为GB并四舍五入。以下是C#的两种实现方式：

方法1：P/Invoke调用Windows API（无需额外引用） #

这种方式不需要添加第三方或系统额外引用，直接调用Windows内核API获取内存信息：

using System;
using System.Runtime.InteropServices;

public static class MemoryHelper
{
    [StructLayout(LayoutKind.Sequential)]
    private struct MEMORYSTATUSEX
    {
        public uint dwLength;
        public uint dwMemoryLoad;
        public ulong ullTotalPhys;
        public ulong ullAvailPhys;
        public ulong ullTotalPageFile;
        public ulong ullAvailPageFile;
        public ulong ullTotalVirtual;
        public ulong ullAvailVirtual;
        public ulong ullAvailExtendedVirtual;
    }

    [DllImport("kernel32.dll", CharSet = CharSet.Auto, SetLastError = true)]
    private static extern bool GlobalMemoryStatusEx(ref MEMORYSTATUSEX lpBuffer);

    public static int GetAvailableMemoryInGB()
    {
        MEMORYSTATUSEX status = new MEMORYSTATUSEX();
        status.dwLength = (uint)Marshal.SizeOf(status);
        GlobalMemoryStatusEx(ref status);
        
        // 转换为GB（1GB = 1024^3字节）
        double availableGB = status.ullAvailPhys / (1024.0 * 1024.0 * 1024.0);
        return (int)Math.Round(availableGB);
    }
}

方法2：使用System.Management（需添加引用） #

通过WMI查询系统内存信息，需要在项目中添加System.Management引用：

using System;
using System.Management;

public static class MemoryHelper
{
    public static int GetAvailableMemoryInGB()
    {
        using (var searcher = new ManagementObjectSearcher("SELECT FreePhysicalMemory FROM Win32_OperatingSystem"))
        {
            foreach (ManagementObject obj in searcher.Get())
            {
                // FreePhysicalMemory单位为KB，转换为GB
                ulong availableKB = Convert.ToUInt64(obj["FreePhysicalMemory"]);
                double availableGB = availableKB / (1024.0 * 1024.0);
                return (int)Math.Round(availableGB);
            }
        }
        return 0;
    }
}

两种方法都能实现和你Python代码完全一致的逻辑：获取系统可用物理内存，转换为GB后四舍五入。

2. 基于可用内存配置ThreadPool #

根据获取到的可用内存GB数，按每GB对应200个线程的规则配置线程池，示例如下：

using System;
using System.Threading;

class Program
{
    static void Main()
    {
        int availableGB = MemoryHelper.GetAvailableMemoryInGB();
        int targetThreadCount = availableGB * 200;

        // 获取当前线程池的线程数配置
        ThreadPool.GetMinThreads(out int currentMinWorkers, out int currentMinIO);
        ThreadPool.GetMaxThreads(out int currentMaxWorkers, out int currentMaxIO);

        // 调整线程池最大工作线程数（如果目标线程数超过默认上限）
        if (targetThreadCount > currentMaxWorkers)
        {
            ThreadPool.SetMaxThreads(targetThreadCount, currentMaxIO);
        }
        // 设置线程池最小工作线程数，确保线程池提前预热足够线程
        ThreadPool.SetMinThreads(targetThreadCount, currentMinIO);

        Console.WriteLine($"可用内存：{availableGB} GB，线程池配置最小工作线程数：{targetThreadCount}");

        // 示例：提交任务到线程池
        for (int i = 0; i < targetThreadCount; i++)
        {
            int taskId = i;
            ThreadPool.QueueUserWorkItem(_ =>
            {
                Console.WriteLine($"任务 {taskId} 执行，线程ID：{Thread.CurrentThread.ManagedThreadId}");
                // 替换为你的实际任务逻辑
                Thread.Sleep(1000); // 模拟任务耗时
            });
        }

        Console.ReadLine();
    }
}

关键注意点 #

• ThreadPool是线程池管理器，会自动处理线程的创建、复用和回收，无需手动创建线程。上述代码通过设置最小线程数预热线程池，或直接提交对应数量的任务让线程池按需扩容。
• 线程数并非越多越好：CPU密集型任务建议线程数接近CPU核心数；IO密集型任务可适当增加线程数，但需避免因上下文切换过多导致性能下降。
• 在.NET Core/.NET 5+中，ThreadPool的默认调度已做优化，也可通过ThreadPool.SetMinThreads快速预热线程池，避免任务等待线程创建的延迟。

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