bugfix: 二进制环境使用NFS作为存储后端时分发本地文件失败 TencentBlueKing#1779

jsonwan · Feb 28, 2023 · c512a0e · c512a0e
1 parent 7f1bbd2
commit c512a0e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...rc/main/java/com/tencent/bk/job/execute/engine/prepare/local/LocalFilePrepareService.java b/...rc/main/java/com/tencent/bk/job/execute/engine/prepare/local/LocalFilePrepareService.java
@@ -89,6 +89,8 @@ public void prepareLocalFilesAsync(
         List<FileSourceDTO> fileSourceList,
         LocalFilePrepareTaskResultHandler resultHandler
     ) {
+        fillLocalFileSourceHost(fileSourceList, stepInstanceId);
+
         if (!JobConstants.FILE_STORAGE_BACKEND_ARTIFACTORY.equals(
             localFileConfigForExecute.getStorageBackend()
         )) {
@@ -117,6 +119,16 @@ public void prepareLocalFilesAsync(
         task.execute();
     }
 
+    private void fillLocalFileSourceHost(List<FileSourceDTO> fileSourceList, long stepInstanceId) {
+        fileSourceList.forEach(fileSourceDTO -> {
+            if (fileSourceDTO.getFileType() == TaskFileTypeEnum.LOCAL.getType() || fileSourceDTO.isLocalUpload()) {
+                fileSourceDTO.setServers(agentService.getLocalServersDTO());
+            }
+        });
+        // 更新本地文件任务内容
+        taskInstanceService.updateResolvedSourceFile(stepInstanceId, fileSourceList);
+    }
+
     public void clearPreparedTmpFile(long stepInstanceId) {
         // 本地文件暂不支持实时清理，依赖定时清理
     }