Go语言os/exec包执行外部命令后环境状态捕获机制解析

package main

import (
    "bytes"
    "fmt"
    "log"
    "os"
    "os/exec"
    "strings"
)

func main() {
    // 1. 定义一个模拟修改环境变量的Bash脚本
    // 脚本会设置或修改MY_VAR和ANOTHER_VAR，然后打印所有环境变量
    scriptContent := `#!/bin/bash
# 确保脚本是可执行的
set -e

# 修改或设置环境变量
export MY_VAR="modified_value_by_child"
export ANOTHER_VAR="new_value_from_child"

echo "--- Child Process Environment ---"
# 打印所有环境变量，每行一个 KEY=VALUE 格式
env
echo "--- Child Process End ---"
`
    // 将脚本内容写入临时文件，并赋予执行权限
    scriptPath := "./temp_env_script.sh"
    err := os.WriteFile(scriptPath, []byte(scriptContent), 0755)
    if err != nil {
        log.Fatalf("无法创建脚本文件: %v", err)
    }
    defer os.Remove(scriptPath) // 确保脚本文件在程序结束时被删除

    fmt.Println("--- 父进程启动时的相关环境变量 ---")
    // 打印父进程中可能存在的MY_VAR和ANOTHER_VAR，用于对比
    fmt.Printf("父进程 MY_VAR: %s\n", os.Getenv("MY_VAR"))
    fmt.Printf("父进程 ANOTHER_VAR: %s\n", os.Getenv("ANOTHER_VAR"))
    fmt.Println("---------------------------------")

    // 2. 准备执行外部命令
    // 使用Bash解释器执行脚本，确保脚本的执行环境一致
    cmd := exec.Command("/bin/bash", scriptPath)

    // 可以选择性地为子进程设置初始环境
    // cmd.Env = append(os.Environ(), "INITIAL_CHILD_VAR=initial")

    // 捕获子进程的标准输出
    var stdout bytes.Buffer
    cmd.Stdout = &stdout
    // 将子进程的错误输出重定向到父进程的stderr，便于调试
    cmd.Stderr = os.Stderr

    fmt.Println("\n--- 执行外部命令 ---")
    err = cmd.Run() // 运行命令并等待其完成
    if err != nil {
        log.Fatalf("命令执行失败: %v, 输出: %s", err, stdout.String())
    }
    fmt.Println("外部命令执行完成。")
    fmt.Println("--------------------")

    // 3. 解析外部命令的输出以捕获环境变化
    fmt.Println("\n--- 捕获到的外部命令环境 ---")
    capturedEnv := make(map[string]string)
    outputLines := strings.Split(stdout.String(), "\n")

    // 查找并解析子进程输出的环境变量部分
    inEnvSection := false
    for _, line := range outputLines {
        if strings.Contains(line, "--- Child Process Environment ---") {
            inEnvSection = true
            continue
        }
        if strings.Contains(line, "--- Child Process End ---") {
            inEnvSection = false
            break
        }
        if inEnvSection && strings.Contains(line, "=") {
            parts := strings.SplitN(line, "=", 2)
            if len(parts) == 2 {
                capturedEnv[parts[0]] = parts[1]
            }
        }
    }

    // 打印捕获到的特定环境变量
    if val, ok := capturedEnv["MY_VAR"]; ok {
        fmt.Printf("捕获到 MY_VAR: %s\n", val)
    } else {
        fmt.Println("MY_VAR 未在子进程输出中捕获到")
    }
    if val, ok := capturedEnv["ANOTHER_VAR"]; ok {
        fmt.Printf("捕获到 ANOTHER_VAR: %s\n", val)
    } else {
        fmt.Println("ANOTHER_VAR 未在子进程输出中捕获到")
    }
    fmt.Println("--------------------------")

    fmt.Println("\n--- 验证父进程环境未受影响 ---")
    // 再次打印父进程中的环境变量，验证其未被子进程修改
    fmt.Printf("父进程 MY_VAR: %s\n", os.Getenv("MY_VAR"))
    fmt.Printf("父进程 ANOTHER_VAR: %s\n", os.Getenv("ANOTHER_VAR"))
    fmt.Println("----------------------------")
}