附件上传修改

src/spiderutil/upload.go

@@ -69,12 +69,19 @@ func UploadFile(code, fileName, url string, bt []byte) (string, string, string,
 	} else {
 		newFileType = util.GetFileType(bt)
 	}
-	if !commUsedReg.MatchString(newFileType) {
-		if !commUsedReg.MatchString(oldFileType) { //都不是常见的附件不上传
+	fileName = name
+	if !commUsedReg.MatchString(newFileType) { //
+		if !commUsedReg.MatchString(oldFileType) { //都不是常见的附件类型不上传
 			logger.Error("上传文件出错，该文件类型禁止上传！", code, " upload file "+fileName, url)
 			return "", fileName, size, newFileType, ""
-		} else if newFileType != oldFileType && oldFileType != "" {
+		} else {
 			newFileType = oldFileType
+			fileName = name + "." + newFileType
+		}
+	} else {
+		fileName = name + "." + newFileType
+		if commUsedReg.MatchString(oldFileType) && newFileType != oldFileType {
+			logger.Info("附件类型不同：", newFileType, oldFileType, code, url)
 		}
 	}
 	//newFileType字节解析出来的文件类型不是常见类型时，替换成oldFileType
@@ -82,22 +89,22 @@ func UploadFile(code, fileName, url string, bt []byte) (string, string, string,
 	// 	newFileType = oldFileType
 	// }
 	//禁止上传
-	if filterTypeReg.MatchString(newFileType) {
-		if oldFileType != "doc" && oldFileType != "docx" {
-			fileName = name + "." + newFileType
-			logger.Error("上传文件出错，该文件类型禁止上传！", code, " upload file "+fileName, url)
-			return "", fileName, size, newFileType, ""
-		} else {
-			newFileType = oldFileType
-		}
-	}
-	if oldFileType != "" {
-		fileName = name + "." + oldFileType
-	} else if newFileType != "" {
-		fileName = name + "." + newFileType
-	} else {
-		fileName = name
-	}
+	// if filterTypeReg.MatchString(newFileType) {
+	// 	if oldFileType != "doc" && oldFileType != "docx" {
+	// 		fileName = name + "." + newFileType
+	// 		logger.Error("上传文件出错，该文件类型禁止上传！", code, " upload file "+fileName, url)
+	// 		return "", fileName, size, newFileType, ""
+	// 	} else {
+	// 		newFileType = oldFileType
+	// 	}
+	// }
+	// if oldFileType != "" {
+	// 	fileName = name + "." + oldFileType
+	// } else if newFileType != "" {
+	// 	fileName = name + "." + newFileType
+	// } else {
+	// 	fileName = name
+	// }
 	key := getHashKey(bt) + TypeByExt(fileName)
 	b, _ := OssPutObject(key, io.MultiReader(bs))
 	if b {