package cn.huge.module.file.utils;
|
|
import cn.huge.base.common.constant.FileCatEnum;
|
import cn.huge.base.common.exception.ServiceException;
|
import cn.huge.base.common.utils.DateUtils;
|
import cn.huge.module.constant.BaseConsts;
|
import cn.huge.module.file.config.FtpFileConfig;
|
import cn.huge.module.file.domain.po.FileInfo;
|
import org.apache.commons.codec.digest.DigestUtils;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.IOException;
|
import java.math.BigDecimal;
|
import java.math.RoundingMode;
|
import java.util.Calendar;
|
import java.util.Date;
|
|
/**
|
* @title: 上传附件包装处理类
|
* @description: 上传附件包装处理类
|
* @company: hugeinfo
|
* @author: liyj
|
* @time: 2021-11-05 16:51:48
|
* @version: 1.0.0
|
*/
|
public class FtpMultipartFileWrapper {
|
|
static final long kilobyte = 1024;
|
static final long megabyte = 1024 * 1024L;
|
static final long gigabyte = 1024 * 1024 * 1024L;
|
|
static final BigDecimal kilobyteDividend = new BigDecimal(kilobyte);
|
static final BigDecimal megabyteDividend = new BigDecimal(megabyte);
|
static final BigDecimal gigabyteDividend = new BigDecimal(gigabyte);
|
|
static final String SHOW_URL_WEB = "/api/web/fileInfo/show";
|
static final String DOWN_URL_WEB = "/api/web/fileInfo/down";
|
static final String SHOW_URL_WECHAT = "/api/wechat/fileInfo/show";
|
static final String DOWN_URL_WECHAT = "/api/wechat/fileInfo/down";
|
|
static final String STORE_WAY = "ftp";
|
|
private MultipartFile file;
|
|
public FtpMultipartFileWrapper(MultipartFile file) {
|
this.file = file;
|
}
|
|
/**
|
* web端上传附件
|
* @param fileId
|
* @return
|
*/
|
public FileInfo toWebFileInfo(String fileId) {
|
FileInfo fileInfo = new FileInfo();
|
fileInfo.setId(fileId);
|
fileInfo.setSuffix(this.getFileSuffix());
|
fileInfo.setTrueName(this.getFileName() + BaseConsts.DOT + fileInfo.getSuffix());
|
fileInfo.setFileName(this.getFileName() + BaseConsts.UNDER + fileInfo.getId() + BaseConsts.DOT + fileInfo.getSuffix());
|
fileInfo.setCat(FileCatEnum.getFileCat(fileInfo.getSuffix()));
|
fileInfo.setSize(this.getFileSize());
|
fileInfo.setUnit(this.getFileUnit());
|
fileInfo.setMd5Code(this.getFileVerifyCode());
|
fileInfo.setStoreWay(STORE_WAY);
|
fileInfo.setPath(this.getFileDir());
|
fileInfo.setFullPath(fileInfo.getPath() + BaseConsts.SLANT + fileInfo.getFileName());
|
fileInfo.setShowUrl(SHOW_URL_WEB + BaseConsts.SLANT + fileInfo.getId());
|
fileInfo.setDownUrl(DOWN_URL_WEB + BaseConsts.SLANT + fileInfo.getId());
|
// todo
|
fileInfo.setZipUrl(fileInfo.getShowUrl());
|
Date nowDate = DateUtils.getNowDate();
|
fileInfo.setCreateTime(nowDate);
|
fileInfo.setUpdateTime(nowDate);
|
fileInfo.setDeleteStatus(BaseConsts.DELETE_STATUS_0);
|
return fileInfo;
|
}
|
|
/**
|
* wechat上传附件
|
* @param fileId
|
* @return
|
*/
|
public FileInfo toWechatFileInfo(String fileId) {
|
FileInfo fileInfo = new FileInfo();
|
fileInfo.setId(fileId);
|
fileInfo.setSuffix(this.getFileSuffix());
|
fileInfo.setTrueName(this.getFileName() + BaseConsts.DOT + fileInfo.getSuffix());
|
// fileInfo.setFileName(this.getFileName() + BaseConsts.UNDER + fileInfo.getId() + BaseConsts.DOT + fileInfo.getSuffix());
|
fileInfo.setFileName(this.getFileVerifyCode() + BaseConsts.DOT + fileInfo.getSuffix());
|
fileInfo.setCat(FileCatEnum.getFileCat(fileInfo.getSuffix()));
|
fileInfo.setSize(this.getFileSize());
|
fileInfo.setUnit(this.getFileUnit());
|
fileInfo.setMd5Code(this.getFileVerifyCode());
|
fileInfo.setStoreWay(STORE_WAY);
|
fileInfo.setPath(this.getFileDir());
|
fileInfo.setFullPath(fileInfo.getPath() + BaseConsts.SLANT + fileInfo.getFileName());
|
fileInfo.setShowUrl(SHOW_URL_WECHAT + BaseConsts.SLANT + fileInfo.getId());
|
fileInfo.setDownUrl(DOWN_URL_WECHAT + BaseConsts.SLANT + fileInfo.getId());
|
// todo
|
fileInfo.setZipUrl(fileInfo.getShowUrl());
|
Date nowDate = DateUtils.getNowDate();
|
fileInfo.setCreateTime(nowDate);
|
fileInfo.setUpdateTime(nowDate);
|
fileInfo.setDeleteStatus(BaseConsts.DELETE_STATUS_0);
|
return fileInfo;
|
}
|
|
/**
|
* 获取文件名称
|
*/
|
public String getFileName() {
|
return StringUtils.substringBeforeLast(this.file.getOriginalFilename(), BaseConsts.DOT);
|
}
|
|
/**
|
* 获取文件后缀
|
*/
|
public String getFileSuffix() {
|
return StringUtils.substringAfterLast(this.file.getOriginalFilename(), BaseConsts.DOT);
|
}
|
|
/**
|
* 获取文件存储目录，以斜杠开头(/home/ftp/huge-dyh/2022/3)
|
*/
|
public String getFileDir() {
|
//按照年月构建目录
|
Calendar ca = Calendar.getInstance();
|
StringBuilder builder = new StringBuilder();
|
builder.append(FtpFileConfig.getRootdir());
|
builder.append(BaseConsts.SLANT);
|
builder.append(ca.get(Calendar.YEAR));
|
builder.append(BaseConsts.SLANT);
|
builder.append(ca.get(Calendar.MONTH) + 1);
|
return builder.toString();
|
}
|
|
/**
|
* 获取文件大小, 根据不同的单位进行运算
|
*/
|
public double getFileSize() {
|
long size = this.file.getSize();
|
BigDecimal divisor = new BigDecimal(size);
|
String unit = this.getFileUnit();
|
|
if (BaseConsts.B.equals(unit)) {
|
return size;
|
} else if (BaseConsts.KB.equals(unit)) {
|
return divisor.divide(kilobyteDividend, 2, RoundingMode.HALF_UP).doubleValue();
|
} else if (BaseConsts.MB.equals(unit)) {
|
return divisor.divide(megabyteDividend, 2, RoundingMode.HALF_UP).doubleValue();
|
} else if (BaseConsts.GB.equals(unit)) {
|
return divisor.divide(gigabyteDividend, 2, RoundingMode.HALF_UP).doubleValue();
|
}
|
return 0.0;
|
}
|
|
/**
|
* 获取文件单位, B|KB|MB|GB
|
*/
|
public String getFileUnit() {
|
long size = this.file.getSize();
|
if (size > 0 && size <= kilobyte) {
|
return BaseConsts.B;
|
} else if (size > kilobyte && size <= megabyte) {
|
return BaseConsts.KB;
|
} else if (size > megabyte && size <= gigabyte) {
|
return BaseConsts.MB;
|
} else if (size > gigabyte) {
|
return BaseConsts.GB;
|
}
|
return BaseConsts.B;
|
}
|
|
/**
|
* 获取文件校验码, MD5
|
*/
|
public String getFileVerifyCode() {
|
try {
|
return DigestUtils.md5Hex(this.file.getBytes());
|
} catch (IOException e) {
|
throw new ServiceException("MultipartFileWrapper.getFileVerifyCode", e);
|
}
|
}
|
}
|
