Chang lue Tsen
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package tool
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"strings"
|
|
)
|
|
|
|
// IsValidImageSize 检查base64图片是否有效且未超出大小限制
|
|
// base64Str: base64编码的图片字符串
|
|
// maxSizeKB: 最大允许大小（KB），int64类型
|
|
// 返回: bool - true表示图片有效且未超限，false表示无效或超限
|
|
func IsValidImageSize(base64Str string, maxSizeKB int64) bool {
|
|
// 输入验证
|
|
if base64Str == "" || maxSizeKB < 0 {
|
|
return false
|
|
}
|
|
|
|
// 提取纯base64数据
|
|
data := base64Str
|
|
if idx := strings.Index(base64Str, ","); idx != -1 {
|
|
data = base64Str[idx+1:]
|
|
}
|
|
|
|
// 快速估算大小（避免完整解码）
|
|
// base64编码后每3字节原始数据变成4字节，解码后大小约为输入长度的3/4
|
|
approxSizeBytes := int64(len(data)) * 3 / 4
|
|
approxSizeKB := approxSizeBytes / 1024
|
|
|
|
// 如果估算大小已超限，无需解码
|
|
if approxSizeKB > maxSizeKB {
|
|
return false
|
|
}
|
|
|
|
// 验证base64有效性
|
|
decoded, err := base64.StdEncoding.DecodeString(data)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
// 精确检查大小
|
|
return int64(len(decoded))/1024 <= maxSizeKB
|
|
}
|