core-ai 文档解析限制文件后缀名

src/main/java/com/qmth/ops/biz/service/DocClientService.java

@@ -2,6 +2,8 @@ package com.qmth.ops.biz.service;
 
 import com.qmth.boot.core.ai.model.doc.ParseDocTask;
 import com.qmth.boot.core.ai.model.doc.ParseDocTaskResult;
+import com.qmth.boot.core.exception.StatusException;
+import com.qmth.boot.tools.io.FileUtils;
 import com.qmth.ops.biz.ai.client.DocApiClient;
 import com.qmth.ops.biz.ai.client.DocApiConfig;
 import com.qmth.ops.biz.ai.exception.DocClientNotFound;
@@ -63,6 +65,12 @@ public class DocClientService {
         if (defaultClient == null) {
             throw new DocClientNotFound();
         }
+
+        String fileSuffix = FileUtils.getFileSuffix(fileName);
+        if (!FileUtils.matchSuffix("(.txt|.pdf|.doc|.docx|.xls|.xlsx|.ppt|.pptx)$", fileSuffix)) {
+            throw new StatusException(fileName + "暂不支持该文件类型");
+        }
+
         return defaultClient.parseDocTask(fileData, fileName);
     }