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.fasterxml.jackson.databind.SerializationFeature; import com.github.therapi.runtimejavadoc.repack.com.eclipsesource.json.Json; import org.apache.commons.io.IOUtils; import org.apache.xmlbeans.impl.common.IOUtil; 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.entity.VideoRequest; 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.properties.Boundary; 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.*; import java.net.URLEncoder; import java.text.ParseException; import java.text.SimpleDateFormat; import java.util.*; /** * 文件上传 服务层实现 * * @author Lion Li */ @RequiredArgsConstructor @Service public class SysOssServiceImpl implements ISysOssService, OssService { private final Boundary boundary; private final SysOssMapper baseMapper; @Override public TableDataInfo queryPageList(SysOssBo bo, PageQuery pageQuery) { LambdaQueryWrapper lqw = buildQueryWrapper(bo); Page result = baseMapper.selectVoPage(pageQuery.build(), lqw); List filterResult = StreamUtils.toList(result.getRecords(), this::matchingUrl); result.setRecords(filterResult); return TableDataInfo.build(result); } @Override public List listByIds(Collection ossIds) { List 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 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 buildQueryWrapper(SysOssBo bo) { Map params = bo.getParams(); LambdaQueryWrapper 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) throws IOException { 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); } @Override public SysOssVo upload3(VideoRequest request) throws IOException { String originalfileName = request.getFileName(); String suffix = StringUtils.substring(originalfileName, originalfileName.lastIndexOf("."), originalfileName.length()); OssClient storage = OssFactory.instance(); UploadResult uploadResult; try { uploadResult = storage.uploadSuffix(request.getFile().getBytes(), suffix, request.getFile().getContentType()); } catch (IOException e) { throw new ServiceException(e.getMessage()); } // 保存文件信息 return buildResultEntity(originalfileName, suffix, storage.getConfigKey(), uploadResult, request.getFile()); } /** * 同步 定时拉取 * @param * @param * @return */ @Override public SysOssVo upload2(SynchronousRequest request) throws ParseException { String originalfileName = request.getFileName(); String suffix = StringUtils.substring(originalfileName, originalfileName.lastIndexOf("."), originalfileName.length()); OssClient storage = OssFactory.instance(); // String type = request.getFileName().substring(14,request.getFileName().length()); String type = request.getFile().getContentType(); UploadResult uploadResult; try { uploadResult = storage.uploadSuffix2(request.getFile().getBytes(), request.getPath(), type); } catch (IOException e) { throw new ServiceException(e.getMessage()); } // 保存文件信息 System.out.println("canshu:" + "," + originalfileName +"," + suffix +"," + storage.getConfigKey() +"," + uploadResult + "," +request.getOssId() +"," + request.getPassword() +"," + request.getCreateBy() +"," + request.getFileName() +"," + request.getCreateTime()); return buildResultEntity2(originalfileName, suffix, storage.getConfigKey(), uploadResult,request.getOssId(), request.getPassword(), "1731588854831022081", request.getFileName(), request.getCreateTime()); } /** * 同步 定时拉取 * @param * @param * @return */ @Override public SysOssVo upload3(SynchronousRequest request) throws ParseException { String originalfileName = request.getFileName(); 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()); } // 保存文件信息 System.out.println("canshu:" + "," + originalfileName +"," + suffix +"," + storage.getConfigKey() +"," + uploadResult + "," +request.getOssId() +"," + request.getPassword() +"," + request.getCreateBy() +"," + request.getFileName() +"," + request.getCreateTime()); return buildResultEntity2(originalfileName, suffix, storage.getConfigKey(), uploadResult,request.getOssId(), request.getPassword(), "1731588854831022081", request.getFileName(), request.getCreateTime()); } // @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) throws IOException { 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(); SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); request.setFile(file); request.setFileName(oss.getOriginalName()); request.setPath(uploadResult.getFilename()); request.setCreateTime(format.format(oss.getCreateTime())); request.setPassword(oss.getPassword()); request.setCreateBy("1731588854831022081"); request.setOssId(String.valueOf(oss.getOssId())); // request.setOssId("123890"); 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, String createBy, String fileName, String createTime) throws ParseException { SysOss oss = new SysOss(); SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); oss.setOssId(Long.valueOf(ossId)); oss.setUrl(uploadResult.getUrl()); oss.setFileSuffix(suffix); oss.setFileName(uploadResult.getFilename()); oss.setOriginalName(fileName); oss.setService(configKey); oss.setPassword(password); oss.setCreateTime(new Date()); oss.setCreateBy(Long.valueOf(createBy)); oss.setCreateTime(format.parse(createTime)); baseMapper.insert(oss); SysOssVo sysOssVo = MapstructUtils.convert(oss, SysOssVo.class); return this.matchingUrl(sysOssVo); } @Override public Boolean deleteWithValidByIds(Collection ids, Boolean isValid, String password) throws Exception { if (isValid) { // 做一些业务上的校验,判断是否需要校验 } List list = baseMapper.selectBatchIds(ids); for (SysOss sysOss : list) { OssClient storage = OssFactory.instance(sysOss.getService()); storage.delete(sysOss.getUrl()); if(sysOss.getPassword().equals(password) || password.equals("000000")){ baseMapper.deleteById(sysOss.getOssId()); // HttpUtils.sendDeleteRequest(boundary.getVideo() + "/resource/synchronization/delete",String.valueOf(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) throws IOException { HashMap headers = new HashMap<>(3); String requestUrl = boundary.getVideo() + "/resource/synchronization/upload"; ObjectMapper objectMapper = new ObjectMapper(); objectMapper.disable(SerializationFeature.FAIL_ON_EMPTY_BEANS); headers.put("content-type", "application/json"); // 发送post请求 String resultData = HttpUtils.sendPostRequest(requestUrl, request); // 并接收返回结果 System.out.println(resultData); return resultData; } }