Go语言中动态管理命令行参数：使用map存储flag值指针的实践

package main

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

func main() {
    // 模拟命令行参数，例如：go run your_program.go -flagA "valueA" -flagB "valueB"
    // 为了演示，我们直接设置os.Args，实际运行中用户会通过命令行输入
    // 如果不设置，可以使用默认的flag.CommandLine，或者通过传入参数给Parse()
    originalArgs := os.Args
    defer func() { os.Args = originalArgs }() // 确保在函数退出时恢复原始os.Args
    os.Args = []string{"program", "-flagA", "valueA_from_args", "-flagB", "valueB_from_args"}

    // 1. 创建一个新的FlagSet
    // FlagSet名称通常用于帮助信息，flag.ExitOnError表示解析失败时退出程序
    fs := flag.NewFlagSet("myCommand", flag.ExitOnError)

    // 2. 定义需要动态管理的flag名称列表
    requiredFlags := []string{"flagA", "flagB", "optionalFlag"}

    // 3. 创建一个map来存储flag名称到其*string指针的映射
    flags := make(map[string]*string)

    // 4. 遍历requiredFlags，为每个flag定义并存储其指针
    for _, f := range requiredFlags {
        // fs.String() 返回一个 *string，我们直接将其存储到map中
        // 第一个参数是flag名称，第二个是默认值，第三个是使用说明
        flags[f] = fs.String(f, "default_"+f, "This is "+f)
    }

    // 5. 解析命令行参数
    // 这一步是关键，它会根据os.Args（或传入的参数）更新所有已定义的flag指针所指向的值
    err := fs.Parse(os.Args[1:]) // os.Args[1:] 排除程序名
    if err != nil {
        fmt.Printf("Error parsing flags: %v\n", err)
        os.Exit(1)
    }

    // 6. 访问解析后的flag值
    fmt.Println("--- Parsed Flag Values ---")
    for name, ptr := range flags {
        // 在这里，ptr是一个*string。我们需要解引用它来获取实际的字符串值
        fmt.Printf("%s: %s\n", name, *ptr)
    }

    // 验证某个特定flag的值
    if valA, ok := flags["flagA"]; ok {
        fmt.Printf("Specific flagA value: %s\n", *valA)
    }

    // 尝试访问一个未通过命令行提供的可选flag，它将是默认值
    if valOptional, ok := flags["optionalFlag"]; ok {
        fmt.Printf("Specific optionalFlag value (not provided): %s\n", *valOptional)
    }

    // 演示如果map存储的是值而不是指针会发生什么（错误示例）
    fmt.Println("\n--- Incorrect Approach (Storing Values) ---")
    incorrectFlags := make(map[string]string)
    for _, f := range requiredFlags {
        // 错误：这里存储的是fs.String()返回的*string的当前值（默认值），而不是指针
        incorrectFlags[f] = *fs.String(f+"_incorrect", "default_incorrect_"+f, "This is "+f+"_incorrect")
    }
    // 即使再次解析，incorrectFlags中的值也不会更新
    // fs.Parse(os.Args[1:]) // 再次解析无意义，因为值已经拷贝
    for name, val := range incorrectFlags {
        fmt.Printf("%s: %s (will be default/empty)\n", name, val)
    }
}