package com.fdkankan.scene.service.impl;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.common.constant.ErrorCode;
import com.fdkankan.common.constant.ServerCode;
import com.fdkankan.common.constant.TbStatus;
import com.fdkankan.common.constant.UploadFilePath;
import com.fdkankan.common.exception.BusinessException;
import com.fdkankan.common.response.ResultData;
import com.fdkankan.common.user.SSOLoginHelper;
import com.fdkankan.common.user.SSOUser;
import com.fdkankan.common.util.BASE64DecodedMultipartFile;
import com.fdkankan.common.util.FileUtil;
import com.fdkankan.fyun.oss.UploadToOssUtil;
import com.fdkankan.scene.entity.ScenePro;
import com.fdkankan.scene.entity.SceneUpload;
import com.fdkankan.scene.mapper.ISceneUploadMapper;
import com.fdkankan.scene.service.ISceneProService;
import com.fdkankan.scene.service.ISceneUploadService;
import com.fdkankan.scene.vo.DeleteFileParamVO;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import java.io.File;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.UUID;
/**
*
* 服务实现类
*
*
* @author
* @since 2022-01-19
*/
@Service
public class SceneUploadServiceImpl extends ServiceImpl implements ISceneUploadService {
@Resource
UploadToOssUtil uploadToOssUtil;
@Value("${oss.point:http://oss-cn-shenzhen-internal.aliyuncs.com}")
private String point;
@Value("${oss.key:LTAIUrvuHqj8pvry}")
private String key;
@Value("${oss.secrey:JLOVl0k8Ke0aaM8nLMMiUAZ3EiiqI4}")
private String secrey;
@Value("${oss.bucket:4dkankan}")
private String bucket;
@Autowired
private SSOLoginHelper ssoLoginHelper;
@Override
public String uploads(String imgData,String fileName,String blzType,MultipartFile[] files,String sceneCode,Integer type,String token) throws Exception{
SSOUser ssoUser = ssoLoginHelper.loginCheck(token);
Long userId = ssoUser == null ? null :ssoUser.getId();
List multipartFiles = new ArrayList<>();
if(StringUtils.isNotBlank(imgData)){
MultipartFile file = BASE64DecodedMultipartFile.base64ToMultipart(imgData);
multipartFiles.add(file);
}
if(files !=null && files.length >0){
multipartFiles.addAll(Arrays.asList(files));
}
return this.uploadFiles(fileName,blzType,multipartFiles,sceneCode,type,userId);
}
@Override
public ResultData delete(DeleteFileParamVO param) throws Exception {
String key = String.format(UploadFilePath.USER_EDIT_PATH, param.getNum()) + param.getFileName();
uploadToOssUtil.delete(key);
this.update(new LambdaUpdateWrapper()
.set(SceneUpload::getTbStatus, TbStatus.DELETE.code())
.eq(SceneUpload::getNum, param.getNum())
.eq(SceneUpload::getBizType, param.getBizType())
.eq(SceneUpload::getFilePath, key)
.eq(SceneUpload::getTbStatus, TbStatus.VALID.code()));
return ResultData.ok();
}
public String uploadFiles(String sendFileName,String bizType,List files, String sceneCode, Integer type,Long userId) throws Exception{
if (StrUtil.isEmpty(sceneCode))
throw new BusinessException(ServerCode.PARAM_REQUIRED, "num");
if(CollUtil.isEmpty(files))
throw new BusinessException(ServerCode.PARAM_REQUIRED, "files");
if(StrUtil.isEmpty(bizType))
throw new BusinessException(ServerCode.PARAM_REQUIRED, "bizType");
List urlList = new ArrayList<>();
if(type != null && 1 == type){
this.updateFileByPreFix(sceneCode,bizType);
}
for (MultipartFile file : files) {
String fileName = file.getOriginalFilename();
// 获取文件后缀
String prefix = fileName.substring(fileName.lastIndexOf("."));
File newFile = File.createTempFile(UUID.randomUUID().toString() ,prefix);
file.transferTo(newFile);
String realFileName = fileName;
if(files.size() ==1 && StringUtils.isNotBlank(sendFileName)){
realFileName = sendFileName ;
}
String ossPath = String.format(UploadFilePath.USER_EDIT_PATH ,sceneCode) + realFileName;
try {
uploadToOssUtil.upload2(newFile.getPath(),ossPath);
}catch (Exception e){
log.error(ossPath+"上传文件失败"+e);
throw new BusinessException(-1,"上传文件失败");
}
//String url =String.format("http://%s.%s/%s", bucket, point.replace("http://",""), ossPath);
urlList.add(realFileName);
FileUtil.delFile(newFile.getPath());
//添加记录
this.saveData(sceneCode,ossPath,bizType,userId);
}
StringBuilder returnString = new StringBuilder();
for (String res : urlList) {
if(StringUtils.isNotBlank(returnString)){
returnString.append(",");
}
returnString.append(res);
}
return returnString.toString();
}
private void updateFileByPreFix(String sceneCode, String blzType) {
LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(SceneUpload::getNum,sceneCode)
.eq(SceneUpload::getBizType,blzType)
.eq(SceneUpload::getTbStatus,0);
List list = this.list(queryWrapper);
if(list != null && list.size() >0){
for (SceneUpload sceneUpload : list) {
try {
uploadToOssUtil.delete(sceneUpload.getFilePath());
this.removeEntity(sceneUpload);
}catch (Exception e){
log.error(sceneUpload.getFilePath()+"删除oss文件失败",e);
}
}
}
}
private void removeEntity(SceneUpload sceneUpload) {
sceneUpload.setTbStatus(1);
this.updateById(sceneUpload);
}
private void saveData(String sceneCode, String ossPath, String bizType,Long userId) {
SceneUpload sceneUpload = new SceneUpload();
sceneUpload.setNum(sceneCode);
sceneUpload.setFilePath(ossPath);
sceneUpload.setBizType(bizType);
sceneUpload.setUploadUser(userId);
this.save(sceneUpload);
}
}