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
|
||
}
|