💩 update getFileExt

2021-11-11 15:19:57 +08:00 · 2021-11-11 15:19:57 +08:00 · 687f95c7e9
commit 687f95c7e9
parent c6226d4a09
1 changed files with 5 additions and 4 deletions
--- a/parser.go
+++ b/parser.go
@ -7,6 +7,7 @@ import (
 	"math/rand"
 	"mime/multipart"
 	"net/http"
+	"path/filepath"
 	"reflect"
 	"strconv"
 	"strings"
@ -63,11 +64,11 @@ func GetFileContentType(file multipart.File) (string, error) {
 }

 func getFileExt(name string) string {
-	results := strings.Split(name, ".")
-	if len(results) == 1 {
-		return "undefined"
+	ext := filepath.Ext(name)
+	if len(ext) == 0 {
+		ext = "undefined"
 	}
-	return results[len(results)-1]
+	return ext
 }

 const (