18 行
345 B
Go
18 行
345 B
Go
package util
|
|
|
|
import (
|
|
"errors"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func ValidateFilename(filename string) error {
|
|
switch filepath.Ext(strings.ToLower(filename)) {
|
|
case ".pdf", ".doc", ".jpg", ".gif", ".png", ".jpeg",
|
|
".docx", ".xls", ".xlsx", ".zip", ".rar", ".7z", ".txt":
|
|
return nil
|
|
default:
|
|
return errors.New("扩展名不支持")
|
|
}
|
|
}
|