lukeye/doc-parser-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
luke 2025-06-17 08:16:09 +08:00
parent 618652784e
commit 48ad6da53a
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/main/java/com/knowledge/base/infrastructure/north/controller/FileWriteController.java
View File

@ -6,6 +6,7 @@ import com.knowledge.base.application.service.DocAppService;
import com.knowledge.base.application.service.UserAppService; import com.knowledge.base.application.service.UserAppService;
import com.knowledge.base.domain.doc.service.iface.DocumentImporter; import com.knowledge.base.domain.doc.service.iface.DocumentImporter;
import com.knowledge.base.infrastructure.cache.FileCacheService; import com.knowledge.base.infrastructure.cache.FileCacheService;
import com.knowledge.base.infrastructure.config.ConstantConfig;
import com.knowledge.base.infrastructure.config.ObjectStorageProperties; import com.knowledge.base.infrastructure.config.ObjectStorageProperties;
import com.knowledge.base.infrastructure.north.dto.doc.OSRecordDTO; import com.knowledge.base.infrastructure.north.dto.doc.OSRecordDTO;
import com.knowledge.base.infrastructure.north.dto.user.UserDTO; import com.knowledge.base.infrastructure.north.dto.user.UserDTO;
@ -136,11 +137,13 @@ public class FileWriteController {
@PostMapping("/upload-files") @PostMapping("/upload-files")
public ResponseEntity<?> uploadFolderToOS( public ResponseEntity<?> uploadFolderToOS(
@CookieValue(value = ConstantConfig.COOKIE_KEY, required = false) String cookieToken,
@RequestHeader("Authorization") String token, @RequestHeader("Authorization") String token,
@RequestParam("files") MultipartFile[] files, @RequestParam("files") MultipartFile[] files,
@RequestParam("bucket") String bucket @RequestParam("bucket") String bucket
) { ) {
try { try {
token = StrUtil.isBlank(token) ? cookieToken : token;
objectStorageGateway.ensureBucketExists(bucket); objectStorageGateway.ensureBucketExists(bucket);
Map<String, String> filePathMap = new LinkedHashMap<>(); Map<String, String> filePathMap = new LinkedHashMap<>();
@ -170,9 +173,10 @@ public class FileWriteController {
// 异步写入 OSS 上传记录 // 异步写入 OSS 上传记录
String finalSafeName = safeName; String finalSafeName = safeName;
String finalToken = token;
ThreadPoolUtil.execute(() -> { ThreadPoolUtil.execute(() -> {
try { try {
Optional<UserTokenDTO> userTokenDTO = userAppService.findToken(token); Optional<UserTokenDTO> userTokenDTO = userAppService.findToken(finalToken);
if (userTokenDTO.isEmpty()) { if (userTokenDTO.isEmpty()) {
logger.warn("异步写入失败：无效 token"); logger.warn("异步写入失败：无效 token");
return; return;