Day 24 | 并发编程 (Goroutine)

package main

import (
    "fmt"
    "time"
)

func task(name string) {
    for i := 1; i <= 3; i++ {
        // 模拟干活耗时 100ms
        time.Sleep(100 * time.Millisecond)
        fmt.Printf("[%s] is processing step %d\n", name, i)
    }
}

func main() {
    // 瞬间派出两名员工，它们会同时开始跑
    go task("Agent-001")
    go task("Agent-002")

    // ⚠️ 极其重要: 主线程必须等待
    // 如果没有下面这行 Sleep，main 函数会立即退出
    // 导致上面的两个协程还没来得及打印就被系统杀死了
    time.Sleep(500 * time.Millisecond) 
    fmt.Println("Main thread quit.")
}

package main

import (
    "fmt"
    "time"
)

// LogProducer 模拟日志抓取
// chan<- 声明该 channel 只能用于发送内容 (Send Only)
func LogProducer(logQueue chan<- string) {
    logs := []string{"User login", "API 404 error", "DB query timeout", "Backup success"}
    
    for _, msg := range logs {
        fmt.Println("🚀 [Scraper] Captured log:", msg)
        logQueue <- msg // 将日志塞入传送带 (如果没人接，这里会阻塞)
        time.Sleep(200 * time.Millisecond)
    }
    // 重要：发完了必须关闭，否则消费者会永远等待（导致死锁）
    close(logQueue)
}

// LogConsumer 模拟日志持久化
// <-chan 声明该 channel 只能用于接收内容 (Receive Only)
func LogConsumer(logQueue <-chan string, done chan<- bool) {
    // 使用 range 优雅地从 Channel 读取
    // 它会一直循环，直到 Channel 被 Close() 且数据读完
    for msg := range logQueue {
        fmt.Printf("💾 [Storage] Saving to DB: %s...\n", msg)
        time.Sleep(300 * time.Millisecond) // 模拟存入数据库比较慢
    }
    fmt.Println("✅ [Storage] All logs processed.")
    
    // 任务全部搞定，给主线程发个信号
    done <- true
}

func main() {
    // 1. 创建无缓冲通道（传送带）
    logQueue := make(chan string)
    // 2. 用于同步的完成状态通道
    done := make(chan bool)

    // 启动并发流程
    go LogProducer(logQueue)
    go LogConsumer(logQueue, done)

    // 3. 阻塞主线程，等待 done 信号
    fmt.Println("System running...")
    <-done // 如果没有这行，主线程会直接退出
    fmt.Println("System stopped.")
}

package main

import (
    "fmt"
    "sync"
    "time"
)

// 注意：wg 必须传指针 *sync.WaitGroup，否则是拷贝，Done() 不会生效！
func checkHost(ip string, wg *sync.WaitGroup) {
    // 函数退出前，一定要通知计数器 -1
    // 使用 defer 是为了防止中间代码崩了导致没减计数，导致主线程死锁
    defer wg.Done()
    
    fmt.Printf("🔍 Checking host: %s\n", ip)
    time.Sleep(1 * time.Second) // 模拟网络延迟
    fmt.Printf("✅ %s is ONLINE\n", ip)
}

func main() {
    var wg sync.WaitGroup
    hosts := []string{"10.0.0.1", "10.0.0.2", "192.168.1.100", "8.8.8.8"}

    for _, ip := range hosts {
        wg.Add(1) // 派人，计数器 +1
        
        // 启动协程
        // 注意：这里传的是 &wg (指针)
        go checkHost(ip, &wg)
    }

    fmt.Println("⏳ Monitoring started. Waiting for all results...")
    
    // 这里会卡住，直到计数器归零
    wg.Wait()
    
    fmt.Println("🚀 All hosts checked. Report generated.")
}