package org.dromara.system.service.impl;
|
|
import cn.hutool.core.convert.Convert;
|
import cn.hutool.core.io.IoUtil;
|
import cn.hutool.core.util.ObjectUtil;
|
import cn.hutool.json.JSON;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import com.github.therapi.runtimejavadoc.repack.com.eclipsesource.json.Json;
|
import org.dromara.common.core.constant.CacheNames;
|
import org.dromara.common.core.domain.R;
|
import org.dromara.common.core.domain.model.LoginUser;
|
import org.dromara.common.core.exception.ServiceException;
|
import org.dromara.common.core.service.OssService;
|
import org.dromara.common.core.utils.MapstructUtils;
|
import org.dromara.common.core.utils.SpringUtils;
|
import org.dromara.common.core.utils.StreamUtils;
|
import org.dromara.common.core.utils.StringUtils;
|
import org.dromara.common.core.utils.file.FileUtils;
|
import org.dromara.common.mybatis.core.page.PageQuery;
|
import org.dromara.common.mybatis.core.page.TableDataInfo;
|
import org.dromara.common.oss.core.OssClient;
|
import org.dromara.common.oss.entity.SynchronousRequest;
|
import org.dromara.common.oss.entity.UploadResult;
|
import org.dromara.common.oss.enumd.AccessPolicyType;
|
import org.dromara.common.oss.factory.OssFactory;
|
import org.dromara.common.satoken.utils.LoginHelper;
|
import org.dromara.system.domain.SysOss;
|
import org.dromara.system.domain.bo.SysOssBo;
|
import org.dromara.system.domain.vo.SysOssVo;
|
import org.dromara.system.mapper.SysOssMapper;
|
import org.dromara.system.service.ISysOssService;
|
import jakarta.servlet.http.HttpServletResponse;
|
import lombok.RequiredArgsConstructor;
|
import org.dromara.system.uitil.HttpUtils;
|
import org.dromara.system.uitil.PasswordUtil;
|
import org.jetbrains.annotations.NotNull;
|
import org.springframework.cache.annotation.Cacheable;
|
import org.springframework.http.MediaType;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.File;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.util.*;
|
|
/**
|
* 文件上传 服务层实现
|
*
|
* @author Lion Li
|
*/
|
@RequiredArgsConstructor
|
@Service
|
public class SysOssServiceImpl implements ISysOssService, OssService {
|
|
private final SysOssMapper baseMapper;
|
|
@Override
|
public TableDataInfo<SysOssVo> queryPageList(SysOssBo bo, PageQuery pageQuery) {
|
LambdaQueryWrapper<SysOss> lqw = buildQueryWrapper(bo);
|
Page<SysOssVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
|
List<SysOssVo> filterResult = StreamUtils.toList(result.getRecords(), this::matchingUrl);
|
result.setRecords(filterResult);
|
return TableDataInfo.build(result);
|
}
|
|
@Override
|
public List<SysOssVo> listByIds(Collection<Long> ossIds) {
|
List<SysOssVo> list = new ArrayList<>();
|
for (Long id : ossIds) {
|
SysOssVo vo = SpringUtils.getAopProxy(this).getById(id);
|
if (ObjectUtil.isNotNull(vo)) {
|
list.add(this.matchingUrl(vo));
|
}
|
}
|
return list;
|
}
|
|
@Override
|
public String selectUrlByIds(String ossIds) {
|
List<String> list = new ArrayList<>();
|
for (Long id : StringUtils.splitTo(ossIds, Convert::toLong)) {
|
SysOssVo vo = SpringUtils.getAopProxy(this).getById(id);
|
if (ObjectUtil.isNotNull(vo)) {
|
list.add(this.matchingUrl(vo).getUrl());
|
}
|
}
|
return String.join(StringUtils.SEPARATOR, list);
|
}
|
|
private LambdaQueryWrapper<SysOss> buildQueryWrapper(SysOssBo bo) {
|
Map<String, Object> params = bo.getParams();
|
LambdaQueryWrapper<SysOss> lqw = Wrappers.lambdaQuery();
|
lqw.select(SysOss.class,x->get(x.getColumn()));
|
lqw.like(StringUtils.isNotBlank(bo.getFileName()), SysOss::getFileName, bo.getFileName());
|
lqw.like(StringUtils.isNotBlank(bo.getOriginalName()), SysOss::getOriginalName, bo.getOriginalName());
|
lqw.eq(StringUtils.isNotBlank(bo.getFileSuffix()), SysOss::getFileSuffix, bo.getFileSuffix());
|
lqw.eq(StringUtils.isNotBlank(bo.getUrl()), SysOss::getUrl, bo.getUrl());
|
lqw.between(params.get("beginCreateTime") != null && params.get("endCreateTime") != null,
|
SysOss::getCreateTime, params.get("beginCreateTime"), params.get("endCreateTime"));
|
lqw.eq(ObjectUtil.isNotNull(bo.getCreateBy()), SysOss::getCreateBy, bo.getCreateBy());
|
lqw.eq(StringUtils.isNotBlank(bo.getService()), SysOss::getService, bo.getService());
|
lqw.orderByDesc(SysOss::getCreateTime);
|
return lqw;
|
}
|
|
public boolean get(String str){
|
if(str.equals("password") || str.equals("url") || str.equals("file_name")){
|
return false;
|
}
|
return true;
|
}
|
|
@Cacheable(cacheNames = CacheNames.SYS_OSS, key = "#ossId")
|
@Override
|
public SysOssVo getById(Long ossId) {
|
return baseMapper.selectVoById(ossId);
|
}
|
|
@Override
|
public void download(Long ossId, HttpServletResponse response) throws IOException {
|
SysOssVo sysOss = SpringUtils.getAopProxy(this).getById(ossId);
|
if (ObjectUtil.isNull(sysOss)) {
|
throw new ServiceException("文件数据不存在!");
|
}
|
FileUtils.setAttachmentResponseHeader(response, sysOss.getOriginalName());
|
response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE + "; charset=UTF-8");
|
OssClient storage = OssFactory.instance(sysOss.getService());
|
try(InputStream inputStream = storage.getObjectContent(sysOss.getUrl())) {
|
int available = inputStream.available();
|
IoUtil.copy(inputStream, response.getOutputStream(), available);
|
response.setContentLength(available);
|
} catch (Exception e) {
|
throw new ServiceException(e.getMessage());
|
}
|
}
|
|
@Override
|
public SysOssVo upload(MultipartFile file) {
|
String originalfileName = file.getOriginalFilename();
|
String suffix = StringUtils.substring(originalfileName, originalfileName.lastIndexOf("."), originalfileName.length());
|
OssClient storage = OssFactory.instance();
|
UploadResult uploadResult;
|
try {
|
uploadResult = storage.uploadSuffix(file.getBytes(), suffix, file.getContentType());
|
} catch (IOException e) {
|
throw new ServiceException(e.getMessage());
|
}
|
// 保存文件信息
|
return buildResultEntity(originalfileName, suffix, storage.getConfigKey(), uploadResult, file);
|
}
|
|
/**
|
* 同步
|
* @param
|
* @param
|
* @return
|
*/
|
@Override
|
public SysOssVo upload2(SynchronousRequest request) {
|
String originalfileName = request.getFile().getOriginalFilename();
|
String suffix = StringUtils.substring(originalfileName, originalfileName.lastIndexOf("."), originalfileName.length());
|
OssClient storage = OssFactory.instance();
|
UploadResult uploadResult;
|
try {
|
uploadResult = storage.uploadSuffix2(request.getFile().getBytes(), request.getPath(), request.getFile().getContentType());
|
} catch (IOException e) {
|
throw new ServiceException(e.getMessage());
|
}
|
// 保存文件信息
|
return buildResultEntity2(originalfileName, suffix, storage.getConfigKey(), uploadResult,request.getOssId(),
|
request.getPassword(), request.getCreateTime(), request.getCreateBy());
|
}
|
|
// @Override
|
// public SysOssVo upload(File file) {
|
// String originalfileName = file.getName();
|
// String suffix = StringUtils.substring(originalfileName, originalfileName.lastIndexOf("."), originalfileName.length());
|
// OssClient storage = OssFactory.instance();
|
// UploadResult uploadResult = storage.uploadSuffix(file, suffix);
|
// // 保存文件信息
|
// return buildResultEntity(originalfileName, suffix, storage.getConfigKey(), uploadResult);
|
// }
|
|
@NotNull
|
private SysOssVo buildResultEntity(String originalfileName, String suffix, String configKey, UploadResult uploadResult, MultipartFile file) {
|
SysOss oss = new SysOss();
|
oss.setUrl(uploadResult.getUrl());
|
oss.setFileSuffix(suffix);
|
oss.setFileName(uploadResult.getFilename());
|
oss.setOriginalName(originalfileName);
|
oss.setService(configKey);
|
oss.setPassword(PasswordUtil.randomPassword(6));
|
baseMapper.insert(oss);
|
//拼接同步信息
|
SynchronousRequest request = new SynchronousRequest();
|
request.setFile(file);
|
request.setPath(uploadResult.getFilename());
|
request.setCreateTime(oss.getCreateTime());
|
request.setPassword(oss.getPassword());
|
request.setCreateBy(String.valueOf(oss.getCreateBy()));
|
request.setOssId(String.valueOf(oss.getOssId()));
|
// post(request);
|
|
SysOssVo sysOssVo = MapstructUtils.convert(oss, SysOssVo.class);
|
return this.matchingUrl(sysOssVo);
|
}
|
|
/**
|
* 同步
|
* @param originalfileName
|
* @param suffix
|
* @param configKey
|
* @param uploadResult
|
* @param ossId
|
* @param password
|
* @param createTime
|
* @param createBy
|
* @return
|
*/
|
@NotNull
|
private SysOssVo buildResultEntity2(String originalfileName, String suffix, String configKey,
|
UploadResult uploadResult, String ossId, String password,
|
Date createTime, String createBy) {
|
SysOss oss = new SysOss();
|
oss.setOssId(Long.valueOf(ossId));
|
oss.setUrl(uploadResult.getUrl());
|
oss.setFileSuffix(suffix);
|
oss.setFileName(uploadResult.getFilename());
|
oss.setOriginalName(originalfileName);
|
oss.setService(configKey);
|
oss.setPassword(password);
|
oss.setCreateTime(createTime);
|
oss.setCreateBy(Long.valueOf(createBy));
|
baseMapper.insert(oss);
|
SysOssVo sysOssVo = MapstructUtils.convert(oss, SysOssVo.class);
|
return this.matchingUrl(sysOssVo);
|
}
|
|
@Override
|
public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid, String password) {
|
if (isValid) {
|
// 做一些业务上的校验,判断是否需要校验
|
}
|
List<SysOss> list = baseMapper.selectBatchIds(ids);
|
for (SysOss sysOss : list) {
|
OssClient storage = OssFactory.instance(sysOss.getService());
|
storage.delete(sysOss.getUrl());
|
if(sysOss.getPassword().equals(password)){
|
baseMapper.deleteById(sysOss.getOssId());
|
return true;
|
}
|
return false;
|
}
|
return true;
|
}
|
|
@Override
|
public R viewPasswordById(Long ossId) {
|
LoginUser loginUser = LoginHelper.getLoginUser();
|
SysOss oss = baseMapper.selectById(ossId);
|
if(oss == null){
|
return R.fail("文件已被清理，请联系管理员！");
|
}
|
if(!oss.getCreateBy().equals(loginUser.getUserId())){
|
return R.fail("您没有权限查看！");
|
}
|
return R.ok(oss.getPassword());
|
}
|
|
/**
|
* 匹配Url
|
*
|
* @param oss OSS对象
|
* @return oss 匹配Url的OSS对象
|
*/
|
private SysOssVo matchingUrl(SysOssVo oss) {
|
OssClient storage = OssFactory.instance(oss.getService());
|
// 仅修改桶类型为 private 的URL，临时URL时长为120s
|
if (AccessPolicyType.PRIVATE == storage.getAccessPolicy()) {
|
oss.setUrl(storage.getPrivateUrl(oss.getFileName(), 120));
|
}
|
return oss;
|
}
|
|
private String post(SynchronousRequest request){
|
HashMap<String, String> headers = new HashMap<>(3);
|
String requestUrl = "http://192.168.3.43:8080/resource/oss/synchronization";
|
ObjectMapper objectMapper = new ObjectMapper();
|
String jsonString = "";
|
try {
|
// 将Java对象转换为JSON字符串
|
jsonString = objectMapper.writeValueAsString(request);
|
System.out.println(jsonString);
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
headers.put("content-type", "application/json");
|
// 发送post请求
|
String resultData = HttpUtils.sendPostWithJson(requestUrl, jsonString,headers);
|
// 并接收返回结果
|
System.out.println(resultData);
|
return resultData;
|
}
|
}
|
