Go Humanize 怎样解析用户输入的容量字符串

如果配置文件里写的是 42 MB、42 mib 或 1.5GB，不要先用字符串替换单位再自己乘 1000。Go 的 github.com/dustin/go-humanize 提供了 ParseBytes，可以直接把容量字符串转换成字节数；它返回 uint64 和 error，因此调用方可以把格式错误和超大值挡在业务逻辑外。

官方地址：https://pkg.go.dev/github.com/dustin/go-humanize

最小写法是 bytes, err := humanize.ParseBytes(input)。其中 MB 按十进制容量解释，MiB 按 1024 进制解释；例如官方文档给出的 42 MB 是 42000000，42 mib 是 44040192。

用 ParseBytes 接收用户输入

先把依赖加入 Go Modules，再把解析函数放在配置读取边界。这样后面的限流、缓存或上传逻辑只接收已经转换好的字节数，不需要重复理解单位。

# 安装容量解析依赖，版本由当前模块的依赖策略决定
go get github.com/dustin/go-humanize

package main

import (
    "fmt"
    "os"

    "github.com/dustin/go-humanize"
)

func parseLimit(input string) (uint64, error) {
    // 把单位解析集中在入口，业务层只处理字节数。
    size, err := humanize.ParseBytes(input)
    if err != nil {
        return 0, fmt.Errorf("解析容量 %q: %w", input, err)
    }
    return size, nil
}

func main() {
    // 示例输入来自命令行；没有参数时给出明确的使用提示。
    if len(os.Args) != 2 {
        fmt.Println("用法: go run . ")
        return
    }

    size, err := parseLimit(os.Args[1])
    if err != nil {
        // 不把非法输入当成 0，避免错误配置悄悄放宽限制。
        fmt.Println(err)
        return
    }
    fmt.Printf("%s = %d 字节\\n", os.Args[1], size)
}

解析函数只做一件事：把第三方库的错误补上输入值，再交给上层处理。生产代码可以把 size 保存到配置结构体里，后续比较文件大小时直接使用字节数。

MB、MiB 和没有后缀的数字有什么区别

ParseBytes 会先读取数字部分，再根据后缀选择容量基数。十进制单位包括 kB、MB、GB；二进制单位包括 KiB、MiB、GiB。库的文档也展示了小写 mib 的写法，调用时不必强制用户输入固定大小写。

最容易出错的是把“磁盘厂商的十进制容量”和“内存或系统工具常见的二进制容量”混为一谈。建议在配置文档中直接写明采用 MB 还是 MiB，不要让调用者根据显示习惯猜测。

错误和超大值要怎样处理

输入不是“数字 + 已知单位”时，ParseBytes 会返回错误；乘上单位后超过 uint64 可表达范围时，也不会静默截断。业务侧应保留错误，而不是把失败结果当作零继续运行。

func parseOptionalLimit(input string) (uint64, bool) {
    // 空配置表示“不设置限制”，与非法配置严格区分。
    if input == "" {
        return 0, false
    }

    size, err := humanize.ParseBytes(input)
    if err != nil {
        // 调用方可以把这条错误映射成配置文件的行号提示。
        return 0, false
    }
    return size, true
}

如果产品允许空值，应在调用 ParseBytes 之前单独定义空值语义；不要依赖库的报错去推断“未配置”。同样，1.5GB 这类小数输入要结合业务的取整要求处理，容量限制通常应在进入比较逻辑前明确采用向下取整还是拒绝小数。

什么时候改用 ParseBigBytes

ParseBytes 的返回类型是 uint64，适合常规文件、缓存和请求体上限。如果配置的是超大容量，或者后续计算必须保留大整数精度，可以使用 ParseBigBytes，它返回 *big.Int 和 error。

func parseHugeSize(input string) (*big.Int, error) {
    // big.Int 用于不适合放进 uint64 的容量计算，不要先转 float64。
    value, err := humanize.ParseBigBytes(input)
    if err != nil {
        return nil, fmt.Errorf("解析大容量 %q: %w", input, err)
    }
    return value, nil
}

选择标准很简单：能明确落在 uint64 范围内就用 ParseBytes，接口更轻；需要大整数精度时用 ParseBigBytes，并继续遵守 big.Int 的指针生命周期和比较方式。

相关问题

用户输入的单位大小写必须固定吗？

不必把大小写写死在业务层。解析函数会把单位部分转成小写后匹配，文档示例也包含 mib；但产品文档仍建议统一展示一种写法，减少人工配置时的歧义。

为什么不直接用 strconv.ParseFloat？

ParseFloat 只能处理数字，不能替你定义 MB 与 MiB 的基数、单位表和溢出边界。容量字符串应先由 ParseBytes 或 ParseBigBytes 完成单位解析，再进入业务比较。