fix: 业务组件新增、更新时保存封面图片

业务组件新增、更新时保存封面图片

DataRoom/dataroom-core/src/main/java/com/gccloud/dataroom/core/module/biz/component/service/impl/BizComponentServiceImpl.java

@@ -2,6 +2,7 @@ package com.gccloud.dataroom.core.module.biz.component.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.gccloud.dataroom.core.config.DataRoomConfig;
 import com.gccloud.dataroom.core.module.biz.component.dao.BizComponentDao;
 import com.gccloud.dataroom.core.module.biz.component.dto.BizComponentSearchDTO;
 import com.gccloud.dataroom.core.module.biz.component.entity.BizComponentEntity;
@@ -9,9 +10,16 @@ import com.gccloud.dataroom.core.module.biz.component.service.IBizComponentServi
 import com.gccloud.dataroom.core.utils.CodeGenerateUtils;
 import com.gccloud.common.exception.GlobalException;
 import com.gccloud.common.vo.PageVO;
+import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
+import org.apache.commons.lang3.exception.ExceptionUtils;
 import org.springframework.stereotype.Service;
 
+import javax.annotation.Resource;
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.util.Base64;
 import java.util.List;
 
 /**
@@ -19,9 +27,13 @@ import java.util.List;
  * @version 1.0
  * @date 2023/6/5 13:35
  */
+@Slf4j
 @Service
 public class BizComponentServiceImpl extends ServiceImpl<BizComponentDao, BizComponentEntity> implements IBizComponentService {
 
+    @Resource
+    private DataRoomConfig bigScreenConfig;
+
     @Override
     public PageVO<BizComponentEntity> getPage(BizComponentSearchDTO searchDTO) {
         LambdaQueryWrapper<BizComponentEntity> queryWrapper = new LambdaQueryWrapper<>();
@@ -61,6 +73,11 @@ public class BizComponentServiceImpl extends ServiceImpl<BizComponentDao, BizCom
         }
         String code = CodeGenerateUtils.generate("bizComponent");
         entity.setCode(code);
+        if (StringUtils.isNotBlank(entity.getCoverPicture())) {
+            String base64Str = entity.getCoverPicture();
+            String fileUrl = this.saveCoverPicture(base64Str, entity.getCode());
+            entity.setCoverPicture(fileUrl);
+        }
         this.save(entity);
         return code;
     }
@@ -71,9 +88,55 @@ public class BizComponentServiceImpl extends ServiceImpl<BizComponentDao, BizCom
         if (repeat) {
             throw new GlobalException("组件名称重复");
         }
+        if (StringUtils.isNotBlank(entity.getCoverPicture())) {
+            String base64Str = entity.getCoverPicture();
+            String fileUrl = this.saveCoverPicture(base64Str, entity.getCode());
+            entity.setCoverPicture(fileUrl);
+        }
         this.updateById(entity);
     }
 
+
+    /**
+     * 将base64字符串转为图片文件存储
+     *
+     * @param base64String
+     * @param fileName
+     * @return
+     */
+    private String saveCoverPicture(String base64String, String fileName) {
+        String fileUrl = "";
+        if (StringUtils.isBlank(base64String)) {
+            return fileUrl;
+        }
+        try {
+            // 去除base64字符串前缀，从初始位置，到逗号位置
+            base64String = base64String.substring(base64String.indexOf(",") + 1);
+            // 解码base64字符串
+            byte[] imageBytes = Base64.getDecoder().decode(base64String);
+            String basePath = bigScreenConfig.getFile().getBasePath();
+            // 不是/结尾，加上/
+            if (!basePath.endsWith("/") || !basePath.endsWith("\\")) {
+                basePath += File.separator;
+            }
+            // 检查目录是否存在，不存在则创建
+            File file = new File(basePath + "cover");
+            if (!file.exists()) {
+                file.mkdirs();
+            }
+            // 保存为图片文件
+            String filePath = basePath + "cover" + File.separator + fileName + ".png";
+            fileUrl = "cover" + File.separator + fileName + ".png";
+            FileOutputStream outputStream = new FileOutputStream(filePath);
+            outputStream.write(imageBytes);
+            outputStream.close();
+            log.info("大屏封面保存至：{}", filePath);
+        } catch (IOException e) {
+            log.error(ExceptionUtils.getStackTrace(e));
+        }
+        return fileUrl;
+    }
+
     @Override
     public String copy(String code) {
         BizComponentEntity copyFrom = this.getInfoByCode(code);

DataRoom/dataroom-core/src/main/java/com/gccloud/dataroom/core/module/manage/service/impl/DataRoomPageServiceImpl.java

@@ -66,7 +66,7 @@ public class DataRoomPageServiceImpl extends ServiceImpl<PageDao, PageEntity> im
             }
             chart.setCode(CodeGenerateUtils.generate(chart.getType() == null ? "chart" : chart.getType()));
         }
-        if (StringUtils.isBlank(bigScreenPageDTO.getCoverPicture())) {
+        if (StringUtils.isNotBlank(bigScreenPageDTO.getCoverPicture())) {
             String base64Str = bigScreenPageDTO.getCoverPicture();
             String fileUrl = this.saveCoverPicture(base64Str, bigScreenPageDTO.getCode());
             bigScreenPageDTO.setCoverPicture(fileUrl);