Go程序中处理混合命令行参数：flag 包与位置参数的最佳实践

go run main.go http://example.com --m=2 --strat=par

go run main.go --m=2 --strat=par http://example.com

package main

import (
    "flag"
    "fmt"
    "log"
    "os"
    // ... 其他导入，如 "webcrawler/crawler"
)

func main() {
    // 错误示范：在解析flag之前尝试获取os.Args[1]
    // 此时os.Args[1]可能是一个flag，而不是期望的URL
    if len(os.Args) < 2 {
        log.Fatal("Url must be provided as first argument")
    }

    strategy := flag.String("strat", "par", "par for parallel OR seq for sequential crawling strategy")
    routineMultiplier := flag.Int("m", 1, "Goroutine multiplier. Default 1x logical CPUs. Only works in parallel strategy")

    // 此时 os.Args[1] 的内容取决于用户命令行输入的顺序
    // 如果用户输入 `go run main.go --m=2 http://example.com`，os.Args[1] 就是 "--m=2"
    // 导致 url 变量获取到错误的值
    url := os.Args[1] // 错误示范：过早使用os.Args[1]

    flag.Parse() // 在此之后，flag才会被解析，但url变量已经错误赋值

    // ... 后续逻辑使用url, *strategy, *routineMultiplier
    fmt.Printf("URL: %s, Strategy: %s, Multiplier: %d\n", url, *strategy, *routineMultiplier)
}

package main

import (
    "flag"
    "fmt"
    "log"
    "os"
    // "webcrawler/crawler" // 假设存在这些包，此处为示例注释
    // "webcrawler/model"
    // "webcrawler/urlutils"
)

func main() {
    // 1. 定义所有命令行标志
    strategy := flag.String("strat", "par", "par for parallel OR seq for sequential crawling strategy")
    routineMultiplier := flag.Int("m", 1, "Goroutine multiplier. Default 1x logical CPUs. Only works in parallel strategy")

    // 2. 调用 flag.Parse() 来解析标志
    // 这会将所有定义的标志从os.Args中解析出来，
    // 并将剩余的非标志参数保留在flag.Args()中。
    // 无论标志和位置参数在命令行中的顺序如何，flag.Parse()都能正确处理。
    flag.Parse()

    // 3. 使用 flag.Args() 获取所有非标志参数（即位置参数）
    args := flag.Args()

    // 4. 校验位置参数的数量
    if len(args) != 1 {
        // 根据需求，URL是强制性的一个位置参数
        fmt.Println("Usage: go run main.go [OPTIONS] <URL>")
        flag.PrintDefaults() // 打印所有标志的默认值和说明，帮助用户理解
        log.Fatalf("Error: Exactly one argument (URL) is required, but got %d.", len(args))
    }

    // 5. 获取并使用位置参数
    targetURL := args[0]

    // 以下是假设的爬虫初始化和执行逻辑，仅作示例
    // page := model.NewBasePage(targetURL)
    // urlutils.BASE_URL = targetURL // 设置全局或配置
    // pages := crawler.Crawl(&page, *strategy, *routineMultiplier)
    // fmt.Printf("Crawled: %d pages\n", len(pages))

    fmt.Printf("Parsed arguments:\n")
    fmt.Printf("  Target URL: %s\n", targetURL)
    fmt.Printf("  Strategy: %s\n", *strategy)
    fmt.Printf("  Routine Multiplier: %d\n", *routineMultiplier)

    // 示例：根据策略值执行不同逻辑
    if *strategy == "par" {
        fmt.Println("  Executing parallel crawl...")
    } else if *strategy == "seq" {
        fmt.Println("  Executing sequential crawl...")
    } else {
        log.Fatalf("  Invalid strategy: %s. Must be 'par' or 'seq'.", *strategy)
    }
}

# 正常情况：URL在flags之后
go run main.go --m=5 --strat=par http://example.com/path

# 正常情况：URL在flags之前
go run main.go http://example.com/another --strat=seq --m=2

# 错误情况：缺少URL
go run main.go --m=5 --strat=par

# 错误情况：提供了多个URL
go run main.go http://example.com/one http://example.com/two --m=1