本文详解在 go 中模拟 bash 的 `source` 行为——即执行脚本并将其导出的环境变量注入当前进程，涵盖直接调用、环境捕获解析、安全注意事项及实用封装建议。
在 Go 中无法直接通过 exec.Command("bash", "source", script) 实现 source，因为 source 是 Bash 内置命令，不是独立可执行文件。但可通过组合 Bash 解释器与 -c 参数间接执行，并配合环境变量提取，达到“真正加载脚本环境”的效果。
✅ 推荐方案：执行 + 解析 env 输出
核心思路是：
- 启动子 shell 执行 source script.sh && env；
- 捕获其标准输出；
- 解析 KEY=VALUE 格式的环境变量行；
- 使用 os.Setenv() 将关键变量写入当前 Go 进程环境。
以下是一个健壮、可复用的示例（已适配 Go 1.16+，使用 os.ReadFile 替代已弃用的 ioutil.ReadFile）：
package main
import (
"bufio"
"bytes"
"fmt"
"os"
"os/exec"
"strings"
)
// SourceShellScript 执行指定 shell 脚本并将其 export 的变量注入当前进程环境
func SourceShellScript(scriptPath string) error {
cmd := exec.Command("bash", "-c", fmt.Sprintf("source %q 2>/dev/null && env", scriptPath))
output, err := cmd.CombinedOutput()
if err != nil {
return fmt.Errorf("failed to source %s: %w\noutput: %s", scriptPath, err, string(output))
}
scanner := bufio.NewScanner(bytes.NewReader(output))
for scanner.Scan() {
line := strings.TrimSpace(scanner.Text())
if line == "" || strings.HasPrefix(line, "#") {
continue
}
parts := strings.SplitN(line, "=", 2)
if len(parts) == 2 {
key := strings.TrimSpace(parts[0])
value := strings.TrimSpace(parts[1])
// 仅设置合法的环境变量名（避免空键或非法字符）
if key != "" && strings.IndexFunc(key, func(r rune) bool {
return !((r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') || r == '_' || (r >= '0' && r <= '9'))
}) == -1 {
os.Setenv(key, value)
}
}
}
if err := scanner.Err(); err != nil {
return fmt.Errorf("error scanning env output: %w", err)
}
return nil
}
func main() {
// 示例：生成临时测试脚本
script := `export HELLO="Go sourced me!"
export VERSION=1.2.3
export PATH="/opt/bin:$PATH"`
err := os.WriteFile("test_env.sh", []byte(script), 0755)
if err != nil {
panic(err)
}
defer os.Remove("test_env.sh")
// 执行 source 并注入环境
if err := SourceShellScript("test_env.sh"); err != nil {
panic(err)
}
fmt.Println("HELLO =", os.Getenv("HELLO")) // 输出：Go sourced me!
fmt.Println("VERSION =", os.Getenv("VERSION")) // 输出：1.2.3
fmt.Println("PATH contains /opt/bin:", strings.Contains(os.Getenv("PATH"), "/opt/bin"))
}⚠️ 关键注意事项
- 安全性：scriptPath 必须可信。拼接进 bash -c 的字符串若含用户输入，极易引发命令注入（如 scriptPath = "foo.sh; rm -rf /"）。生产环境应校验路径合法性（如 filepath.Clean() + 白名单目录），或改用绝对路径 + os.Stat 验证。
- 变量作用域：source 仅影响子 shell；本方案通过解析 env 并调用 os.Setenv，将变量显式写入 Go 主进程，但不会影响后续 fork 出的子进程的默认环境（除非显式传入 cmd.Env）。
- 非 export 变量无效：Shell 脚本中未 export 的变量不会出现在 env 输出中，因此不会被加载。
- 编码与换行：确保脚本为 UTF-8 编码，且无 Windows 风格 CRLF（可能干扰解析）。
- 错误处理：务必检查 CombinedOutput() 的错误和输出内容，source 失败时 env 仍会输出当前环境，易造成误判。
? 进阶建议
- 对于复杂场景（如需支持 .bashrc 风格初始化、函数导入、别名等），推荐使用 go-basher 库，它提供更完整的 shell 交互抽象。
- 若只需读取少量变量，可改用 bash -c "source ... && printf '%s\0' \$VAR1 \$VAR2" 配合 strings.Split() 提取，减少解析开销。
- 在容器或 CI 环境中，优先考虑将环境变量预置到 os.Environ() 或通过配置文件管理，而非动态 source，以提升可维护性与可观测性。
掌握此模式后，你便能在 Go 应用中无缝集成传统 Shell 生态（如加载 .env、初始化工具链、复用现有部署脚本），兼顾现代语言的安全性与 Shell 的灵活性。
