Go语言os/exec包执行外部命令后环境变量变更的捕获与处理

#!/bin/bash

# 模拟子进程修改环境变量
export MY_VAR="modified_by_child_$(date +%s)" # 修改现有变量
export NEW_VAR="hello_from_child"           # 添加新变量
export ANOTHER_VAR="some_other_value"

# 打印出我们关心的环境变量，每行一个 KEY=VALUE 格式
echo "MY_VAR=$MY_VAR"
echo "NEW_VAR=$NEW_VAR"
echo "ANOTHER_VAR=$ANOTHER_VAR"

# 如果需要，也可以打印所有环境变量 (可能会包含不必要的输出)
# env

package main

import (
    "bufio"
    "bytes"
    "fmt"
    "io"
    "os"
    "os/exec"
    "strings"
)

func main() {
    // 定义子进程的初始环境变量（可选，但通常需要传递一些父进程的环境）
    // 这里我们只传递PATH，并设置一个初始的MY_VAR值
    initialEnv := []string{
        "PATH=" + os.Getenv("PATH"), // 确保子进程能找到命令
        "MY_VAR=initial_value_from_parent", // 父进程设置的初始值
    }

    // 创建一个命令对象，执行我们的Shell脚本
    cmd := exec.Command("./child_process.sh")
    cmd.Env = initialEnv // 为子进程设置其环境变量

    // 创建一个缓冲区来捕获子进程的标准输出
    var stdoutBuf bytes.Buffer
    cmd.Stdout = &stdoutBuf
    cmd.Stderr = os.Stderr // 将子进程的标准错误直接输出到父进程的标准错误

    fmt.Println("正在执行子进程...")
    err := cmd.Run() // 执行命令并等待其完成
    if err != nil {
        fmt.Printf("执行命令出错: %v\n", err)
        return
    }

    fmt.Println("子进程执行完毕。正在捕获环境变量变更...")

    // 解析子进程的标准输出，提取环境变量
    modifiedEnv := make(map[string]string)
    scanner := bufio.NewScanner(&stdoutBuf) // 使用 bufio.Scanner 逐行读取输出
    for scanner.Scan() {
        line := scanner.Text()
        parts := strings.SplitN(line, "=", 2) // 按第一个等号分割 KEY=VALUE
        if len(parts) == 2 {
            modifiedEnv[parts[0]] = parts[1]
        }
    }
    if err := scanner.Err(); err != nil {
        fmt.Printf("读取子进程输出时出错: %v\n", err)
    }

    fmt.Println("\n从子进程捕获的环境变量:")
    for k, v := range modifiedEnv {
        fmt.Printf("%s = %s\n", k, v)
    }

    // 演示如何使用这些捕获到的环境变量
    fmt.Println("\n模拟后续操作中使用捕获到的环境变量:")
    if val, ok := modifiedEnv["MY_VAR"]; ok {
        fmt.Printf("  MY_VAR 的最新值: %s\n", val)
    }
    if val, ok := modifiedEnv["NEW_VAR"]; ok {
        fmt.Printf("  NEW_VAR 的值: %s\n", val)
    }
    // 实际应用中，你可以将这些变量用于后续的 exec.Command 调用，
    // 或者更新当前父进程的环境（通过 os.Setenv，但这只影响当前进程及其未来的子进程）。
}

正在执行子进程...
子进程执行完毕。正在捕获环境变量变更...

从子进程捕获的环境变量:
MY_VAR = modified_by_child_1678886400
NEW_VAR = hello_from_child
ANOTHER_VAR = some_other_value

模拟后续操作中使用捕获到的环境变量:
  MY_VAR 的最新值: modified_by_child_1678886400
  NEW_VAR 的值: hello_from_child