CpuTool.cs 1.54 KB
/********************************************************************************
** 类名称: CpuTool
** 描述  : 用于线程绑定CPU的处理类
** 作者  : 丁书杰
** 创建时间:2019/03/26
** 版权所有 (C) :中科视语(北京)科技有限公司
*********************************************************************************/
using System;
using System.Runtime.InteropServices;

namespace OS.Spin.Common.Machine
{
    public class CpuTool
    {
        /// <summary>
        /// SetThreadAffinityMask 指定hThread 运行在 核心 dwThreadAffinityMask
        /// </summary>
        /// <param name="hThread">线程句柄</param>
        /// <param name="dwThreadAffinityMask">绑定的CPU标记</param>
        /// <returns></returns>
        [DllImport("kernel32.dll")]
        public static extern UIntPtr SetThreadAffinityMask(IntPtr hThread,
        UIntPtr dwThreadAffinityMask);

        /// <summary>
        /// 得到当前线程的handler
        /// </summary>
        /// <returns></returns>
        [DllImport("kernel32.dll")]
        public static extern IntPtr GetCurrentThread();

        /// <summary>
        /// 获取线程ID
        /// </summary>
        /// <param name="id">设置</param>
        /// <returns>返回的线程Id</returns>
       public static ulong SetCpuID(int id)
        {
            ulong cpuid = 0;
            if (id < 0 || id >= System.Environment.ProcessorCount)
            {
                id = 0;
            }
            cpuid |= 1UL << id;

            return cpuid;
        }
    }
}