test/DWSurvey
1
0
Fork 0
mirror of https://gitee.com/wkeyuan/DWSurvey.git synced 2024-12-01 19:49:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化上传文件类型过滤的方法

Browse Source
This commit is contained in:
wkeyuan 2023-04-28 15:04:24 +08:00
parent e9f328b936
commit ad7d4d8927
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/main/java/com/baidu/ueditor/upload/StorageManager.java
View File

@ -48,7 +48,6 @@ public class StorageManager {
public static State saveFileByInputStream(InputStream is, String path, public static State saveFileByInputStream(InputStream is, String path,
long maxSize) { long maxSize) {
System.out.println("saveFileByInputStream");
BaseState validateState = isUserUpFileType(is,path.substring(path.lastIndexOf("."))); BaseState validateState = isUserUpFileType(is,path.substring(path.lastIndexOf(".")));
if(!validateState.isSuccess()) return validateState; if(!validateState.isSuccess()) return validateState;
State state = new BaseState(false, AppInfo.IO_ERROR); State state = new BaseState(false, AppInfo.IO_ERROR);

13
src/main/java/net/diaowen/common/plugs/file/FileMagicUtils.java
View File

@ -26,8 +26,17 @@ public class FileMagicUtils {
//根据文件获取对应的文件类型 //根据文件获取对应的文件类型
public static FileMagic getFileMagic(File inp, String fileSuffix) throws Exception { public static FileMagic getFileMagic(File inp, String fileSuffix) throws Exception {
FileInputStream fis = new FileInputStream(inp); FileMagic fileMagic = null;
return getFileMagic(fis,fileSuffix); FileInputStream fis = null;
try{
fis = new FileInputStream(inp);
fileMagic = getFileMagic(fis,fileSuffix);
}catch (Exception e){
e.printStackTrace();
}finally {
if (fis!=null) fis.close();
}
return fileMagic;
} }
//切换到使用最新的tika验测 //切换到使用最新的tika验测