package com.zt.life.sys.service;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import cn.hutool.core.convert.Convert;
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.fasterxml.jackson.annotation.ObjectIdGenerator;
|
import com.zt.common.constant.Constant;
|
import com.zt.common.db.query.QueryFilter;
|
import com.zt.common.service.BaseService;
|
import com.zt.common.utils.CacheUtils;
|
import com.zt.common.utils.UUIDUtil;
|
import com.zt.core.oss.service.ISysOssConfigService;
|
import com.zt.core.oss.service.ISysOssService;
|
import com.zt.life.core.constant.Cache;
|
import com.zt.life.export.service.DownloadService;
|
import com.zt.life.oss.OssEncryptService;
|
import com.zt.life.sys.dao.SysOssConfigDao;
|
import com.zt.life.sys.dto.OssDto;
|
import com.zt.life.sys.mapstruct.OssExMapper;
|
import com.zt.life.sys.model.SysOssConfig;
|
import com.zt.life.util.ZipCipherUtil;
|
import com.zt.modules.oss.model.SysOss;
|
import com.zt.modules.oss.service.SysOssService;
|
import org.apache.commons.io.FileUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.cache.annotation.CacheEvict;
|
import org.springframework.stereotype.Service;
|
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.nio.file.Files;
|
import java.nio.file.Paths;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
/**
|
* SYS_OSS_CONFIG
|
*
|
* @author zt generator
|
* @since 1.0.0 2020-08-25
|
*/
|
@Service
|
public class SysOssConfigService extends BaseService<SysOssConfigDao, SysOssConfig> implements ISysOssConfigService {
|
|
@Autowired
|
private OssExMapper ossExMapper;
|
@Autowired
|
private ISysOssService sysOssService;
|
|
@Autowired
|
private SysOssService sysOssServices;
|
|
@Value("${zt.oss.local-path}")
|
private String localPath;
|
|
@Value("${data.key}")
|
private String key;
|
|
@Autowired
|
private OssEncryptService ossEncryptService;
|
|
/**
|
* 分页查询
|
*
|
* @param queryFilter
|
* @return
|
*/
|
public List<SysOssConfig> page(QueryFilter queryFilter) {
|
return queryFilter.getPageList(baseDao.getList(queryFilter.getQueryParams()));
|
}
|
|
@Override
|
@CacheEvict(value = Cache.CONFIG, allEntries = true)
|
public void insert(SysOssConfig entity) {
|
super.insert(entity);
|
}
|
|
@Override
|
@CacheEvict(value = Cache.CONFIG, allEntries = true)
|
public void update(SysOssConfig entity) {
|
super.update(entity);
|
}
|
|
/**
|
* 删除
|
*
|
* @param ids
|
*/
|
@CacheEvict(value = Cache.CONFIG, allEntries = true)
|
public void delete(Long[] ids) {
|
super.deleteLogic(ids);
|
}
|
|
public List<OssDto> getOssConfig() {
|
// 这里手动缓存，不然在当前方法内调用缓存无效
|
List<OssDto> all = null;//(List<OssDto>) CacheUtils.get(Cache.CONFIG, "all");
|
if (all == null) {
|
all = new ArrayList<>();
|
List<SysOssConfig> configs = baseDao.selectList(new QueryWrapper<SysOssConfig>()
|
.eq(Constant.TableColumn.IS_DELETE, Constant.Bool.NO).lambda().orderByAsc(SysOssConfig::getSort));
|
if (configs.size() > 0) {
|
Map<String, OssDto> busiMap = new HashMap<>();
|
Map<String, OssDto.OssFieldGroupDto> groupMap = new HashMap<>();
|
|
for (SysOssConfig config : configs) {
|
OssDto dto = busiMap.get(config.getBusiType());
|
if (dto == null) {
|
dto = new OssDto(config.getBusiType(), config.getBusiTypeName());
|
busiMap.put(config.getBusiType(), dto);
|
all.add(dto);
|
}
|
|
String group = config.getBusiFieldGroup();
|
OssDto.OssFieldGroupDto groupDto = groupMap.get(group);
|
if (groupDto == null) {
|
groupDto = new OssDto.OssFieldGroupDto(group);
|
dto.getGroups().add(groupDto);
|
groupMap.put(group, groupDto);
|
}
|
groupDto.getFields().add(ossExMapper.toDto(config));
|
}
|
}
|
CacheUtils.put(Cache.CONFIG, "all", all);
|
}
|
return all.stream().map(ossDto -> {
|
try {
|
return ossDto.clone();
|
} catch (CloneNotSupportedException e) {
|
}
|
return null;
|
}).collect(Collectors.toList());
|
}
|
|
/**
|
* 查询附件
|
*
|
* @param busiId
|
* @param busiType
|
* @return
|
*/
|
public OssDto getOssByBusiType(Long busiId, String busiType) {
|
List<OssDto> dtos = this.getOssConfig().stream().filter(config -> config.getBusiType().equals(busiType))
|
.collect(Collectors.toList());
|
|
if (dtos.size() > 0) {
|
OssDto dto = dtos.get(0);
|
if (busiId != null && busiId > 0) {
|
for (OssDto.OssFieldGroupDto groupDto : dto.getGroups()) {
|
for (OssDto.OssFieldDto fieldDto : groupDto.getFields()) {
|
fieldDto.getFiles().addAll(sysOssService.getOsses(busiId, busiType, fieldDto.getBusiField()));
|
}
|
}
|
}
|
return dto;
|
}
|
return null;
|
}
|
|
|
/**
|
* 修改附件http
|
*
|
* @param busiId
|
* @param dto
|
*/
|
public void updateOss(Long busiId, OssDto dto) {
|
if (dto != null) {
|
for (OssDto.OssFieldGroupDto groupDto : dto.getGroups()) {
|
for (OssDto.OssFieldDto fieldDto : groupDto.getFields()) {
|
if (CollectionUtil.isNotEmpty(fieldDto.getFiles())) {
|
List<Long> ids = fieldDto.getFiles().stream().map(file -> file.getId())
|
.collect(Collectors.toList());
|
sysOssService.updateBusiInfo(busiId, dto.getBusiType(), dto.getBusiTypeName(),
|
fieldDto.getBusiField(), fieldDto.getBusiFieldName(), ids, true);
|
} else {// 没有附件了
|
sysOssService.deleteBusiInfo(busiId, dto.getBusiType(), fieldDto.getBusiField());
|
}
|
}
|
}
|
}
|
}
|
|
public void updateOssBatch(String[] bizIds, OssDto dto) {
|
if (dto != null) {
|
for (OssDto.OssFieldGroupDto groupDto : dto.getGroups()) {
|
for (OssDto.OssFieldDto fieldDto : groupDto.getFields()) {
|
if (CollectionUtil.isNotEmpty(fieldDto.getFiles())) {
|
List<Long> ids = fieldDto.getFiles().stream().map(file -> file.getId())
|
.collect(Collectors.toList());
|
SysOss sysOss = sysOssServices.get(ids.get(0));
|
for (String bizId : bizIds) {
|
Long uuId = UUIDUtil.generateId();
|
sysOss.setId(uuId);
|
sysOssServices.insert(sysOss);
|
List<Long> ids2 = new ArrayList<>();
|
ids2.add(uuId);
|
sysOssServices.updateBusiInfo(Long.parseLong(bizId), dto.getBusiType(), dto.getBusiTypeName(),
|
fieldDto.getBusiField(), fieldDto.getBusiFieldName(), ids2, true);
|
}
|
|
} else {// 没有附件了
|
//sysOssService.deleteBusiInfo(busiId, dto.getBusiType(), fieldDto.getBusiField());
|
}
|
}
|
}
|
}
|
}
|
|
public void updateOssBatch2(String bizIds, OssDto dto) {
|
if (dto != null) {
|
SysOss sysOss = null;
|
for (OssDto.OssFieldGroupDto groupDto : dto.getGroups()) {
|
for (OssDto.OssFieldDto fieldDto : groupDto.getFields()) {
|
if (CollectionUtil.isNotEmpty(fieldDto.getFiles())) {
|
List<Long> ids = fieldDto.getFiles().stream().map(file -> file.getId())
|
.collect(Collectors.toList());
|
sysOss = sysOssServices.get(ids.get(0));
|
}
|
}
|
}
|
if (sysOss != null) {
|
for (String bizId : bizIds.split(",")) {
|
sysOss.setBusiId(Convert.toLong(bizId));
|
Long uuId = UUIDUtil.generateId();
|
sysOss.setId(uuId);
|
sysOssServices.insert(sysOss);
|
}
|
}
|
}
|
}
|
/**
|
* 批量上传
|
*
|
* @param dto
|
*/
|
public void uploadingOss(List<Long> busiIdList, OssDto dto) {
|
List<SysOss> all = new ArrayList<>();
|
String busiType = "";
|
String busiTypeName = "";
|
String busiField = "";
|
String busiFieldName = "";
|
for (OssDto.OssFieldGroupDto groupDto : dto.getGroups()) {
|
for (OssDto.OssFieldDto fieldDto : groupDto.getFields()) {
|
if (CollectionUtil.isNotEmpty(fieldDto.getFiles())) {
|
List<Long> ids = fieldDto.getFiles().stream().map(file -> file.getId()).collect(Collectors.toList());
|
if (ids.size() > 0) {
|
all = sysOssServices.getBySysOss(ids);
|
}
|
}
|
}
|
}
|
|
for (Long busiId : busiIdList) {
|
for (OssDto.OssFieldGroupDto groupDto : dto.getGroups()) {
|
for (OssDto.OssFieldDto fieldDto : groupDto.getFields()) {
|
if (CollectionUtil.isNotEmpty(fieldDto.getFiles())) {
|
List<Long> ids = fieldDto.getFiles().stream().map(file -> file.getId()).collect(Collectors.toList());
|
busiType = dto.getBusiType();
|
busiTypeName = dto.getBusiTypeName();
|
busiField = fieldDto.getBusiField();
|
busiFieldName = fieldDto.getBusiFieldName();
|
sysOssService.updateBusiInfo(busiId, busiType, busiTypeName, busiField, busiFieldName, ids, false);
|
}
|
// else {// 没有附件了
|
// sysOssService.deleteBusiInfo(busiId, dto.getBusiType(), fieldDto.getBusiField());
|
// }
|
}
|
}
|
}
|
List<com.zt.core.oss.dto.OssDto> ossDtoList = new ArrayList<>();
|
if (all != null && all.size() > 0 && busiIdList.size() > 1) {
|
for (int i = 0; i < busiIdList.size() - 1; i++) {
|
for (int j = 0; j < all.size(); j++) {
|
Long uuid = generateId();
|
SysOss sysOss = all.get(j);
|
sysOss.setId(uuid);
|
String urlOssIdStr = sysOss.getUrl().substring(sysOss.getUrl().indexOf("_zt_oss_id="));
|
String newUrlOssIdStr = "_zt_oss_id=" + uuid;
|
sysOss.setUrl(sysOss.getUrl().replace(urlOssIdStr, newUrlOssIdStr));
|
sysOss.setBusiId(busiIdList.get(i + 1));
|
sysOss.setBusiType(busiType);
|
sysOss.setBusiTypeName(busiTypeName);
|
sysOss.setBusiField(busiField);
|
sysOss.setBusiFieldName(busiFieldName);
|
sysOssServices.insert(sysOss);
|
ossDtoList.add(JSONObject.toJavaObject(JSONObject.parseObject(JSONObject.toJSONString(sysOss)), com.zt.core.oss.dto.OssDto.class));
|
}
|
}
|
}
|
}
|
|
|
public static Long generateId() {
|
String code = String.valueOf(new Date().getTime());
|
float fl = new Random().nextFloat();
|
fl = fl < 0.1 ? (float) (fl + 0.1) : fl;
|
Integer ran = Integer.parseInt((fl * 1000000 + "").split("\\.")[0]);
|
char[] temp = code.substring(code.length() - 6, code.length()).toCharArray();
|
for (char c : temp) {
|
int i = new Random().nextInt(6);
|
ran += Integer.parseInt(((Integer.parseInt(c + "") * Math.pow(10, i)) + "").split("\\.")[0]);
|
}
|
code = code.substring(0, code.length() - 6) + ran;
|
String buwei = "";
|
int codel = code.length();
|
if (codel < 17) {
|
double f = Math.pow(10, 17 - codel);
|
fl = new Random().nextFloat();
|
fl = fl < 0.1 ? (float) (fl + 0.1) : fl;
|
String a1 = fl * f + "";
|
buwei = a1.split("\\.")[0];
|
}
|
Random runnable = new Random();
|
int num = runnable.nextInt(90) + 10;
|
return Long.parseLong(code + buwei + num);
|
}
|
|
public Integer existsSameTypeAndFiledGroup(String busiType, String busiFieldGroup, Long id) {
|
return baseDao.existsSameTypeAndFiledGroup(busiType, busiFieldGroup, id);
|
}
|
|
public Integer existsSameGroupById(String busiFiledGroup, Long id) {
|
return baseDao.existsSameGroupById(busiFiledGroup, id);
|
}
|
|
public SysOssConfig getByBusiTypeAndBusiFiledGroup(String busiType, String busiFiledGroup) {
|
return baseDao.selectOne(new QueryWrapper<SysOssConfig>().eq("busi_type", busiType).eq("busi_field_group", busiFiledGroup));
|
}
|
|
public void deleteByBusiTypes(Long[] busiTypes) {
|
for (Long busiType : busiTypes) {
|
baseDao.updateByBusiType(String.valueOf(busiType));
|
}
|
}
|
|
public List<SysOssConfig> selectList(String busiType) {
|
return baseDao.selectList(new QueryWrapper<SysOssConfig>().eq(Constant.TableColumn.IS_DELETE, Constant.Bool.NO).lambda()
|
.eq(SysOssConfig::getBusiType, busiType));
|
}
|
|
public void downloadFilesByosList(HttpServletRequest request, HttpServletResponse response, List<SysOss> list, String projectName) {
|
try {
|
String uuid = UUIDUtil.getUUID();
|
String remoteFileFullPath = Paths.get(localPath, "temp", uuid).toString();
|
Files.createDirectories(Paths.get(remoteFileFullPath));
|
|
for (SysOss so : list) {
|
try {
|
String localFileFullPath = Paths.get(localPath, so.getPath()).toString();
|
File localFile = new File(localFileFullPath);
|
if (localFile.exists()) {
|
try (InputStream in = ossEncryptService.decrypt(localFile);
|
BufferedInputStream bis = new BufferedInputStream(in);
|
BufferedOutputStream bos = new BufferedOutputStream(
|
new FileOutputStream(remoteFileFullPath + File.separator + so.getName()))) {
|
|
byte[] buffer = new byte[1024];
|
int len;
|
while ((len = bis.read(buffer)) != -1) {
|
bos.write(buffer, 0, len);
|
}
|
}
|
}
|
} catch (Exception e) {
|
e.printStackTrace(); // 或者记录日志
|
}
|
}
|
|
String fileNameZip = Paths.get(localPath, "temp", uuid + ".zip").toString();
|
new ZipCipherUtil().encryptZip(remoteFileFullPath, fileNameZip, key);
|
|
File zipFile = new File(fileNameZip);
|
response.reset();
|
response.setContentLength((int) zipFile.length());
|
response.setContentType("application/octet-stream;charset=UTF-8");
|
String encodedFilename = DownloadService.getNameEncoder(request, projectName + ".zip");
|
response.addHeader("Content-Disposition", "attachment;filename=" + encodedFilename);
|
|
try (FileInputStream fileInputStream = new FileInputStream(zipFile);
|
ServletOutputStream outputStream = response.getOutputStream()) {
|
|
byte[] buffers = new byte[1024];
|
int length;
|
while ((length = fileInputStream.read(buffers)) > 0) {
|
outputStream.write(buffers, 0, length);
|
}
|
|
} catch (IOException e) {
|
e.printStackTrace(); // 或者记录日志
|
} finally {
|
FileUtils.deleteQuietly(new File(remoteFileFullPath));
|
FileUtils.deleteQuietly(new File(fileNameZip));// 删除临时文件夹及其内容
|
}
|
} catch (Exception ex) {
|
ex.printStackTrace(); // 或者记录日志
|
}
|
}
|
}
|
