修改镜像删除mapper文件

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/mapper/ImageDao.java

@@ -80,5 +80,7 @@ public interface ImageDao {
     int deleteById(Integer id);
 
     List<Image> queryByName(String name);
+
+    Image getByName(String name);
 }
 

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/ImageService.java

@@ -26,6 +26,20 @@ public interface ImageService {
      */
     Image queryById(Integer id);
 
+
+    Page<Image> queryByName(String name);
+
+    /**
+     * 通过名字精确查询单条数据
+     *
+     * @param name 名字
+     * @return 实例对象
+     */
+
+    Image getByName(String name);
+
+
+
     /**
      * 分页查询
      *
@@ -59,9 +73,11 @@ public interface ImageService {
      */
     boolean deleteById(Integer id);
 
-    String removeById(Integer id);
+    String removeById(Integer id) throws Exception;
+
+
+
 
-    Page<Image> queryByName(String name);
 
     String insertImageAndVersion(ImageVo imageVo) throws Exception;
 
@@ -77,4 +93,6 @@ public interface ImageService {
     Map<String, String> createImageFromNet(String imageName, String imageTag, String NetPath) throws Exception;
     Map<String, String> uploadImageFiles(MultipartFile file) throws Exception;
 
+
+
 }

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ImageServiceImpl.java

@@ -88,6 +88,8 @@ public class ImageServiceImpl implements ImageService {
         return new PageImpl<>(this.imageDao.queryAllByLimit(image, pageRequest), pageRequest, total);
     }
 
+
+
     /**
      * 新增数据
      *
@@ -138,10 +140,10 @@ public class ImageServiceImpl implements ImageService {
     }
 
     @Override
-    public String removeById(Integer id) {
+    public String removeById(Integer id) throws Exception {
         Image image = this.imageDao.queryById(id);
         if (image == null){
-            return "镜像不存在";
+            throw new Exception("镜像不存在");
         }
 
         //判断权限，只有admin和创建者本身可以删除该数据集
@@ -150,11 +152,11 @@ public class ImageServiceImpl implements ImageService {
 
 
         String createdBy = image.getCreateBy();
-        if (!(StringUtils.equals(username,"admin") || StringUtils.equals(username,createdBy))){
-            return "无权限删除该镜像";
+        if (!(StringUtils.equals(username,"admin") || !StringUtils.equals(username,createdBy))){
+            throw new Exception("无权限删除该镜像");
         }
         if (!imageVersionService.queryByImageId(id).isEmpty()){
-            return "请先删除该镜像下的版本文件";
+            throw new Exception("请先删除该镜像下的版本文件");
         }
         image.setState(0);
         return this.imageDao.update(image)>0?"删除成功":"删除失败";
@@ -167,26 +169,46 @@ public class ImageServiceImpl implements ImageService {
         return new PageImpl<>(this.imageDao.queryByName(name));
     }
 
+
+    /**
+     * 通过名字精确查询镜像
+     *
+     * @param name 名字
+     * @return 镜像
+     */
+    @Override
+    public Image getByName(String name) {
+        return this.imageDao.getByName(name);
+    }
+
     @Override
     @Transactional
     public String insertImageAndVersion(ImageVo imageVo) throws Exception {
-        Image image = new Image();
-        image.setName(imageVo.getName());
-        image.setDescription(imageVo.getDescription());
-        image.setImageType(imageVo.getImageType());
-        Image imageInsert = this.insert(image);
-        if (imageInsert == null){
-            throw new Exception("新增镜像失败");
+        Image existingImage = getByName(imageVo.getName());
+        Image imageToUse;
+        if(existingImage == null) {
+            // 如果不存在相同名称的镜像，则创建新的镜像记录
+            Image newImage = new Image();
+            newImage.setName(imageVo.getName());
+            newImage.setDescription(imageVo.getDescription());
+            newImage.setImageType(imageVo.getImageType());
+            imageToUse = this.insert(newImage);
+            if (imageToUse == null) {
+                throw new Exception("新增镜像失败");
+            }
+        }else{
+            // 如果已存在相同名称的镜像，使用已存在的镜像
+            imageToUse = existingImage;
         }
         ImageVersion imageVersion = new ImageVersion();
-        imageVersion.setImageId(imageInsert.getId());
+        imageVersion.setImageId(imageToUse.getId());
         imageVersion.setVersion(imageVo.getVersion());
         imageVersion.setTagName(imageVo.getTagName());
         imageVersion.setFileSize(imageVo.getFileSize());
         imageVersion.setStatus("building");
         ImageVersion imageVersionInsert = this.imageVersionService.insert(imageVersion);
         if (imageVersionInsert == null) {
-            throw new Exception("新增镜像失败");
+            throw new Exception("新增镜像版本失败");
         }
         // 使用CompletableFuture异步执行不同的镜像构建逻辑
         CompletableFuture.supplyAsync(() -> {
@@ -206,7 +228,7 @@ public class ImageServiceImpl implements ImageService {
         }).thenAccept(resultMap ->{
             try {
                 String imageUrl = resultMap.get("url");
-                String fileSize = resultMap.get("filesize");
+                String fileSize = resultMap.get("fileSize");
                 imageVersion.setUrl(imageUrl);
                 imageVersion.setFileSize(fileSize);
                 imageVersion.setStatus("available");
@@ -240,7 +262,7 @@ public class ImageServiceImpl implements ImageService {
         String logs2 = k8sClientUtil.executeCommand(pod,"docker pull "+ netPath);
         //   在容器里执行 docker tag name:tag nexus3.kube-system.svc:8083/imageName:imageTag
         if (StringUtils.isNoneBlank(logs2)){
-            String substring = logs2.substring(logs2.lastIndexOf(harborUrl), logs2.length());
+            String substring = logs2.substring(logs2.lastIndexOf(harborUrl));
             String cleanedString = substring.replaceAll("(\\r|\\n)", "");
             String tagCmd = "docker tag " + cleanedString + " " + harborUrl + "/" + repository + "/" + username + "/" + imageName + ":" + imageTag;
             String imageUrl = harborUrl + "/" + repository + "/" + username + "/" + imageName + ":" + imageTag;
@@ -282,7 +304,7 @@ public class ImageServiceImpl implements ImageService {
         String logs2 = k8sClientUtil.executeCommand(pod,"docker load -i "+filePath);
         //   在容器里执行 docker tag name:tag nexus3.kube-system.svc:8083/imageName:imageTag
         if (StringUtils.isNoneBlank(logs2)){
-            String substring = logs2.substring(logs2.lastIndexOf(harborUrl), logs2.length());
+            String substring = logs2.substring(logs2.lastIndexOf(harborUrl));
             String cleanedString = substring.replaceAll("(\\r|\\n)", "");
             String tagCmd = "docker tag " + cleanedString + " " + harborUrl + "/" + repository + "/" + username + "/" + imageName + ":" + imageTag;
             String imageUrl = harborUrl + "/" + repository + "/" + username + "/" + imageName + ":" + imageTag;

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/K8sClientUtil.java

@@ -410,13 +410,12 @@ public class K8sClientUtil {
             int exitValue = proc.waitFor();
             if (exitValue != 0) {
                 // 如果进程的退出值不为0，表示命令执行失败
-                throw new RuntimeException("命令执行失败，退出值：" + exitValue);
+                throw new RuntimeException("容器中命令执行失败，退出值：" + exitValue);
             }
-
             return builder.toString();
         } catch (Exception e) {
-            log.error("执行命令异常", e);
-            throw new RuntimeException("执行命令异常");
+            log.error("容器执行命令异常", e);
+            throw new RuntimeException("容器执行命令异常");
         }
     }
 

ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/ImageDaoMapper.xml

@@ -26,7 +26,7 @@
     <select id="queryAllByLimit" resultMap="ImageMap">
         select
         img.id, img.name, img.description, img.image_type, img.create_by, img.create_time, img.update_by, img.update_time, img.state,
-        (SELECT COUNT(*) FROM image_version WHERE image_version.image_id = img.id) as versionCount
+        (SELECT COUNT(*) FROM image_version WHERE image_version.image_id = img.id and image_version.state = 1) as versionCount
         from image img
         <where>
             img.state = 1
@@ -75,6 +75,18 @@
         </where>
     </select>
 
+    <select id="getByName" resultMap="ImageMap">
+        select
+        id, name, description, image_type, create_by, create_time, update_by, update_time, state
+        from image
+        <where>
+            state = 1
+            <if test="name != null and name != ''">
+                and name = #{name}
+            </if>
+        </where>
+    </select>
+
 
     <!--统计总行数-->
     <select id="count" resultType="java.lang.Long">
@@ -111,6 +123,7 @@
             </if>
         </where>
     </select>
+
     <!--新增所有列-->
     <insert id="insert" keyProperty="id" useGeneratedKeys="true">
         insert into image(name,description,image_type,create_by,create_time,update_by,update_time,state)
@@ -161,7 +174,7 @@ name = values(name)description = values(description)image_type = values(image_ty
             <if test="image.updateTime != null">
                 update_time = #{image.updateTime},
             </if>
-            <if test="image.state != null and image.state != ''">
+            <if test="image.state != null">
                 state = #{image.state},
             </if>
         </set>