package org.jeecg.modules.system.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.apache.commons.lang3.StringUtils;
|
import org.jeecg.common.api.vo.FileUploadResult;
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.common.exception.ExceptionCast;
|
import org.jeecg.common.exception.JeecgBootException;
|
import org.jeecg.common.util.FileUtil;
|
import org.jeecg.modules.system.entity.FileDocInfo;
|
import org.jeecg.modules.system.mapper.DocInfoMapper;
|
import org.jeecg.modules.system.service.IDocInfoService;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.util.List;
|
|
@Service
|
public class DocInfoServiceImpl extends ServiceImpl<DocInfoMapper, FileDocInfo> implements IDocInfoService {
|
|
@Override
|
public boolean addDocInfo(MultipartFile file) {
|
if (file == null || file.isEmpty()) {
|
throw new JeecgBootException("非法参数");
|
}
|
FileUploadResult fileUploadResult = FileUtil.uploadFile(file);
|
if(fileUploadResult == null)
|
throw new JeecgBootException("文档导入失败");
|
|
String fileName = FileUtil.getFilenameNonSuffix(file.getOriginalFilename());
|
List<FileDocInfo> fileDocInfoList = this.list(new LambdaQueryWrapper<FileDocInfo>().eq(FileDocInfo::getFileName, fileName));
|
if (CollectionUtils.isNotEmpty(fileDocInfoList))
|
throw new JeecgBootException("该文档已存在");
|
|
FileDocInfo docInfo = new FileDocInfo();
|
String docId = IdWorker.getIdStr();
|
docInfo.setId(docId);
|
docInfo.setFileName(fileUploadResult.getFileName());
|
docInfo.setFilePath(fileUploadResult.getFilePath());
|
docInfo.setFileEncodeName(fileUploadResult.getFileEncodeName());
|
docInfo.setFileSize(fileUploadResult.getFileSize());
|
docInfo.setFileSuffix(fileUploadResult.getFileSuffix());
|
return super.save(docInfo);
|
}
|
|
@Override
|
public boolean deleteDocInfo(String id) {
|
|
if(StringUtils.isBlank(id))
|
throw new JeecgBootException("非法参数");
|
FileDocInfo en = super.getById(id);
|
if(en == null)
|
throw new JeecgBootException("未查询到文件信息");
|
FileUtil.deleteFile(en.getFilePath(), en.getFileName());
|
return super.removeById(id);
|
}
|
|
@Override
|
public Result<?> downloadDocFile(HttpServletResponse response, String id) {
|
|
if(StringUtils.isBlank(id))
|
throw new JeecgBootException("非法参数");
|
FileDocInfo en = super.getById(id);
|
if(en == null)
|
throw new JeecgBootException("未查询到文件信息");
|
|
String fileName;
|
if(StringUtils.isNotBlank(en.getFileSuffix())) {
|
fileName = en.getFileName() + "." + en.getFileSuffix();
|
}else {
|
fileName = en.getFileName();
|
}
|
FileUtil.downLoadFile(response, en.getFileEncodeName(), en.getFilePath(), fileName);
|
return null;
|
}
|
}
|
