本文详解如何在 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("设置亲和性失败: %vn", 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 跨节点调度等）。