package com.ruoyi.personnel.service.impl;
|
|
import cn.hutool.core.lang.UUID;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.ruoyi.common.utils.SecurityUtils;
|
import com.ruoyi.personnel.enumeration.AttachmentType;
|
import com.ruoyi.personnel.mapper.PersonTrainingFileMapper;
|
import com.ruoyi.personnel.pojo.PersonTrainingFile;
|
import com.ruoyi.personnel.service.FileGeneralService;
|
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.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.IOException;
|
import java.time.LocalDateTime;
|
import java.util.List;
|
|
@Service
|
|
public class FileGeneralServiceImpl implements FileGeneralService {
|
|
@Autowired
|
private PersonTrainingFileMapper fileMapper;
|
|
@Value("${wordUrl}")
|
private String wordUrl;
|
|
@Value("${file.path}")
|
private String imgUrl;
|
|
@Value("${excelUrl}")
|
private String excelUrl;
|
|
@Override
|
public void fileUpload(MultipartFile file, String suffix, Integer id,Integer type) {
|
String name = file.getOriginalFilename();
|
UUID uuid = UUID.randomUUID();
|
String fileName = uuid + name;
|
String path = "";
|
// 根据后缀判断属于什么类型
|
if(suffix.toLowerCase().contains("xls")) {
|
path = excelUrl;
|
} else if (suffix.toLowerCase().contains("doc") || suffix.toLowerCase().contains("pdf")) {
|
path = wordUrl;
|
} else {
|
path = imgUrl;
|
}
|
File file1 = new File(path);
|
if(!file1.exists()) {
|
file1.mkdir();
|
}
|
try {
|
File file2 = new File(file1, fileName);
|
file.transferTo(file2);
|
// 保存数据库
|
PersonTrainingFile personTrainingFile = new PersonTrainingFile();
|
personTrainingFile.setDetailId(id);
|
personTrainingFile.setFileUrl("/" + fileName);
|
personTrainingFile.setFileName(file.getOriginalFilename());
|
personTrainingFile.setCreateUser(SecurityUtils.getUserId().intValue());
|
personTrainingFile.setCreateTime(LocalDateTime.now());
|
personTrainingFile.setEnumAttachmentType(AttachmentType.getTypeValue(type));
|
fileMapper.insert(personTrainingFile);
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
}
|
|
@Override
|
public List<PersonTrainingFile> selectFile(Integer id, Integer type) {
|
List<PersonTrainingFile> personTrainingFileList = fileMapper.selectList(new LambdaQueryWrapper<PersonTrainingFile>()
|
.eq(PersonTrainingFile::getDetailId, id)
|
.eq(PersonTrainingFile::getEnumAttachmentType, AttachmentType.getTypeValue(type)));
|
if(CollectionUtils.isNotEmpty(personTrainingFileList)) {
|
for(PersonTrainingFile a : personTrainingFileList) {
|
// 设置mime
|
int i = a.getFileUrl().lastIndexOf(".");
|
String contentType = getContentType(a.getFileUrl().substring(i + 1));
|
a.setMime(contentType);
|
}
|
}
|
return personTrainingFileList;
|
}
|
|
@Override
|
public void delFile(Integer id) {
|
fileMapper.deleteById(id);
|
}
|
|
@Override
|
public void fileDownLoad(Integer id, HttpServletResponse response) {
|
PersonTrainingFile personTrainingFile = fileMapper.selectById(id);
|
// 获取文件后缀
|
String suffix = personTrainingFile.getFileName().substring(personTrainingFile.getFileName().lastIndexOf(".") + 1);
|
String path = "";
|
if(suffix.toLowerCase().contains("xls")) {
|
path = excelUrl;
|
} else if (suffix.toLowerCase().contains("doc") || suffix.toLowerCase().contains("pdf")) {
|
path = wordUrl;
|
} else {
|
path = imgUrl;
|
}
|
File file = new File(path + File.separator + personTrainingFile.getFileUrl());
|
if(!file.exists()) {
|
throw new RuntimeException("文件不存在");
|
}
|
try(FileInputStream fileInputStream = new FileInputStream(file);
|
ServletOutputStream stream = response.getOutputStream()) {
|
String contentType = getContentType(suffix);
|
response.setContentType(contentType);
|
response.setHeader("Content-disposition", "attachment;filename=" + personTrainingFile.getFileName());
|
byte[] bytes = new byte[1024];
|
int i;
|
while((i = fileInputStream.read(bytes)) != -1) {
|
stream.write(bytes, 0, i);
|
stream.flush();
|
}
|
}catch (Exception e) {
|
throw new RuntimeException(e);
|
}
|
|
|
|
|
|
|
|
}
|
// 根据文件扩展名获取 MIME 类型
|
private String getContentType(String fileExtension) {
|
switch (fileExtension) {
|
case "jpg":
|
case "jpeg":
|
return "image/jpeg";
|
case "png":
|
return "image/png";
|
case "gif":
|
return "image/gif";
|
case "pdf":
|
return "application/pdf";
|
case "doc":
|
return "application/msword";
|
case "docx":
|
return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
|
case "xls":
|
return "application/vnd.ms-excel";
|
case "xlsx":
|
return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
|
case "ppt":
|
case "pptx":
|
return "application/vnd.ms-powerpoint";
|
default:
|
return "application/octet-stream"; // 默认二进制流
|
}
|
}
|
}
|
