package com.ruoyi.inspect.service.impl;
|
|
import cn.hutool.core.lang.UUID;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.ruoyi.framework.exception.ErrorException;
|
import com.ruoyi.inspect.mapper.InsUnqualifiedHandlerFileMapper;
|
import com.ruoyi.inspect.pojo.InsUnqualifiedHandlerFile;
|
import com.ruoyi.inspect.service.InsUnqualifiedHandlerFileService;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.File;
|
|
/**
|
* @author 27233
|
* @description 针对表【ins_unqualified_handler_file(不合格处理附件记录表)】的数据库操作Service实现
|
* @createDate 2024-07-31 13:38:38
|
*/
|
@Service
|
@Slf4j
|
public class InsUnqualifiedHandlerFileServiceImpl extends ServiceImpl<InsUnqualifiedHandlerFileMapper, InsUnqualifiedHandlerFile>
|
implements InsUnqualifiedHandlerFileService {
|
|
@Value("${wordUrl}")
|
private String wordUrl;
|
|
@Value("${file.path}")
|
private String imgUrl;
|
|
@Override
|
public InsUnqualifiedHandlerFile uploadFile(Long handlerId, MultipartFile file) {
|
String urlString;
|
String pathName;
|
String path;
|
String prefix;
|
String filename = file.getOriginalFilename();
|
String contentType = file.getContentType();
|
InsUnqualifiedHandlerFile unqualifiedHandlerFile = new InsUnqualifiedHandlerFile();
|
// unqualifiedHandlerFile.setUnqualifiedId(handlerId);
|
unqualifiedHandlerFile.setFileName(filename);
|
if (contentType != null && contentType.startsWith("image/")) {
|
// 是图片
|
path = imgUrl;
|
prefix = "/image/";
|
unqualifiedHandlerFile.setType(1);
|
} else {
|
// 是文件
|
path = wordUrl;
|
prefix = "/word/";
|
unqualifiedHandlerFile.setType(2);
|
}
|
try {
|
File realpath = new File(path);
|
if (!realpath.exists()) {
|
realpath.mkdirs();
|
}
|
pathName = UUID.randomUUID() + "_" + file.getOriginalFilename();
|
urlString = realpath + "/" + pathName;
|
file.transferTo(new File(urlString));
|
unqualifiedHandlerFile.setFileUrl(pathName);
|
baseMapper.insert(unqualifiedHandlerFile);
|
return unqualifiedHandlerFile;
|
} catch (Exception e) {
|
e.printStackTrace();
|
log.error("附件上传错误");
|
throw new ErrorException(e.getMessage());
|
}
|
}
|
|
}
|
