From c305b09fbd20849a39631ddd43980cb09e1805e9 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Mon, 04 Jul 2022 17:10:45 +0800 Subject: [PATCH] fix seeColumnInfo --- src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java | 33 ++++++++++++++++++++++++++++++++- 1 files changed, 32 insertions(+), 1 deletions(-) diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java index f1795c0..f1e072f 100644 --- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java +++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CmsTemplateServiceImpl.java @@ -4,6 +4,9 @@ import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.xcong.farmer.cms.common.exception.GlobalException; +import com.xcong.farmer.cms.common.utils.FileUtils; +import com.xcong.farmer.cms.configurations.GlobalExceptionHandler; import com.xcong.farmer.cms.core.template.TemplateConfiguration; import com.xcong.farmer.cms.modules.system.dto.TemplateListDto; import com.xcong.farmer.cms.modules.system.entity.CmsTemplateEntity; @@ -13,6 +16,12 @@ import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.web.multipart.MultipartFile; + +import java.io.File; +import java.io.IOException; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; /** * @author wzy @@ -24,10 +33,32 @@ @Autowired private TemplateConfiguration cfg; + private List<String> fileSuffix = Arrays.asList(".zip", ".html"); + @Override - public void updateTemplate(MultipartFile file) { + public void updateTemplate(MultipartFile upload) { String templatePath = cfg.templatePath; String staticPath = cfg.staticPath; + + String filename = upload.getOriginalFilename(); + String suffix = filename.substring(filename.lastIndexOf(".")); + + if (!fileSuffix.contains(suffix)) { + throw new GlobalException("请上传正确格式文件"); + } + + try { + if (".zip".equals(suffix)) { + String path = FileUtils.path(templatePath, filename); + System.out.println(path); + File file = new File(path); + upload.transferTo(file); + FileUtils.zipUpload(file, templatePath, staticPath); + } + } catch (IOException e) { + e.printStackTrace(); + } + System.out.println(suffix); } @Override -- Gitblit v1.9.1