Browse Source

Merge branch 'dev-opt' of https://gitlink.org.cn/ci4s/ci4sManagement-cloud into dev-opt

dev-opt-homepage
chenzhihang 7 months ago
parent
commit
a117d92e7e
2 changed files with 18 additions and 6 deletions
  1. +16
    -4
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java
  2. +2
    -2
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/FileUtil.java

+ 16
- 4
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java View File

@@ -18,6 +18,7 @@ import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.ByteArrayResource;
import org.springframework.core.io.InputStreamResource;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
@@ -517,10 +518,21 @@ public class NewDatasetServiceImpl implements NewDatasetService {
headers.put("Authorization", "Token " + labelDatasetVersionVo.getToken());
InputStream inputStream = HttpUtils.getIntputStream(labelloaclUrl + "/api/projects/" + labelDatasetVersionVo.getProjectId() + "/export?exportType=" + labelDatasetVersionVo.getExportType(), headers);

// 2. 打包成zip包
// String zipFileName = labelDatasetVersionVo.getName() + "_" + labelDatasetVersionVo.getVersion() + ".zip";
String zipFileName = labelDatasetVersionVo.getName() + "_" + labelDatasetVersionVo.getVersion();
MultipartFile[] files = FileUtil.toMultipartFiles(inputStream, zipFileName);
// 2. 直接保存zip文件,不要重新打包
String zipFileName = labelDatasetVersionVo.getName() + "_" + labelDatasetVersionVo.getVersion() + ".zip";

// 将InputStream转换为byte数组
ByteArrayOutputStream baos = new ByteArrayOutputStream();
byte[] buffer = new byte[1024];
int length;
while ((length = inputStream.read(buffer)) >= 0) {
baos.write(buffer, 0, length);
}
inputStream.close();

// 创建MultipartFile
ByteArrayResource resource = new ByteArrayResource(baos.toByteArray());
MultipartFile[] files = new MultipartFile[]{new FileUtil.CustomMultipartFile(resource, zipFileName)};

// 3. 上传到minio
List<Map<String, String>> maps = uploadDatasetlocal(files, UUID.randomUUID().toString());


+ 2
- 2
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/FileUtil.java View File

@@ -136,12 +136,12 @@ public class FileUtil {
return new MultipartFile[]{new CustomMultipartFile(resource, fileName)};
}

private static class CustomMultipartFile implements MultipartFile {
public static class CustomMultipartFile implements MultipartFile {

private final ByteArrayResource resource;
private final String fileName;

CustomMultipartFile(ByteArrayResource resource, String fileName) {
public CustomMultipartFile(ByteArrayResource resource, String fileName) {
this.resource = resource;
this.fileName = fileName;
}


Loading…
Cancel
Save