怎样用Golang开发混沌工程工具 模拟系统故障方案

package main

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

func simulateCPUUsage(wg *sync.WaitGroup) {
    defer wg.Done()
    for {
        // 模拟 CPU 密集型任务
        _ = computeSomething()
    }
}

func computeSomething() int {
    result := 0
    for i := 0; i < 1000000; i++ {
        result += i * i
    }
    return result
}

func main() {
    runtime.GOMAXPROCS(runtime.NumCPU()) // 使用所有 CPU 核心
    var wg sync.WaitGroup
    numCPU := runtime.NumCPU()

    for i := 0; i < numCPU; i++ {
        wg.Add(1)
        go simulateCPUUsage(&wg)
    }

    fmt.Println("模拟 CPU 负载...")
    time.Sleep(10 * time.Second) // 模拟 10 秒
    fmt.Println("停止模拟 CPU 负载...")
    //  退出所有 goroutine，这里省略了优雅退出的代码，实际应用中需要实现优雅退出
    // 比如使用 context.WithCancel
    wg.Wait()
}

context.WithCancel

tc

os/exec

package main

import (
    "fmt"
    "os/exec"
    "time"
)

func simulateNetworkLatency(interfaceName string, latency string) error {
    // 清除之前的规则 (可选)
    cmdClear := exec.Command("sudo", "tc", "qdisc", "del", "dev", interfaceName, "root")
    err := cmdClear.Run()
    if err != nil {
        fmt.Printf("清除规则失败，可能不存在: %v\n", err) // 不一定有规则，所以不算是错误
    }

    // 添加网络延迟规则
    cmdAdd := exec.Command("sudo", "tc", "qdisc", "add", "dev", interfaceName, "root", "netem", "delay", latency)
    output, err := cmdAdd.CombinedOutput()
    if err != nil {
        return fmt.Errorf("添加网络延迟规则失败: %v, output: %s", err, string(output))
    }

    fmt.Printf("成功添加网络延迟 %s 到接口 %s\n", latency, interfaceName)
    return nil
}

func removeNetworkLatency(interfaceName string) error {
    cmd := exec.Command("sudo", "tc", "qdisc", "del", "dev", interfaceName, "root")
    output, err := cmd.CombinedOutput()
    if err != nil {
        return fmt.Errorf("移除网络延迟规则失败: %v, output: %s", err, string(output))
    }

    fmt.Printf("成功移除接口 %s 的网络延迟\n", interfaceName)
    return nil
}

func main() {
    interfaceName := "eth0" // 替换为你的网络接口名称
    latency := "100ms"       // 延迟时间

    err := simulateNetworkLatency(interfaceName, latency)
    if err != nil {
        fmt.Println("模拟网络延迟失败:", err)
        return
    }

    fmt.Println("模拟网络延迟 10 秒...")
    time.Sleep(10 * time.Second)

    err = removeNetworkLatency(interfaceName)
    if err != nil {
        fmt.Println("移除网络延迟失败:", err)
        return
    }

    fmt.Println("网络延迟已移除")
}

tc

interfaceName