校验文件名称和类型的正确性

controller/image.go

@@ -41,6 +41,13 @@ func (a *ImageController) UploadNiupic(c *gin.Context) {
 		fileName := util.Md5Encode(fmt.Sprintf("%s%s", f.Filename, time.Now().String()))
 		fildDir := fmt.Sprintf("%s%d%s_", "upload_tmp_", time.Now().Year(), time.Now().Month().String())
 		subfilepath := fmt.Sprintf("%s%s%s", fildDir, fileName, fileExt)
+
+		err := c.SaveUploadedFile(f, subfilepath)
+		if err != nil {
+			log.Println(err)
+			resultRes.Error(1, "图片保存失败")
+		} else {
+			//校验文件名称和类型的正确性
 			if strings.Contains(strings.ToUpper(fileExt), "JPEG") || strings.Contains(strings.ToUpper(fileExt), "JPG") || strings.Contains(strings.ToUpper(fileExt), "PNG") || strings.Contains(strings.ToUpper(fileExt), "GIF") {
 				_, err := util.CheckImageFile(subfilepath)
 				if err != nil {
@@ -52,11 +59,6 @@ func (a *ImageController) UploadNiupic(c *gin.Context) {
 				}
 			}
 
-		err := c.SaveUploadedFile(f, subfilepath)
-		if err != nil {
-			log.Println(err)
-			resultRes.Error(1, "图片保存失败")
-		} else {
 			url, err := service.UploadNiupic(subfilepath)
 			if err != nil {
 				resultRes.Error(1, "图片上传失败")

util/imageCheck_test.go

@@ -14,6 +14,16 @@ import (
 *
 */
 func TestCheckImageFile(t *testing.T) {
+	path := "D:\\Desktop\\icon\\m_487c6a80-458c-11eb-a0b7-0016a.jpg"
+	_, err := CheckImageFile(path)
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+	fmt.Println("success")
+}
+
+func TestCheckImageFil2e(t *testing.T) {
 	path := "D:\\Desktop\\icon\\avatar.png"
 	_, err := CheckImageFile(path)
 	if err != nil {
@@ -22,9 +32,17 @@ func TestCheckImageFile(t *testing.T) {
 	}
 	fmt.Println("success")
 }
-func TestIndex(t *testing.T) {
+
+func TestIndex2(t *testing.T) {
 	path := "343455.sdsdsdsd.jpg"
 	lastIndex := strings.LastIndex(path, ".")
 	extend := path[lastIndex+1:]
 	fmt.Println(extend)
 }
+
+func TestIndex(t *testing.T) {
+	path := "D:\\Desktop\\icon\\m_487c6a80-458c-11eb-a0b7-0016a.jpg"
+	lastIndex := strings.LastIndex(path, ".")
+	extend := path[lastIndex+1:]
+	fmt.Println(extend)
+}