Go语言中动态管理命令行参数：使用指针和Map存储FlagSet值

import (
    "flag"
    "strings"
)

// 假设 commands 是一个字符串切片，例如 []string{"command1"}
// requiredFlags 是需要定义的参数名，例如 []string{"flagA", "flagB"}

fs := flag.NewFlagSet(strings.Join(commands, " "), flag.ExitOnError)
requiredFlags := []string{"flagA", "flagB"}
flags := make(map[string]string) // 注意这里 map 的值类型是 string

for _, f := range requiredFlags {
    // 错误的做法：fs.String(f, "", "") 返回 *string，但这里对其进行了立即解引用
    // 此时 *fs.String(...) 得到的是默认值 ""，而不是一个可更新的引用
    flags[f] = *fs.String(f, "", "")
}

// 在此之后调用 fs.Parse(os.Args[1:])
// 即使解析成功，flags["flagA"] 的值仍然是 ""，因为它存储的是一个字符串副本

package main

import (
    "flag"
    "fmt"
    "os"
    "strings"
)

func main() {
    // 模拟命令和所需参数
    commands := []string{"my-command"}
    requiredFlags := []string{"flagA", "flagB", "output"}

    // 创建一个新的 FlagSet
    fs := flag.NewFlagSet(strings.Join(commands, " "), flag.ExitOnError)

    // 正确的做法：map 的值类型应为 *string，用于存储指针
    flags := make(map[string]*string)

    for _, f := range requiredFlags {
        // 将 fs.String() 返回的 *string 指针直接存储到 map 中
        // 此时 map[f] 存储的是一个内存地址，该地址在 fs.Parse() 后会被更新
        flags[f] = fs.String(f, "", fmt.Sprintf("Help message for %s", f))
    }

    // 模拟命令行参数，例如：go run your_program.go --flagA=valueA --flagB=valueB --output=file.txt
    // 为了在示例中运行，我们手动设置参数，实际应用中通常是 os.Args[1:]
    // fs.Parse(os.Args[1:])
    // 为了演示，我们直接传入模拟的参数切片
    err := fs.Parse([]string{"--flagA=hello", "--flagB=world", "--output=result.log"})
    if err != nil {
        fmt.Printf("Error parsing flags: %v\n", err)
        return
    }

    fmt.Println("FlagSet parsed successfully!")

    // 访问更新后的值：需要对 map 中存储的指针进行解引用
    fmt.Printf("Value of flagA: %s\n", *flags["flagA"])
    fmt.Printf("Value of flagB: %s\n", *flags["flagB"])
    fmt.Printf("Value of output: %s\n", *flags["output"])

    // 验证未定义的参数（如果存在）
    // 注意：如果 flagA 未在命令行中提供，*flags["flagA"] 将是其默认值 ""
    // 但如果 flags["nonExistentFlag"] 尝试访问，会引发运行时错误，
    // 因为 map 返回 nil，nil 指针解引用
    // 更好的做法是检查是否存在或使用默认值
    if val, ok := flags["flagA"]; ok {
        fmt.Printf("Safely accessed flagA: %s\n", *val)
    } else {
        fmt.Println("flagA not found in map (should not happen in this example)")
    }
}