| @@ -80,5 +80,6 @@ public interface DatasetVersionDao { | |||||
| */ | */ | ||||
| int deleteById(Integer id); | int deleteById(Integer id); | ||||
| List<DatasetVersion> queryByDatasetId(Integer datasetId); | |||||
| } | } | ||||
| @@ -80,5 +80,7 @@ public interface ModelsVersionDao { | |||||
| */ | */ | ||||
| int deleteById(Integer id); | int deleteById(Integer id); | ||||
| List<ModelsVersion> queryByModelsId(Integer modelsId); | |||||
| } | } | ||||
| @@ -7,6 +7,8 @@ import com.ruoyi.platform.domain.DatasetVersion; | |||||
| import org.springframework.data.domain.Page; | import org.springframework.data.domain.Page; | ||||
| import org.springframework.data.domain.PageRequest; | import org.springframework.data.domain.PageRequest; | ||||
| import java.util.List; | |||||
| /** | /** | ||||
| * (DatasetVersion)表服务接口 | * (DatasetVersion)表服务接口 | ||||
| * | * | ||||
| @@ -58,4 +60,5 @@ public interface DatasetVersionService { | |||||
| String removeById(Integer id); | String removeById(Integer id); | ||||
| List<DatasetVersion> queryByDatasetId(Integer datasetId); | |||||
| } | } | ||||
| @@ -7,6 +7,8 @@ import com.ruoyi.platform.domain.ModelsVersion; | |||||
| import org.springframework.data.domain.Page; | import org.springframework.data.domain.Page; | ||||
| import org.springframework.data.domain.PageRequest; | import org.springframework.data.domain.PageRequest; | ||||
| import java.util.List; | |||||
| /** | /** | ||||
| * (ModelsVersion)表服务接口 | * (ModelsVersion)表服务接口 | ||||
| * | * | ||||
| @@ -58,4 +60,5 @@ public interface ModelsVersionService { | |||||
| String removeById(Integer id); | String removeById(Integer id); | ||||
| List<ModelsVersion> queryByModelsId(Integer modelsId); | |||||
| } | } | ||||
| @@ -30,6 +30,7 @@ import java.io.ByteArrayOutputStream; | |||||
| import java.io.InputStream; | import java.io.InputStream; | ||||
| import java.text.SimpleDateFormat; | import java.text.SimpleDateFormat; | ||||
| import java.util.Date; | import java.util.Date; | ||||
| import java.util.List; | |||||
| /** | /** | ||||
| * (Dataset)表服务实现类 | * (Dataset)表服务实现类 | ||||
| @@ -42,7 +43,6 @@ public class DatasetServiceImpl implements DatasetService { | |||||
| @Resource | @Resource | ||||
| private DatasetDao datasetDao; | private DatasetDao datasetDao; | ||||
| private DatasetService datasetService; | |||||
| @Resource | @Resource | ||||
| private DatasetVersionDao datasetVersionDao; | private DatasetVersionDao datasetVersionDao; | ||||
| @@ -158,6 +158,9 @@ public class DatasetServiceImpl implements DatasetService { | |||||
| if (!(StringUtils.equals(username,"admin") || StringUtils.equals(username,createdBy))){ | if (!(StringUtils.equals(username,"admin") || StringUtils.equals(username,createdBy))){ | ||||
| return "无权限删除该数据集"; | return "无权限删除该数据集"; | ||||
| } | } | ||||
| if (datasetVersionService.queryByDatasetId(id).size()>0){ | |||||
| return "请先删除该数据集的版本文件"; | |||||
| } | |||||
| dataset.setState(0); | dataset.setState(0); | ||||
| return this.datasetDao.update(dataset)>0?"删除数据集成功":"删除数据集失败"; | return this.datasetDao.update(dataset)>0?"删除数据集成功":"删除数据集失败"; | ||||
| @@ -13,6 +13,7 @@ import org.springframework.stereotype.Service; | |||||
| import javax.annotation.Resource; | import javax.annotation.Resource; | ||||
| import java.util.Date; | import java.util.Date; | ||||
| import java.util.List; | |||||
| /** | /** | ||||
| * (DatasetVersion)表服务实现类 | * (DatasetVersion)表服务实现类 | ||||
| @@ -117,4 +118,9 @@ public class DatasetVersionServiceImpl implements DatasetVersionService { | |||||
| return this.datasetVersionDao.update(datasetVersion)>0?"删除数据集版本成功":"删除数据集版本失败"; | return this.datasetVersionDao.update(datasetVersion)>0?"删除数据集版本成功":"删除数据集版本失败"; | ||||
| } | } | ||||
| @Override | |||||
| public List<DatasetVersion> queryByDatasetId(Integer datasetId) { | |||||
| return datasetVersionDao.queryByDatasetId(datasetId); | |||||
| } | |||||
| } | } | ||||
| @@ -6,6 +6,7 @@ import com.ruoyi.platform.domain.ModelsVersion; | |||||
| import com.ruoyi.platform.mapper.ModelsDao; | import com.ruoyi.platform.mapper.ModelsDao; | ||||
| import com.ruoyi.platform.mapper.ModelsVersionDao; | import com.ruoyi.platform.mapper.ModelsVersionDao; | ||||
| import com.ruoyi.platform.service.ModelsService; | import com.ruoyi.platform.service.ModelsService; | ||||
| import com.ruoyi.platform.service.ModelsVersionService; | |||||
| import com.ruoyi.platform.utils.MinioUtil; | import com.ruoyi.platform.utils.MinioUtil; | ||||
| import com.ruoyi.system.api.model.LoginUser; | import com.ruoyi.system.api.model.LoginUser; | ||||
| import io.minio.MinioClient; | import io.minio.MinioClient; | ||||
| @@ -43,6 +44,8 @@ public class ModelsServiceImpl implements ModelsService { | |||||
| private ModelsVersionDao modelsVersionDao; | private ModelsVersionDao modelsVersionDao; | ||||
| private MinioClient minioClient; | private MinioClient minioClient; | ||||
| @Resource | |||||
| private ModelsVersionService modelsVersionService; | |||||
| @Value("${minio.endpoint}") | @Value("${minio.endpoint}") | ||||
| private String minioEndpoint; | private String minioEndpoint; | ||||
| @@ -148,7 +151,9 @@ public class ModelsServiceImpl implements ModelsService { | |||||
| if (!(StringUtils.equals(username,"admin") || StringUtils.equals(username,createdBy))){ | if (!(StringUtils.equals(username,"admin") || StringUtils.equals(username,createdBy))){ | ||||
| return "无权限删除该模型"; | return "无权限删除该模型"; | ||||
| } | } | ||||
| if (modelsVersionService.queryByModelsId(id).size()>0){ | |||||
| return "请先删除该数据集的版本文件"; | |||||
| } | |||||
| models.setState(0); | models.setState(0); | ||||
| return this.modelsDao.update(models)>0?"删除成功":"删除失败"; | return this.modelsDao.update(models)>0?"删除成功":"删除失败"; | ||||
| } | } | ||||
| @@ -14,6 +14,7 @@ import org.springframework.stereotype.Service; | |||||
| import javax.annotation.Resource; | import javax.annotation.Resource; | ||||
| import java.util.Date; | import java.util.Date; | ||||
| import java.util.List; | |||||
| /** | /** | ||||
| * (ModelsVersion)表服务实现类 | * (ModelsVersion)表服务实现类 | ||||
| @@ -122,4 +123,9 @@ public class ModelsVersionServiceImpl implements ModelsVersionService { | |||||
| modelsVersion.setState(0); | modelsVersion.setState(0); | ||||
| return this.modelsVersionDao.update(modelsVersion)>0?"删除成功":"删除失败"; | return this.modelsVersionDao.update(modelsVersion)>0?"删除成功":"删除失败"; | ||||
| } | } | ||||
| @Override | |||||
| public List<ModelsVersion> queryByModelsId(Integer modelsId){ | |||||
| return modelsVersionDao.queryByModelsId(modelsId); | |||||
| } | |||||
| } | } | ||||
| @@ -18,7 +18,12 @@ | |||||
| <result property="state" column="state" jdbcType="INTEGER"/> | <result property="state" column="state" jdbcType="INTEGER"/> | ||||
| </resultMap> | </resultMap> | ||||
| <select id="queryByDatasetId" resultMap="DatasetVersionMap"> | |||||
| select | |||||
| id,dataset_id,version,url,file_name,file_size,available_cluster,status,create_by,create_time,update_by,update_time,state | |||||
| from dataset_version | |||||
| where id = #{id} and state = 1 | |||||
| </select> | |||||
| <!--查询单个--> | <!--查询单个--> | ||||
| @@ -26,7 +31,7 @@ | |||||
| select | select | ||||
| id,dataset_id,version,url,file_name,file_size,available_cluster,status,create_by,create_time,update_by,update_time,state | id,dataset_id,version,url,file_name,file_size,available_cluster,status,create_by,create_time,update_by,update_time,state | ||||
| from dataset_version | from dataset_version | ||||
| where id = #{id} and state = 1 | |||||
| where dataset_id = #{datasetId} and state = 1 | |||||
| </select> | </select> | ||||
| <!--查询指定行数据--> | <!--查询指定行数据--> | ||||
| @@ -16,7 +16,12 @@ | |||||
| <result property="updateTime" column="update_time" jdbcType="TIMESTAMP"/> | <result property="updateTime" column="update_time" jdbcType="TIMESTAMP"/> | ||||
| <result property="state" column="state" jdbcType="INTEGER"/> | <result property="state" column="state" jdbcType="INTEGER"/> | ||||
| </resultMap> | </resultMap> | ||||
| <select id="queryByModelsId" resultMap="ModelsVersionMap"> | |||||
| select | |||||
| id,models_id,version,url,file_name,file_size,status,create_by,create_time,update_by,update_time,state | |||||
| from models_version | |||||
| where models_id = #{modelsId} and state = 1 | |||||
| </select> | |||||
| <!--查询单个--> | <!--查询单个--> | ||||
| <select id="queryById" resultMap="ModelsVersionMap"> | <select id="queryById" resultMap="ModelsVersionMap"> | ||||
| select | select | ||||