Bläddra i källkod

研报上传文件判断

xyxie 11 månader sedan
förälder
incheckning
4671474a3a
1 ändrade filer med 6 tillägg och 2 borttagningar
  1. 6 2
      controllers/report.go

+ 6 - 2
controllers/report.go

@@ -1286,8 +1286,12 @@ func (this *ReportUploadCommonController) UploadImg() {
 	}
 	pass := filetype.IsImage(fileData)
 	if !pass {
-		err = fmt.Errorf("文件格式有误")
-		return
+		kind, _ := filetype.Match(fileData)
+		if kind.Extension != "pdf" {
+			err = fmt.Errorf("文件格式有误")
+			return
+		}
+		fmt.Printf("File type: %s. MIME: %s\n", kind.Extension, kind.MIME.Value)
 	}
 
 	ext := path.Ext(h.Filename)