本文详解如何在 Windows（包括 Win7）平台下，通过 Go 调用 Win32 API SetProcessAffinityMask 设置进程的 CPU 亲和性掩码，规避 Linux 专用 syscall 的兼容性问题，并提供可运行的完整示例与关键注意事项。
本文详解如何在 windows（包括 win7）平台下，通过 go 调用 win32 api `setprocessaffinitymask` 设置进程的 cpu 亲和性掩码，规避 linux 专用 syscall 的兼容性问题，并提供可运行的完整示例与关键注意事项。
在 Linux 系统中，开发者常借助 sched_setaffinity 等系统调用设置进程 CPU 亲和性，但该机制在 Windows 上并不存在——Go 标准库中的 syscall.SYS_SCHED_SETAFFINITY 仅适用于 Unix-like 系统，因此直接编译会报错 undefined: syscall.SYS_SCHED_SETAFFINITY。Windows 提供了等效的 Win32 API：SetProcessAffinityMask（位于 kernel32.dll），它允许为指定进程句柄（handle）配置 CPU 亲和性掩码（affinity mask），即以位图形式指定进程可运行的逻辑处理器集合。
以下是一个跨版本兼容（支持 Windows 7 及以上）、安全可靠的 Go 实现：
package main
import (
"fmt"
"runtime"
"syscall"
"unsafe"
)
// setProcessAffinityMask 调用 Win32 API SetProcessAffinityMask
// h: 进程句柄（需具备 PROCESS_SET_INFORMATION 权限）
// mask: CPU 亲和性掩码（bit i 表示是否允许在第 i 个逻辑处理器上运行）
func setProcessAffinityMask(h syscall.Handle, mask uintptr) error {
kernel32 := syscall.NewLazyDLL("kernel32.dll")
proc := kernel32.NewProc("SetProcessAffinityMask")
r1, _, e1 := proc.Call(uintptr(h), mask)
if r1 == 0 {
if e1 != 0 {
return error(e1)
}
return syscall.EINVAL
}
return nil
}
// getCurrentProcess 返回当前进程的伪句柄（无需显式 OpenProcess）
func getCurrentProcess() syscall.Handle {
return syscall.Handle(^uintptr(1)) // = CURRENT_PROCESS_HANDLE
}
func main() {
// 示例：将当前进程绑定到 CPU 0（掩码 0x1）
mask := uintptr(1) // 仅启用第 0 号逻辑 CPU
// 获取当前进程句柄（具有足够权限）
h := getCurrentProcess()
// 应用亲和性掩码
if err := setProcessAffinityMask(h, mask); err != nil {
fmt.Printf("设置亲和性失败: %v\n", err)
return
}
fmt.Printf("成功将进程绑定到 CPU 0（掩码: 0x%x）\n", mask)
// 验证：获取当前亲和性（可选，需调用 GetProcessAffinityMask）
// 注意：Go 标准库未封装该函数，如需验证，需自行封装或使用其他工具（如 taskmgr / PowerShell）
}✅ 关键说明与注意事项：
- 进程句柄权限：SetProcessAffinityMask 要求句柄具备 PROCESS_SET_INFORMATION 访问权限。使用 GetCurrentProcess()（即 ^uintptr(1)）返回的伪句柄默认拥有该权限，无需调用 OpenProcess，简化且安全。
-
掩码格式：掩码是 uintptr 类型的位图，第 i 位为 1 表示允许在逻辑处理器 i 上执行。例如：
- 0x1 → 仅 CPU 0
- 0x3 → CPU 0 和 CPU 1
- 0xFF → 前 8 个 CPU
超出系统实际 CPU 数量的位会被忽略。
- 多核/超线程兼容性：该 API 作用于逻辑处理器（logical processor），自动适配物理核心 + 超线程（HT）环境，无需手动区分。
- Windows 7 支持：SetProcessAffinityMask 自 Windows XP 起即已存在，完全兼容 Windows 7。
- 错误处理：返回值 r1 == 0 表示失败；应优先检查 e1（系统错误码），再回退至通用错误（如 EINVAL）。
- 不建议硬编码 PID：原始问题中尝试传入整数 PID 是错误的——Win32 API 接收的是句柄（Handle），而非 PID。若需设置其他进程，请先用 OpenProcess(PROCESS_SET_INFORMATION, false, pid) 获取句柄（需管理员权限及目标进程未受保护）。
? 进阶提示：若需动态获取系统 CPU 总数并构造全核掩码，可结合 runtime.NumCPU() 使用（注意：NumCPU() 返回逻辑 CPU 数，与掩码位宽一致）：
n := runtime.NumCPU()
fullMask := uintptr(0)
for i := 0; i < n; i++ {
fullMask |= (1 << uint(i))
}
// fullMask 即覆盖全部逻辑 CPU 的掩码掌握此方法后，你即可在 Windows Go 程序中精准控制 CPU 资源分配，适用于性能敏感场景（如实时计算、低延迟服务、避免 NUMA 跨节点调度等）。
