Go - GMP 调度模型

// G 代表 Goroutine，包含：
// - 栈信息（栈指针、栈大小）
// - 调度信息（状态、优先级）
// - 上下文信息（寄存器）

type g struct {
    stack       stack   // 栈内存范围 [stack.lo, stack.hi)
    stackguard0 uintptr // 栈溢出检查
    m           *m      // 当前关联的 M
    sched       gobuf   // 调度上下文（PC、SP 等）
    atomicstatus uint32 // G 的状态
    goid         int64  // G 的唯一标识
    // ...
}

// M 代表操作系统线程
type m struct {
    g0      *g     // 调度栈，用于执行调度代码
    curg    *g     // 当前运行的 G
    p       puintptr // 关联的 P
    spinning bool  // 是否在自旋寻找工作
    // ...
}

// P 代表逻辑处理器，是 G 和 M 之间的桥梁
type p struct {
    id          int32
    status      uint32 // P 的状态
    m           muintptr // 关联的 M
    runqhead    uint32   // 本地队列头
    runqtail    uint32   // 本地队列尾
    runq        [256]guintptr // 本地运行队列
    runnext     guintptr // 下一个要运行的 G
    // ...
}

              ┌───────────────────────────────────────┐
              │            全局队列 (Global Queue)     │
              │        ┌──┬──┬──┬──┬──┬──┬──┐        │
              │        │G │G │G │G │G │G │G │        │
              │        └──┴──┴──┴──┴──┴──┴──┘        │
              └───────────────────────────────────────┘
                              ↑  ↓
        ┌─────────────────────┼──┼─────────────────────┐
        │                     │  │                     │
   ┌────┴────┐           ┌────┴──┴───┐           ┌────┴────┐
   │    P    │           │     P     │           │    P    │
   │ ┌─────┐ │           │  ┌─────┐  │           │ ┌─────┐ │
   │ │runq │ │           │  │runq │  │           │ │runq │ │
   │ └─────┘ │           │  └─────┘  │           │ └─────┘ │
   └────┬────┘           └─────┬─────┘           └────┬────┘
        │                      │                      │
   ┌────┴────┐           ┌─────┴─────┐           ┌────┴────┐
   │    M    │           │     M     │           │    M    │
   │  (OS    │           │   (OS     │           │  (OS    │
   │ Thread) │           │  Thread)  │           │ Thread) │
   └─────────┘           └───────────┘           └─────────┘

// 调度循环伪代码
func schedule() {
    // 1. 从 runnext 获取 G
    gp := pp.runnext
    if gp != nil {
        pp.runnext = nil
        return gp
    }
    
    // 2. 从本地队列获取 G
    gp = runqget(pp)
    if gp != nil {
        return gp
    }
    
    // 3. 从全局队列获取 G
    gp = globrunqget(pp, 0)
    if gp != nil {
        return gp
    }
    
    // 4. 从其他 P 偷取 G
    gp = stealWork(pp)
    if gp != nil {
        return gp
    }
    
    // 5. 没有可运行的 G，休眠
    stopm()
}

// go func() 的底层实现
func newproc(fn *funcval) {
    // 1. 获取当前 G
    gp := getg()
    
    // 2. 获取调用者 PC
    pc := getcallerpc()
    
    // 3. 在系统栈上创建新 G
    systemstack(func() {
        newg := newproc1(fn, gp, pc)
        
        // 4. 将新 G 放入 P 的本地队列
        runqput(pp, newg, true)
        
        // 5. 如果有空闲的 P，唤醒一个 M
        if atomic.Load(&sched.npidle) != 0 {
            wakep()
        }
    })
}

// 1. runtime.Gosched()
func example() {
    go func() {
        for i := 0; i < 10; i++ {
            fmt.Println(i)
            runtime.Gosched()  // 主动让出 CPU
        }
    }()
}

// 2. channel 操作
ch := make(chan int)
<-ch  // 阻塞，触发调度

// 3. 锁操作
var mu sync.Mutex
mu.Lock()  // 如果锁被持有，触发调度

// 4. time.Sleep
time.Sleep(time.Second)  // 触发调度

// Go 1.14+ 基于信号的抢占
// 当 G 运行超过 10ms，会被抢占

func longLoop() {
    for {
        // 这个循环可以被抢占
        // 不再需要函数调用
    }
}

// 系统调用导致的调度
func syscall() {
    // 1. 进入系统调用前
    //    - M 解绑 P
    //    - P 可以绑定其他 M
    
    // 2. 系统调用执行中
    //    - M 被阻塞
    //    - P 继续调度其他 G
    
    // 3. 系统调用返回后
    //    - M 尝试获取原来的 P
    //    - 如果 P 被占用，获取空闲 P
    //    - 如果没有空闲 P，将 G 放入全局队列
}

func stealWork(pp *p) *g {
    // 随机选择一个 P
    for i := 0; i < 4; i++ {
        victim := allp[fastrand() % len(allp)]
        if victim == pp {
            continue
        }
        
        // 偷取一半的 G
        n := victim.runqlen() / 2
        if n > 0 {
            return runqsteal(pp, victim, n)
        }
    }
    
    return nil
}

func sysmon() {
    for {
        // 1. 检查死锁
        checkdead()
        
        // 2. 抢占长时间运行的 G
        retake(now)
        
        // 3. 触发 GC
        if gcTrigger {
            gcStart()
        }
        
        // 4. 归还长时间阻塞的 P
        // 5. 唤醒 netpoller
        
        // 休眠一段时间
        usleep(delay)
    }
}

import "runtime"

func init() {
    // 设置 P 的数量
    // 默认等于 CPU 核数
    runtime.GOMAXPROCS(runtime.NumCPU())
}

// 也可以通过环境变量设置
// GOMAXPROCS=4 ./myapp

// 使用 worker pool 限制并发数
func processWithLimit(tasks []Task, limit int) {
    sem := make(chan struct{}, limit)
    var wg sync.WaitGroup
    
    for _, task := range tasks {
        wg.Add(1)
        sem <- struct{}{}  // 获取信号量
        
        go func(t Task) {
            defer wg.Done()
            defer func() { <-sem }()  // 释放信号量
            
            process(t)
        }(task)
    }
    
    wg.Wait()
}

// 设置 GODEBUG 查看调度信息
// GODEBUG=schedtrace=1000 ./myapp

// 输出示例：
// SCHED 1000ms: gomaxprocs=4 idleprocs=2 threads=6 
//               spinningthreads=0 idlethreads=3 
//               runqueue=0 [2 3 1 0]

// 解释：
// gomaxprocs: P 的数量
// idleprocs: 空闲 P 数量
// threads: M 的数量
// runqueue: 全局队列长度
// [2 3 1 0]: 各 P 本地队列长度