初始化Go学习项目

go-goroutine-gpm/main.go

@@ -0,0 +1,76 @@
+package main
+
+import (
+	"fmt"
+	"runtime"
+	"sync"
+	"time"
+)
+
+func say(s string, id int) {
+	for i := 0; i < 3; i++ {
+		fmt.Printf("[%d] %s: step %d\n", id, s, i)
+		time.Sleep(100 * time.Millisecond) // 模拟工作
+	}
+}
+
+func main() {
+	fmt.Println("=== 1. 单线程顺序执行 ===")
+	say("Hello", 1)
+	say("World", 2)
+
+	fmt.Println("\n=== 2. 使用 Goroutine 并发执行 ===")
+	go say("Goroutine-A", 1)
+	go say("Goroutine-B", 2)
+
+	// 主 goroutine 等待 1 秒，否则程序会提前退出
+	time.Sleep(1 * time.Second)
+
+	fmt.Println("\n=== 3. 使用 sync.WaitGroup 安全等待 ===")
+	var wg sync.WaitGroup
+
+	for i := 1; i <= 3; i++ {
+		wg.Add(1)
+		go func(id int) {
+			defer wg.Done()
+			fmt.Printf("Worker %d started\n", id)
+			time.Sleep(time.Duration(id*200) * time.Millisecond)
+			fmt.Printf("Worker %d finished\n", id)
+		}(i)
+	}
+
+	wg.Wait() // 阻塞直到所有 goroutine 完成
+	fmt.Println("All workers done!")
+
+	fmt.Println("\n=== 4. 查看当前 GOMAXPROCS 和 CPU 核心数 ===")
+	fmt.Printf("CPU 核心数: %d\n", runtime.NumCPU())
+	fmt.Printf("GOMAXPROCS: %d\n", runtime.GOMAXPROCS(0)) // 0 表示不修改，仅查询
+
+	fmt.Println("\n=== 5. 手动设置 GOMAXPROCS（通常不需要）===")
+	old := runtime.GOMAXPROCS(2)
+	fmt.Printf("旧 GOMAXPROCS: %d, 新设为: 2\n", old)
+	runtime.GOMAXPROCS(old) // 恢复原值
+
+	fmt.Println("\n=== 6. 观察 Goroutine 数量变化 ===")
+	fmt.Printf("启动前 Goroutine 数: %d\n", runtime.NumGoroutine())
+
+	var wg2 sync.WaitGroup
+	for i := 0; i < 5; i++ {
+		wg2.Add(1)
+		go func(n int) {
+			defer wg2.Done()
+			time.Sleep(200 * time.Millisecond)
+		}(i)
+	}
+
+	time.Sleep(50 * time.Millisecond) // 让 goroutine 启动
+	fmt.Printf("启动后 Goroutine 数: %d\n", runtime.NumGoroutine())
+
+	wg2.Wait()
+	fmt.Printf("全部完成后 Goroutine 数: %d\n", runtime.NumGoroutine())
+
+	// 在程序最后添加
+	time.Sleep(1 * time.Second)
+	fmt.Printf("延迟后 Goroutine 数: %d\n", runtime.NumGoroutine())
+
+}