92 lines
3.6 KiB
Java
92 lines
3.6 KiB
Java
package cn.keking.utils;
|
||
|
||
import cn.keking.config.ConfigConstants;
|
||
import cn.keking.model.FileAttribute;
|
||
import cn.keking.model.ReturnResponse;
|
||
import io.mola.galimatias.GalimatiasParseException;
|
||
import org.apache.commons.io.FileUtils;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
|
||
import java.io.*;
|
||
import java.net.*;
|
||
import java.util.UUID;
|
||
|
||
import static cn.keking.utils.KkFileUtils.isFtpUrl;
|
||
import static cn.keking.utils.KkFileUtils.isHttpUrl;
|
||
|
||
/**
|
||
* @author yudian-it
|
||
*/
|
||
public class DownloadUtils {
|
||
|
||
private final static Logger logger = LoggerFactory.getLogger(DownloadUtils.class);
|
||
private static final String fileDir = ConfigConstants.getFileDir();
|
||
private static final String URL_PARAM_FTP_USERNAME = "ftp.username";
|
||
private static final String URL_PARAM_FTP_PASSWORD = "ftp.password";
|
||
private static final String URL_PARAM_FTP_CONTROL_ENCODING = "ftp.control.encoding";
|
||
|
||
/**
|
||
* @param fileAttribute fileAttribute
|
||
* @param fileName 文件名
|
||
* @return 本地文件绝对路径
|
||
*/
|
||
public static ReturnResponse<String> downLoad(FileAttribute fileAttribute, String fileName) {
|
||
String urlStr = fileAttribute.getUrl().replaceAll("\\+", "%20");
|
||
ReturnResponse<String> response = new ReturnResponse<>(0, "下载成功!!!", "");
|
||
String realPath = DownloadUtils.getRelFilePath(fileName, fileAttribute);
|
||
try {
|
||
URL url = WebUtils.normalizedURL(urlStr);
|
||
if (isHttpUrl(url)) {
|
||
File realFile = new File(realPath);
|
||
FileUtils.copyURLToFile(url,realFile);
|
||
} else if (isFtpUrl(url)) {
|
||
String ftpUsername = WebUtils.getUrlParameterReg(fileAttribute.getUrl(), URL_PARAM_FTP_USERNAME);
|
||
String ftpPassword = WebUtils.getUrlParameterReg(fileAttribute.getUrl(), URL_PARAM_FTP_PASSWORD);
|
||
String ftpControlEncoding = WebUtils.getUrlParameterReg(fileAttribute.getUrl(), URL_PARAM_FTP_CONTROL_ENCODING);
|
||
FtpUtils.download(fileAttribute.getUrl(), realPath, ftpUsername, ftpPassword, ftpControlEncoding);
|
||
} else {
|
||
response.setCode(1);
|
||
response.setMsg("url不能识别url" + urlStr);
|
||
}
|
||
response.setContent(realPath);
|
||
response.setMsg(fileName);
|
||
return response;
|
||
} catch (IOException | GalimatiasParseException e) {
|
||
logger.error("文件下载失败，url：{}", urlStr, e);
|
||
response.setCode(1);
|
||
response.setContent(null);
|
||
if (e instanceof FileNotFoundException) {
|
||
response.setMsg("文件不存在!!!");
|
||
} else {
|
||
response.setMsg(e.getMessage());
|
||
}
|
||
return response;
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* 获取真实文件绝对路径
|
||
*
|
||
* @param fileName 文件名
|
||
* @return 文件路径
|
||
*/
|
||
private static String getRelFilePath(String fileName, FileAttribute fileAttribute) {
|
||
String type = fileAttribute.getSuffix();
|
||
if (null == fileName) {
|
||
UUID uuid = UUID.randomUUID();
|
||
fileName = uuid + "." + type;
|
||
} else { // 文件后缀不一致时，以type为准(针对simText【将类txt文件转为txt】)
|
||
fileName = fileName.replace(fileName.substring(fileName.lastIndexOf(".") + 1), type);
|
||
}
|
||
String realPath = fileDir + fileName;
|
||
File dirFile = new File(fileDir);
|
||
if (!dirFile.exists() && !dirFile.mkdirs()) {
|
||
logger.error("创建目录【{}】失败,可能是权限不够，请检查", fileDir);
|
||
}
|
||
return realPath;
|
||
}
|
||
|
||
}
|