dpa55065
2019-07-06 15:27 阅读 74
已采纳

用类似bash的环境变量引用插入字符串

I have an input string for my Golang CLI tool with some references to environment variables in bash syntax ($VAR and ${VAR}), e.g.:

$HOME/somedir/${SOME_VARIABLE}dir/anotherdir-${ANOTHER_VARIABLE}

What is the most efficient way to interpolate this string by replacing environemnt variables references with actual values? For previous example it can be:

/home/user/somedir/4dir/anotherdir-5

if HOME=/home/user, SOME_VARIABLE=4 and ANOTHER_VARIABLE=5.

Right now I'm using something like:

func interpolate(str string) string {
        for _, e := range os.Environ() {
                parts := strings.SplitN(e, "=", 2)
                name, value := parts[0], parts[1]
                str = strings.Replace(str, "$" + name, value, -1)
        }
        return str
}

but this code doesn't handle ${} variables.

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享

1条回答 默认 最新

  • 已采纳
    droi5225 droi5225 2019-07-06 16:37

    Use os.ExpandEnv:

    s := os.ExpandEnv(
        "$HOME/somedir/${SOME_VARIABLE}dir/anotherdir-${ANOTHER_VARIABLE}")
    

    You can use os.Expand for the case where the values are not taken from environment variables:

    m := map[string]string{
        "HOME":             "/home/user",
        "SOME_VARIABLE":    "4",
        "ANOTHER_VARIABLE": "5",
    }
    s := os.Expand(
        "$HOME/somedir/${SOME_VARIABLE}dir/anotherdir-${ANOTHER_VARIABLE}", 
        func(s string) string { return m[s] })
    

    Run it in the playground.

    点赞 评论 复制链接分享

相关推荐