113 lines
4.2 KiB
Java
113 lines
4.2 KiB
Java
package io.dataease.service.staticResource;
|
|
|
|
import cn.hutool.core.codec.Base64Decoder;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
|
import com.google.gson.Gson;
|
|
import io.dataease.commons.exception.DEException;
|
|
import io.dataease.commons.utils.FileUtils;
|
|
import io.dataease.commons.utils.LogUtil;
|
|
import io.dataease.commons.utils.StaticResourceUtils;
|
|
import io.dataease.controller.request.resource.StaticResourceRequest;
|
|
import io.dataease.exception.DataEaseException;
|
|
import org.apache.commons.lang3.StringUtils;
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.util.Assert;
|
|
import org.springframework.util.FileCopyUtils;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import javax.imageio.ImageIO;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* Author: wangjiahao
|
|
* Date: 2022/4/24
|
|
* Description:
|
|
*/
|
|
@Service
|
|
public class StaticResourceService {
|
|
|
|
private final Path staticDir = Paths.get("/opt/dataease/data/static-resource/");
|
|
|
|
public void upload(String fileId, MultipartFile file) {
|
|
// check if the path is valid (not outside staticDir)
|
|
Assert.notNull(file, "Multipart file must not be null");
|
|
try {
|
|
if (!isImage(file)) {
|
|
DEException.throwException("Multipart file must be image");
|
|
}
|
|
String originName = file.getOriginalFilename();
|
|
String newFileName = fileId + originName.substring(originName.lastIndexOf("."), originName.length());
|
|
Path uploadPath = Paths.get(staticDir.toString(), newFileName);
|
|
// create dir is absent
|
|
FileUtils.createIfAbsent(Paths.get(staticDir.toString()));
|
|
Files.createFile(uploadPath);
|
|
file.transferTo(uploadPath);
|
|
} catch (IOException e) {
|
|
LogUtil.error("文件上传失败", e);
|
|
DataEaseException.throwException("文件上传失败");
|
|
} catch (Exception e) {
|
|
DataEaseException.throwException(e);
|
|
}
|
|
}
|
|
|
|
private boolean isImage(MultipartFile file) {
|
|
BufferedImage image = null;
|
|
try (InputStream input = file.getInputStream()) {
|
|
image = ImageIO.read(input);
|
|
} catch (IOException e) {
|
|
LogUtil.error(e.getMessage(), e);
|
|
return false;
|
|
}
|
|
if (image == null || image.getWidth() <= 0 || image.getHeight() <= 0) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public void saveFilesToServe(String staticResource) {
|
|
Gson gson = new Gson();
|
|
if (StringUtils.isNotEmpty(staticResource)) {
|
|
Map<String, String> resource = gson.fromJson(staticResource, Map.class);
|
|
for (Map.Entry<String, String> entry : resource.entrySet()) {
|
|
String path = entry.getKey();
|
|
String fileName = path.substring(path.lastIndexOf("/") + 1, path.length());
|
|
saveSingleFileToServe(fileName, entry.getValue());
|
|
}
|
|
}
|
|
}
|
|
|
|
public void saveSingleFileToServe(String fileName, String content) {
|
|
Path uploadPath = Paths.get(staticDir.toString(), fileName);
|
|
try {
|
|
if (uploadPath.toFile().exists()) {
|
|
LogUtil.info("file exists");
|
|
} else {
|
|
if (StringUtils.isNotEmpty(content)) {
|
|
Files.createFile(uploadPath);
|
|
FileCopyUtils.copy(Base64Decoder.decode(content), Files.newOutputStream(uploadPath));
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
LogUtil.error("template static resource save error" + e.getMessage());
|
|
}
|
|
}
|
|
|
|
public Map<String, String> findResourceAsBase64(StaticResourceRequest resourceRequest) {
|
|
Map<String, String> result = new HashMap<>();
|
|
if (CollectionUtil.isNotEmpty(resourceRequest.getResourcePathList())) {
|
|
for (String path : resourceRequest.getResourcePathList()) {
|
|
String value = StaticResourceUtils.getImgFileToBase64(path.substring(path.lastIndexOf("/") + 1, path.length()));
|
|
result.put(path, value);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
}
|