feature：增加数据配比功能

backend/services/data-collection-service/src/main/java/com/datamate/collection/interfaces/rest/CollectionTaskController.java

@@ -10,6 +10,7 @@
 import com.datamate.datamanagement.application.DatasetApplicationService;
 import com.datamate.datamanagement.domain.model.dataset.Dataset;
 import com.datamate.datamanagement.interfaces.converter.DatasetConverter;
+import com.datamate.datamanagement.interfaces.dto.DatasetResponse;
 import jakarta.validation.Valid;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -35,11 +36,13 @@ public class CollectionTaskController{
     public ResponseEntity<CollectionTaskResponse> createTask(@Valid @RequestBody CreateCollectionTaskRequest request) {
         CollectionTask task = CollectionTaskConverter.INSTANCE.toCollectionTask(request);
         String datasetId = null;
+        DatasetResponse dataset = null;
         if (Objects.nonNull(request.getDataset())) {
-            datasetId = datasetService.createDataset(request.getDataset()).getId();
+            dataset = DatasetConverter.INSTANCE.convertToResponse(datasetService.createDataset(request.getDataset()));
+            datasetId = dataset.getId();
         }
         CollectionTaskResponse response = CollectionTaskConverter.INSTANCE.toResponse(taskService.create(task, datasetId));
-        response.setDataset(DatasetConverter.INSTANCE.convertToResponse(datasetService.getDataset(datasetId)));
+        response.setDataset(dataset);
         return ResponseEntity.ok().body(response);
     }
 

backend/services/data-management-service/src/main/java/com/datamate/datamanagement/application/DatasetApplicationService.java

@@ -119,6 +119,8 @@ public void deleteDataset(String datasetId) {
     public Dataset getDataset(String datasetId) {
         Dataset dataset = datasetRepository.getById(datasetId);
         BusinessAssert.notNull(dataset, DataManagementErrorCode.DATASET_NOT_FOUND);
+        List<DatasetFile> datasetFiles = datasetFileRepository.findAllByDatasetId(datasetId);
+        dataset.setFiles(datasetFiles);
         return dataset;
     }
 

backend/services/data-management-service/src/main/java/com/datamate/datamanagement/application/DatasetFileApplicationService.java

@@ -102,6 +102,10 @@ public DatasetFile getDatasetFile(String datasetId, String fileId) {
     public void deleteDatasetFile(String datasetId, String fileId) {
         DatasetFile file = getDatasetFile(datasetId, fileId);
         Dataset dataset = datasetRepository.getById(datasetId);
+        dataset.setFiles(new ArrayList<>(Collections.singleton(file)));
+        datasetFileRepository.removeById(fileId);
+        dataset.removeFile(file);
+        datasetRepository.updateById(dataset);
         // 删除文件时，上传到数据集中的文件会同时删除数据库中的记录和文件系统中的文件，归集过来的文件仅删除数据库中的记录
         if (file.getFilePath().startsWith(dataset.getPath())) {
             try {
@@ -111,9 +115,6 @@ public void deleteDatasetFile(String datasetId, String fileId) {
                 throw BusinessException.of(SystemErrorCode.FILE_SYSTEM_ERROR);
             }
         }
-        datasetFileRepository.removeById(fileId);
-        dataset.removeFile(file);
-        datasetRepository.updateById(dataset);
     }
 
     /**

backend/services/data-management-service/src/main/java/com/datamate/datamanagement/application/FileMetadataService.java

@@ -110,7 +110,7 @@ private DatasetFile extractFileMetadata(String filePath, String datasetId) throw
                 .fileType(fileType)
                 .uploadTime(LocalDateTime.now())
                 .lastAccessTime(LocalDateTime.now())
-                .status("UPLOADED")
+                .status("ACTIVE")
                 .build();
     }
 

backend/services/data-management-service/src/main/java/com/datamate/datamanagement/domain/model/dataset/DatasetFile.java

@@ -2,9 +2,13 @@
 
 import com.baomidou.mybatisplus.annotation.TableId;
 import com.baomidou.mybatisplus.annotation.TableName;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import lombok.*;
+import org.apache.commons.lang3.StringUtils;
 
 import java.time.LocalDateTime;
+import java.util.Arrays;
+import java.util.Collections;
 import java.util.List;
 
 /**
@@ -25,11 +29,25 @@ public class DatasetFile {
     private String fileType; // JPG/PNG/DCM/TXT
     private Long fileSize; // bytes
     private String checkSum;
-    private List<String> tags;
+    private String tags;
     private String metadata;
     private String status; // UPLOADED, PROCESSING, COMPLETED, ERROR
     private LocalDateTime uploadTime;
     private LocalDateTime lastAccessTime;
     private LocalDateTime createdAt;
     private LocalDateTime updatedAt;
+
+    /**
+     * 解析标签
+     *
+     * @return 标签列表
+     */
+    public List<String> analyzeTag() {
+        try {
+            ObjectMapper mapper = new ObjectMapper();
+            return mapper.readValue(tags, List.class);
+        } catch (Exception e) {
+            return Collections.emptyList();
+        }
+    }
 }

backend/services/data-management-service/src/main/java/com/datamate/datamanagement/interfaces/converter/DatasetConverter.java

@@ -1,17 +1,24 @@
 package com.datamate.datamanagement.interfaces.converter;
 
+import com.datamate.common.infrastructure.exception.BusinessException;
+import com.datamate.common.infrastructure.exception.SystemErrorCode;
 import com.datamate.datamanagement.interfaces.dto.CreateDatasetRequest;
 import com.datamate.datamanagement.interfaces.dto.DatasetFileResponse;
 import com.datamate.datamanagement.interfaces.dto.DatasetResponse;
 import com.datamate.datamanagement.interfaces.dto.UploadFileRequest;
 import com.datamate.common.domain.model.ChunkUploadRequest;
 import com.datamate.datamanagement.domain.model.dataset.Dataset;
 import com.datamate.datamanagement.domain.model.dataset.DatasetFile;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import org.apache.commons.collections4.CollectionUtils;
 import org.mapstruct.Mapper;
 import org.mapstruct.Mapping;
+import org.mapstruct.Named;
 import org.mapstruct.factory.Mappers;
 
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * 数据集文件转换器
@@ -26,6 +33,7 @@ public interface DatasetConverter {
      */
     @Mapping(source = "sizeBytes", target = "totalSize")
     @Mapping(source = "path", target = "targetLocation")
+    @Mapping(source = "files", target = "distribution", qualifiedByName = "getDistribution")
     DatasetResponse convertToResponse(Dataset dataset);
 
     /**
@@ -49,4 +57,28 @@ public interface DatasetConverter {
      * 将数据集文件转换为响应
      */
     DatasetFileResponse convertToResponse(DatasetFile datasetFile);
+
+    /**
+     * 获取数据文件的标签分布
+     *
+     * @param datasetFiles 数据集文件
+     * @return 标签分布
+     */
+    @Named("getDistribution")
+    default Map<String, Long> getDistribution(List<DatasetFile> datasetFiles) {
+        Map<String, Long> distribution = new HashMap<>();
+        if (CollectionUtils.isEmpty(datasetFiles)) {
+            return distribution;
+        }
+        for (DatasetFile datasetFile : datasetFiles) {
+            List<String> tags = datasetFile.analyzeTag();
+            if (CollectionUtils.isEmpty(tags)) {
+                continue;
+            }
+            for (String tag : tags) {
+                distribution.put(tag, distribution.getOrDefault(tag, 0L) + 1);
+            }
+        }
+        return distribution;
+    }
 }

backend/services/data-management-service/src/main/java/com/datamate/datamanagement/interfaces/dto/DatasetResponse.java

@@ -5,6 +5,7 @@
 
 import java.time.LocalDateTime;
 import java.util.List;
+import java.util.Map;
 
 /**
  * 数据集响应DTO
@@ -42,8 +43,8 @@ public class DatasetResponse {
     private LocalDateTime updatedAt;
     /** 创建者 */
     private String createdBy;
-    /**
-     * 更新者
-     */
+    /** 更新者 */
     private String updatedBy;
+    /** 分布 */
+    private Map<String, Long> distribution ;
 }

frontend/src/mock/ratio.tsx

@@ -1,4 +1,4 @@
-import type { RatioTask } from "@/pages/RatioTask/ratio";
+import type { RatioTask } from "@/pages/RatioTask/ratio.model.ts";
 
 export const mockRatioTasks: RatioTask[] = [
   {

frontend/src/pages/DataManagement/dataset.model.ts

@@ -49,11 +49,13 @@ export interface Dataset {
   status: DatasetStatus;
   size?: string;
   itemCount?: number;
+  fileCount?: number;
   createdBy: string;
   createdAt: string;
   updatedAt: string;
   tags: string[];
   targetLocation?: string;
+  distribution?: Record<string, number>;
 }
 
 export interface TagItem {