修改3.0添加桌面图标不展示问题
parent
9195bda7cf
commit
ee5e1dbddd
@ -0,0 +1,178 @@
|
|||||||
|
package com.zhiyun.zhiyun03.utils.file;
|
||||||
|
|
||||||
|
import cn.hutool.core.util.CharsetUtil;
|
||||||
|
import cn.hutool.core.util.IdUtil;
|
||||||
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
import org.springframework.util.Assert;
|
||||||
|
import org.springframework.util.StringUtils;
|
||||||
|
import org.springframework.web.multipart.MultipartFile;
|
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletResponse;
|
||||||
|
import java.io.*;
|
||||||
|
import java.net.URLEncoder;
|
||||||
|
import java.nio.file.Files;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.Calendar;
|
||||||
|
import java.util.Date;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
public class LocalFileUtil implements IFileUtil{
|
||||||
|
|
||||||
|
private final static List<String> excludeSp = Arrays.asList("bin", "sh");
|
||||||
|
|
||||||
|
private final String localPath;
|
||||||
|
|
||||||
|
public LocalFileUtil(String localPath) {
|
||||||
|
this.localPath = localPath;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String upload(MultipartFile file) {
|
||||||
|
return upload(null, file);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String upload(String objectName, InputStream fileIn) {
|
||||||
|
Assert.notNull(fileIn, "文件不能为空");
|
||||||
|
Assert.hasText(objectName, "文件名为空");
|
||||||
|
Assert.isTrue(objectName.lastIndexOf(".") > 0, "文件名称需携带扩展后缀");
|
||||||
|
FileOutputStream out = null;
|
||||||
|
try {
|
||||||
|
String relativePath = getDiskRelativePath(objectName, null);
|
||||||
|
File file = new File(getFullPath(relativePath));
|
||||||
|
file.getParentFile().mkdirs();
|
||||||
|
out = new FileOutputStream(file);
|
||||||
|
byte[] buff = new byte[4096];
|
||||||
|
int len;
|
||||||
|
while ((len = fileIn.read(buff)) != -1) {
|
||||||
|
out.write(buff, 0, len);
|
||||||
|
}
|
||||||
|
out.flush();
|
||||||
|
return relativePath;
|
||||||
|
} catch (IOException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
throw new IllegalArgumentException("上传文件失败,IO错误");
|
||||||
|
} finally {
|
||||||
|
try {
|
||||||
|
fileIn.close();
|
||||||
|
if (out != null) out.close();
|
||||||
|
} catch (IOException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean remove(String path) {
|
||||||
|
if (!StringUtils.hasText(path)) return false;
|
||||||
|
File file = new File(getFullPath(path));
|
||||||
|
if (!file.exists()) return false;
|
||||||
|
return file.delete();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void download(HttpServletResponse response, String fileName, String relativePath) {
|
||||||
|
Assert.hasText(relativePath, "路径为空");
|
||||||
|
System.out.println("------------------------------------" + getFullPath(relativePath));
|
||||||
|
File file = new File(getFullPath(relativePath));
|
||||||
|
Assert.isTrue(file.exists(), "附件不存在");
|
||||||
|
String sp = relativePath.substring(relativePath.lastIndexOf(".") + 1);
|
||||||
|
Assert.isTrue(!excludeSp.contains(sp.toLowerCase()), "文件类型错误");
|
||||||
|
if (!StringUtils.hasText(fileName)) {
|
||||||
|
fileName = file.getName().substring(0, file.getName().lastIndexOf("."));
|
||||||
|
}
|
||||||
|
InputStream fis = null;
|
||||||
|
try {
|
||||||
|
response.setContentType("application/octet-stream");
|
||||||
|
response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode((fileName + "." + sp), CharsetUtil.UTF_8));
|
||||||
|
fis = new BufferedInputStream(Files.newInputStream(file.toPath()));
|
||||||
|
response.addHeader("Content-Length", String.valueOf(fis.available()));
|
||||||
|
byte[] buff = new byte[4096];
|
||||||
|
int len;
|
||||||
|
while ((len = fis.read(buff)) != -1) {
|
||||||
|
response.getOutputStream().write(buff, 0, len);
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new IllegalArgumentException("下载文件失败: " + e.getMessage());
|
||||||
|
} finally {
|
||||||
|
try {
|
||||||
|
if (fis != null) fis.close();
|
||||||
|
} catch (IOException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void download(HttpServletResponse response, String relativePath) {
|
||||||
|
download(response, null, relativePath);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getFullPath(String relativePath) {
|
||||||
|
if (!relativePath.startsWith("/")) {
|
||||||
|
relativePath = "/" + relativePath;
|
||||||
|
}
|
||||||
|
return localPath + relativePath;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getSuffix(String fileName) {
|
||||||
|
Assert.hasText(fileName, "文件名不存在!");
|
||||||
|
if (fileName.lastIndexOf(".") < 0) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
return fileName.substring(fileName.lastIndexOf(".") + 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
public String upload(String fileName, MultipartFile file) {
|
||||||
|
return upload(fileName, file, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getDiskRelativePath(String fileName, String suffix) {
|
||||||
|
Calendar c = Calendar.getInstance();
|
||||||
|
Date d = new Date();
|
||||||
|
c.setTime(d);
|
||||||
|
String year = Integer.toString(c.get(Calendar.YEAR));
|
||||||
|
String month = Integer.toString(c.get(Calendar.MONTH) + 1);
|
||||||
|
String day = Integer.toString(c.get(Calendar.DATE));
|
||||||
|
StringBuilder path = new StringBuilder();
|
||||||
|
path.append("/").append(year)
|
||||||
|
.append("/").append(month)
|
||||||
|
.append("/").append(day)
|
||||||
|
.append("/").append(fileName);
|
||||||
|
if (StringUtils.hasText(suffix)) path.append(".").append(suffix);
|
||||||
|
return path.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
private String upload(String fileName, MultipartFile file, String relativePath) {
|
||||||
|
Assert.isTrue(!file.isEmpty(), "文件不存在");
|
||||||
|
String originalFilename = file.getOriginalFilename();
|
||||||
|
if (fileName == null) fileName = IdUtil.simpleUUID();
|
||||||
|
String sp = getSuffix(originalFilename);
|
||||||
|
Assert.notNull(sp, "文件类型错误");
|
||||||
|
Assert.isTrue(!excludeSp.contains(sp.toLowerCase()), "文件类型错误");
|
||||||
|
try {
|
||||||
|
String filePath;
|
||||||
|
if (relativePath == null) {
|
||||||
|
filePath = getDiskRelativePath(fileName, sp);
|
||||||
|
} else {
|
||||||
|
relativePath = relativePath.endsWith("/") ? relativePath : relativePath + "/";
|
||||||
|
filePath = relativePath + fileName + "." + sp;
|
||||||
|
}
|
||||||
|
|
||||||
|
File destFile = new File(getFullPath(filePath));
|
||||||
|
destFile.getParentFile().mkdirs();
|
||||||
|
file.transferTo(destFile);
|
||||||
|
return filePath;
|
||||||
|
} catch (FileNotFoundException e) {
|
||||||
|
throw new IllegalArgumentException("上传文件失败,FileNotFound错误");
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new IllegalArgumentException("上传文件失败,IO错误");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue