package com.zt.modules.oss.service;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import cn.hutool.core.io.IoUtil;
|
import cn.hutool.core.util.NumberUtil;
|
import cn.hutool.core.util.ReflectUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.zt.common.constant.Constant;
|
import com.zt.common.db.query.QueryFilter;
|
import com.zt.common.entity.BaseEntity;
|
import com.zt.common.exception.ErrorCode;
|
import com.zt.common.exception.RenException;
|
import com.zt.common.service.BaseService;
|
import com.zt.core.oss.dto.OssDto;
|
import com.zt.core.oss.service.ISysOssService;
|
import com.zt.modules.oss.annotation.DataFieldParser;
|
import com.zt.modules.oss.cloud.AbstractStorageService;
|
import com.zt.modules.oss.constant.OssConstant;
|
import com.zt.modules.oss.dao.SysOssDao;
|
import com.zt.modules.oss.enums.CloudChannel;
|
import com.zt.modules.oss.enums.OssStatus;
|
import com.zt.modules.oss.mapstruct.OssMapper;
|
import com.zt.modules.oss.model.OssFieldDefine;
|
import com.zt.modules.oss.model.SysOss;
|
import org.apache.commons.io.FileUtils;
|
import org.apache.commons.io.FilenameUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.multipart.MultipartFile;
|
import sun.misc.BASE64Decoder;
|
|
import javax.annotation.PostConstruct;
|
import javax.crypto.Cipher;
|
import javax.crypto.KeyGenerator;
|
import java.io.*;
|
import java.security.Key;
|
import java.security.SecureRandom;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* 文件上传
|
*
|
* @author hehz
|
*/
|
@Service
|
public class SysOssService extends BaseService<SysOssDao, SysOss> implements ISysOssService {
|
|
@Autowired
|
private AbstractStorageService storageService;
|
@Autowired
|
private DataFieldParser dataFieldParser;
|
@Autowired
|
private OssMapper ossMapper;
|
|
private Map<String, OssFieldDefine> map = new HashMap<>();
|
private static final String CIPHER_ALGORITHM = "AES";
|
private static final String key = "zhpt-key#%W";
|
|
@PostConstruct
|
private void init() {
|
try {
|
dataFieldParser.getOssFields().stream().forEach(d -> map.put(d.getClassName(), d));
|
} catch (Exception e) {
|
}
|
}
|
|
public List<SysOss> page(QueryFilter queryFilter) {
|
List<SysOss> list = baseDao.getList(queryFilter.getParams());
|
list.stream().forEach(oss -> oss.setUrl(getOssUrl(oss)));
|
return queryFilter.getPageList(list);
|
}
|
|
public SysOss getById(Long id) {
|
SysOss oss = baseDao.selectById(id);
|
oss.setUrl(getOssUrl(oss));
|
return oss;
|
}
|
|
/**
|
* 通过业务id BusiId和业务类型 busiType获取附件数据
|
*
|
* @return
|
*/
|
public List<SysOss> getSysOssByBusiIdAnd(Long busiId, String busiType) {
|
return baseDao.selectList(new QueryWrapper<SysOss>().eq(Constant.TableColumn.IS_DELETE, Constant.Bool.NO).lambda()
|
.eq(SysOss::getBusiId, busiId).eq(SysOss::getBusiType, busiType));
|
}
|
|
/**
|
* 上传
|
*
|
* @param file
|
* @return
|
* @throws IOException
|
*/
|
public SysOss uploadExt(MultipartFile file, Long busiId, String busiType) throws IOException {
|
if (file.isEmpty()) {
|
throw new RenException(ErrorCode.UPLOAD_FILE_EMPTY);
|
}
|
|
// 上传文件
|
String extension = FilenameUtils.getExtension(file.getOriginalFilename());
|
SysOss oss = storageService.uploadSuffix(file.getBytes(), extension);
|
oss.setName(file.getOriginalFilename());
|
oss.setType(extension);
|
String url = oss.getUrl();
|
String idPart = OssConstant.ID + "=" + oss.getId();
|
oss.setUrl(url + (url.indexOf("?") > 0 ? "&" : "?") + idPart);
|
oss.setSize(NumberUtil.round(file.getSize() / 1024, 2).doubleValue());
|
oss.setChannel(storageService.getConfig().getType());// 渠道
|
oss.setStatus(OssStatus.TEMPORARY.getValue());
|
oss.setBusiType(busiType);
|
//oss.setFileType();
|
oss.setBusiId(busiId);
|
this.insert(oss);
|
return oss;
|
}
|
|
/**
|
* 上传
|
*
|
* @param
|
* @return
|
* @throws IOException
|
*/
|
public OssDto uploadNew(@RequestBody Map<String, Object> fileData) throws IOException {
|
if (fileData.get("uid") == null) {
|
throw new RenException(ErrorCode.UPLOAD_FILE_EMPTY);
|
}
|
String name = (String) fileData.get("name");
|
String uid = (String) fileData.get("uid");
|
// 上传文件
|
String extension = FilenameUtils.getExtension(name);
|
String tempUploadDir = storageService.getConfig().getLocalPath() + File.separator + "TEMP_UPLOAD";
|
String filePath = tempUploadDir + File.separator + "file_" + uid;
|
|
File file = new File(filePath);
|
SysOss oss = storageService.uploadSuffix(FileUtils.openInputStream(file), extension);
|
oss.setName(name);
|
oss.setType(extension);
|
String url = oss.getUrl();
|
String idPart = OssConstant.ID + "=" + oss.getId();
|
oss.setUrl(url + (url.indexOf("?") > 0 ? "&" : "?") + idPart);
|
oss.setSize(NumberUtil.round(file.length() / 1024, 2).doubleValue());
|
oss.setChannel(storageService.getConfig().getType());// 渠道
|
oss.setStatus(OssStatus.TEMPORARY.getValue());
|
|
this.insert(oss);
|
return ossMapper.toDto(oss);
|
}
|
|
public OssDto upload(MultipartFile file) throws IOException {
|
if (file.isEmpty()) {
|
throw new RenException(ErrorCode.UPLOAD_FILE_EMPTY);
|
}
|
|
// 上传文件
|
String extension = FilenameUtils.getExtension(file.getOriginalFilename());
|
SysOss oss = storageService.uploadSuffix(file.getBytes(), extension);
|
oss.setName(file.getOriginalFilename());
|
oss.setType(extension);
|
String url = oss.getUrl();
|
String idPart = OssConstant.ID + "=" + oss.getId();
|
oss.setUrl(url + (url.indexOf("?") > 0 ? "&" : "?") + idPart);
|
oss.setSize(NumberUtil.round(file.getSize() / 1024, 2).doubleValue());
|
oss.setChannel(storageService.getConfig().getType());// 渠道
|
oss.setStatus(OssStatus.TEMPORARY.getValue());
|
|
this.insert(oss);
|
return ossMapper.toDto(oss);
|
}
|
|
public void upload2(MultipartFile file, Long busiId, String busiType, String busiTypeName, String busiField, String busiFieldName) throws IOException {
|
if (file.isEmpty()) {
|
throw new RenException(ErrorCode.UPLOAD_FILE_EMPTY);
|
}
|
|
// 上传文件
|
String extension = FilenameUtils.getExtension(file.getOriginalFilename());
|
SysOss oss = storageService.uploadSuffix(file.getBytes(), extension);
|
oss.setName(file.getOriginalFilename());
|
oss.setType(extension);
|
String url = oss.getUrl();
|
String idPart = OssConstant.ID + "=" + oss.getId();
|
oss.setUrl(url + (url.indexOf("?") > 0 ? "&" : "?") + idPart);
|
oss.setBusiId(busiId);
|
oss.setBusiType(busiType);
|
oss.setBusiTypeName(busiTypeName);
|
oss.setBusiField(busiField);
|
oss.setBusiFieldName(busiFieldName);
|
oss.setSize(NumberUtil.round(file.getSize() / 1024, 2).doubleValue());
|
oss.setChannel(storageService.getConfig().getType());// 渠道
|
oss.setStatus(OssStatus.TEMPORARY.getValue());
|
|
this.insert(oss);
|
}
|
|
/**
|
* 删除临时文件
|
*
|
* @param id
|
*/
|
public void deleteTemporary(Long id) {
|
SysOss oss = getById(id);
|
if (!OssStatus.TEMPORARY.getValue().equals(oss.getStatus())) {
|
throw new RenException("正式文件不能删除！");
|
}
|
|
storageService.delete(oss);
|
|
baseDao.deleteById(id);
|
}
|
|
public void setListOsses(List<? extends BaseEntity> dataList, String busiType) {
|
List<Long> ids = dataList.stream().map(f -> f.getId()).collect(Collectors.toList());
|
Map<String, Object> params = new HashMap<String, Object>();
|
params.put("busiIdList", ids);
|
params.put("busiType", busiType);
|
Map<Long, Map<String, String>> accessoryMap = getOsses(params);
|
for (BaseEntity item : dataList) {
|
item.setAccessoryMap(accessoryMap.get(item.getId()));
|
}
|
}
|
|
public Map<Long, Map<String, String>> getOsses(Map<String, Object> params) {
|
List<SysOss> sysOssList = this.getSysOssByBusiType(params);
|
Map<Long, Map<String, String>> accessoryMap = new HashMap<>();
|
String downRouter = storageService.getConfig().getLocalDomain() + "/sys/oss/content?";
|
if (sysOssList != null) {
|
for (SysOss sysOss : sysOssList) {
|
String idPart = OssConstant.ID + "=" + sysOss.getId();
|
String url = downRouter + idPart;
|
if (accessoryMap.containsKey(sysOss.getBusiId())) {
|
accessoryMap.get(sysOss.getBusiId()).put(url, sysOss.getName());
|
} else {
|
Map<String, String> map = new HashMap<>();
|
map.put(url, sysOss.getName());
|
accessoryMap.put(sysOss.getBusiId(), map);
|
}
|
}
|
}
|
return accessoryMap;
|
}
|
|
@Override
|
public List<OssDto> getOsses(Long busiId, String busiType, String busiField) {
|
List<OssDto> list = baseDao
|
.selectList(
|
new QueryWrapper<SysOss>().lambda().eq(SysOss::getBusiType, busiType)
|
.eq(SysOss::getBusiId, busiId).eq(SysOss::getBusiField, busiField)
|
.orderByAsc(true, SysOss::getSort)).stream().map(d -> {
|
OssDto dto = ossMapper.toDto(d);
|
dto.setUrl(getOssUrl(d));
|
return dto;
|
}).collect(Collectors.toList());
|
return list;
|
}
|
|
@Override
|
public void setOssField(BaseEntity entity) {
|
if (entity == null) {
|
return;
|
}
|
String busiType = entity.getClass().getSimpleName();
|
if (map.containsKey(busiType)) {
|
OssFieldDefine define = map.get(entity.getClass().getSimpleName());
|
for (OssFieldDefine.Field field : define.getFields()) {
|
List<OssDto> list = baseDao
|
.selectList(
|
new QueryWrapper<SysOss>().lambda().eq(SysOss::getBusiType, busiType)
|
.eq(SysOss::getBusiId, entity.getId())
|
.eq(SysOss::getBusiField, field.getField()).orderByAsc(true, SysOss::getSort))
|
.stream().map(d -> ossMapper.toDto(d)).collect(Collectors.toList());
|
|
if (CollectionUtil.isNotEmpty(list)) {
|
Class clazz = ReflectUtil.getField(entity.getClass(), field.getField()).getType();
|
if (clazz.getName().equals(List.class.getName())) {
|
ReflectUtil.setFieldValue(entity, field.getField(), list);
|
} else if (clazz.getName().equals(OssDto.class.getName())) {
|
ReflectUtil.setFieldValue(entity, field.getField(), list.get(0));
|
}
|
}
|
}
|
}
|
}
|
|
@Override
|
public void updateBusiInfo(BaseEntity entity) {
|
if (entity == null) {
|
return;
|
}
|
String busiType = entity.getClass().getSimpleName();
|
if (map.containsKey(busiType)) {
|
OssFieldDefine define = map.get(entity.getClass().getSimpleName());
|
for (OssFieldDefine.Field field : define.getFields()) {
|
Object value = ReflectUtil.getFieldValue(entity, field.getField());
|
List<OssDto> datas = new ArrayList<>();
|
if (value != null) {
|
if (value instanceof OssDto) {
|
datas.add((OssDto) value);
|
} else if (value instanceof List) {
|
datas.addAll((Collection<? extends OssDto>) value);
|
} else if (value instanceof String) {// 富文本
|
String text = (String) value;
|
int index;
|
while ((index = text.indexOf(OssConstant.ID)) > 0) {
|
String id = text.substring(index, index + 20);
|
OssDto oss = new OssDto();
|
oss.setId(Long.parseLong(id));
|
datas.add(oss);
|
|
text = text.substring(index + 20);
|
}
|
}
|
}
|
List<Long> ids = datas.stream().map(d -> d.getId()).collect(Collectors.toList());
|
this.updateBusiInfo(entity.getId(), busiType, define.getBusiTypeName(), field.getField(),
|
field.getFieldName(), ids, true);
|
}
|
}
|
}
|
|
@Override
|
public void updateBusiInfo(Long busiId, String busiType, String busiTypeName, String busiField,
|
String busiFieldName, List<Long> ids, Boolean deleteListNotExists) {
|
LambdaQueryWrapper<SysOss> queryWrapper = new LambdaQueryWrapper<SysOss>().and(Wrapper -> Wrapper
|
.eq(SysOss::getBusiType, busiType).eq(SysOss::getBusiId, busiId).eq(SysOss::getBusiField, busiField));
|
if (ids.size() > 0) {
|
queryWrapper.or(Wrapper -> Wrapper.in(SysOss::getId, ids));
|
}
|
List<SysOss> all = baseDao.selectList(queryWrapper);
|
Map<Long, SysOss> map = new HashMap<>();
|
for (SysOss oss : all) {
|
map.put(oss.getId(), oss);
|
}
|
for (int i = 0; i < ids.size(); i++) {
|
Long id = ids.get(i);
|
if (id != null && map.containsKey(id)) {
|
SysOss db = map.get(id);
|
if (db.getBusiId() != null && db.getBusiId() > 0) {// 原来已保存了业务id的
|
if (db.getSort() == null || db.getSort() != (i + 1)) {// 排序变了
|
db.setSort(i + 1);
|
baseDao.updateById(db);
|
}
|
} else {// 新上传需要补充业务id的
|
db.setBusiType(busiType);
|
db.setBusiTypeName(busiTypeName);
|
db.setBusiId(busiId);
|
db.setBusiField(busiField);
|
db.setBusiFieldName(busiFieldName);
|
db.setStatus(OssStatus.PERMANENT.getValue());
|
baseDao.updateById(db);
|
}
|
map.remove(id);
|
}
|
}
|
if (deleteListNotExists) {
|
for (SysOss oss : map.values()) {// 删除数据库和文件
|
storageService.delete(oss);
|
baseDao.deleteById(oss.getId());
|
}
|
}
|
}
|
|
|
@Override
|
public void deleteBusiInfo(String busiType, Long busiId) {
|
if (map.containsKey(busiType)) {
|
baseDao.delete(new QueryWrapper<SysOss>().lambda().eq(SysOss::getBusiType, busiType)
|
.eq(SysOss::getBusiId, busiId));
|
}
|
}
|
|
@Override
|
public void deleteBusiInfo(Long busiId, String busiType, String busiField) {
|
baseDao.delete(new QueryWrapper<SysOss>().lambda().eq(SysOss::getBusiType, busiType)
|
.eq(SysOss::getBusiId, busiId).eq(SysOss::getBusiField, busiField));
|
}
|
|
private String getOssUrl(SysOss oss) {
|
String url = oss.getUrl();
|
String idPart = OssConstant.ID + "=" + oss.getId();
|
oss.setUrl(url + (url.indexOf("?") > 0 ? "&" : "?") + idPart);
|
if (CloudChannel.LOCAL.getValue().equals(oss.getChannel())) {// 本地
|
url = storageService.getConfig().getLocalDomain() + "/sys/oss/content";
|
}
|
return url + (url.indexOf("?") > 0 ? "&" : "?") + idPart;
|
}
|
|
public String getOssUrlCommon(SysOss oss) {
|
String url = oss.getUrl();
|
String idPart = OssConstant.ID + "=" + oss.getId();
|
oss.setUrl(url + (url.indexOf("?") > 0 ? "&" : "?") + idPart);
|
if (CloudChannel.LOCAL.getValue().equals(oss.getChannel())) {// 本地
|
url = storageService.getConfig().getLocalDomain() + "/sys/oss/content";
|
}
|
return url + (url.indexOf("?") > 0 ? "&" : "?") + idPart;
|
}
|
|
/**
|
* 通过id获取指定数据
|
*/
|
public List<SysOss> getBySysOss(List<Long> ids) {
|
return baseDao.selectList(new QueryWrapper<SysOss>().eq(Constant.TableColumn.IS_DELETE, Constant.Bool.NO).lambda().in(SysOss::getId, ids));
|
}
|
|
public Integer getBusiFieldNameCount(Long busiId, String busiType) {
|
return baseDao.getBusiFieldNameCount(busiId, busiType);
|
}
|
|
public List<SysOss> getSysOssByBusiType(Map<String, Object> params) {
|
return baseDao.getSysOssByBusiType(params);
|
}
|
|
public boolean removeBase64File() {
|
List<SysOss> list = baseDao.getOldFileList();
|
|
for (SysOss item : list) {
|
try {
|
String localPath = storageService.getConfig().getLocalPath();
|
String oldFilePath = localPath + "/" + item.getPath();
|
File oldFile = new File(oldFilePath);
|
System.out.println(oldFilePath);
|
String tempFilePath2 = storageService.getConfig().getLocalPath() + "/tempJx2";
|
if (oldFile.exists()) {
|
|
SecureRandom sr = new SecureRandom();
|
Cipher cipher = Cipher.getInstance(CIPHER_ALGORITHM);
|
Key secureKey = getKey(key);
|
cipher.init(Cipher.DECRYPT_MODE, secureKey, sr);
|
byte[] res = new BASE64Decoder().decodeBuffer(IoUtil.toStream(oldFile));
|
res = cipher.doFinal(res);
|
|
|
OutputStream fos = new FileOutputStream(tempFilePath2);
|
fos.write(res);//利
|
fos.close();
|
|
InputStream inputStream = new FileInputStream(tempFilePath2);
|
Cipher cipher2 = Cipher.getInstance(CIPHER_ALGORITHM);
|
secureKey = getKey(key);
|
cipher2.init(Cipher.ENCRYPT_MODE, secureKey);
|
|
FileOutputStream outputStream = new FileOutputStream(oldFilePath);
|
byte[] buffer = new byte[8192];
|
int bytesRead;
|
|
try {
|
while ((bytesRead = inputStream.read(buffer)) != -1) {
|
byte[] encryptedBytes = cipher2.update(buffer, 0, bytesRead);
|
outputStream.write(encryptedBytes);
|
}
|
byte[] finalEncryptedBytes = cipher2.doFinal();
|
outputStream.write(finalEncryptedBytes);
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
// 关闭资源
|
try {
|
inputStream.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
try {
|
outputStream.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
return true;
|
}
|
|
private Key getKey(String strKey) {
|
try {
|
if (strKey == null) {
|
strKey = "";
|
}
|
KeyGenerator _generator = KeyGenerator.getInstance(CIPHER_ALGORITHM);
|
SecureRandom secureRandom = SecureRandom.getInstance("SHA1PRNG");
|
secureRandom.setSeed(strKey.getBytes());
|
_generator.init(128, secureRandom);
|
return _generator.generateKey();
|
} catch (Exception e) {
|
throw new RuntimeException("密钥生成异常");
|
}
|
}
|
}
|
