package org.jeecg.modules.system.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.jeecg.common.api.dto.message.MessageDTO;
|
import org.jeecg.common.constant.CommonConstant;
|
import org.jeecg.common.constant.SymbolConstant;
|
import org.jeecg.common.constant.enums.FileTypeEnum;
|
import org.jeecg.common.constant.enums.MessageTypeEnum;
|
import org.jeecg.common.exception.JeecgBootException;
|
import org.jeecg.common.system.api.ISysBaseAPI;
|
import org.jeecg.common.util.CommonUtils;
|
import org.jeecg.common.util.RedisUtil;
|
import org.jeecg.common.util.oConvertUtils;
|
import org.jeecg.modules.system.entity.SysComment;
|
import org.jeecg.modules.system.entity.SysFiles;
|
import org.jeecg.modules.system.entity.SysFormFile;
|
import org.jeecg.modules.system.mapper.SysCommentMapper;
|
import org.jeecg.modules.system.mapper.SysFilesMapper;
|
import org.jeecg.modules.system.mapper.SysFormFileMapper;
|
import org.jeecg.modules.system.service.ISysCommentService;
|
import org.jeecg.modules.system.vo.SysCommentFileVo;
|
import org.jeecg.modules.system.vo.SysCommentVO;
|
import org.jeecg.modules.system.vo.UserAvatar;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.util.FileCopyUtils;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.io.File;
|
import java.io.IOException;
|
import java.util.*;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
|
/**
|
* @Description: 系统评论回复表
|
* @Author: jeecg-boot
|
* @Date: 2022-07-19
|
* @Version: V1.0
|
*/
|
@Service
|
public class SysCommentServiceImpl extends ServiceImpl<SysCommentMapper, SysComment> implements ISysCommentService {
|
|
@Autowired
|
private ISysBaseAPI sysBaseApi;
|
|
@Autowired
|
private SysFormFileMapper sysFormFileMapper;
|
|
@Autowired
|
private SysFilesMapper sysFilesMapper;
|
|
@Autowired
|
private RedisUtil redisUtil;
|
|
@Value(value = "${jeecg.path.upload}")
|
private String uploadpath;
|
|
@Value(value = "${jeecg.uploadType}")
|
private String uploadType;
|
|
/**
|
* sysFormFile中的表名
|
*/
|
private static final String SYS_FORM_FILE_TABLE_NAME = "sys_comment";
|
|
@Override
|
public List<SysCommentVO> queryFormCommentInfo(SysComment sysComment) {
|
String tableName = sysComment.getTableName();
|
String dataId = sysComment.getTableDataId();
|
//获取评论信息
|
List<SysCommentVO> list = this.baseMapper.queryCommentList(tableName, dataId);
|
// 获取评论相关人员
|
Set<String> personSet = new HashSet<>();
|
if(list!=null && list.size()>0){
|
for(SysCommentVO vo: list){
|
if(oConvertUtils.isNotEmpty(vo.getFromUserId())){
|
personSet.add(vo.getFromUserId());
|
}
|
if(oConvertUtils.isNotEmpty(vo.getToUserId())){
|
personSet.add(vo.getToUserId());
|
}
|
}
|
}
|
if(personSet.size()>0){
|
//获取用户信息
|
Map<String, UserAvatar> userAvatarMap = queryUserAvatar(personSet);
|
for(SysCommentVO vo: list){
|
String formId = vo.getFromUserId();
|
String toId = vo.getToUserId();
|
// 设置头像、用户名
|
if(oConvertUtils.isNotEmpty(formId)){
|
UserAvatar fromUser = userAvatarMap.get(formId);
|
if(fromUser!=null){
|
vo.setFromUserId_dictText(fromUser.getRealname());
|
vo.setFromUserAvatar(fromUser.getAvatar());
|
}
|
}
|
if(oConvertUtils.isNotEmpty(toId)){
|
UserAvatar toUser = userAvatarMap.get(toId);
|
if(toUser!=null){
|
vo.setToUserId_dictText(toUser.getRealname());
|
vo.setToUserAvatar(toUser.getAvatar());
|
}
|
}
|
}
|
}
|
return list;
|
}
|
|
@Transactional(rollbackFor = Exception.class)
|
@Override
|
public void saveOneFileComment(HttpServletRequest request) {
|
String savePath = "";
|
String bizPath = request.getParameter("biz");
|
//LOWCOD-2580 sys/common/upload接口存在任意文件上传漏洞
|
if (oConvertUtils.isNotEmpty(bizPath)) {
|
if (bizPath.contains(SymbolConstant.SPOT_SINGLE_SLASH) || bizPath.contains(SymbolConstant.SPOT_DOUBLE_BACKSLASH)) {
|
throw new JeecgBootException("上传目录bizPath，格式非法！");
|
}
|
}
|
MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
|
// 获取上传文件对象
|
MultipartFile file = multipartRequest.getFile("file");
|
if (oConvertUtils.isEmpty(bizPath)) {
|
if (CommonConstant.UPLOAD_TYPE_OSS.equals(uploadType)) {
|
//未指定目录，则用阿里云默认目录 upload
|
bizPath = "upload";
|
} else {
|
bizPath = "";
|
}
|
}
|
if (CommonConstant.UPLOAD_TYPE_LOCAL.equals(uploadType)) {
|
savePath = this.uploadLocal(file, bizPath);
|
} else {
|
savePath = CommonUtils.upload(file, bizPath, uploadType);
|
}
|
|
String orgName = file.getOriginalFilename();
|
// 获取文件名
|
orgName = CommonUtils.getFileName(orgName);
|
//文件大小
|
long size = file.getSize();
|
//文件类型
|
String type = orgName.substring(orgName.lastIndexOf("."), orgName.length());
|
FileTypeEnum fileType = FileTypeEnum.getByType(type);
|
|
//保存至 SysFiles
|
SysFiles sysFiles = new SysFiles();
|
sysFiles.setFileName(orgName);
|
sysFiles.setUrl(savePath);
|
sysFiles.setFileType(fileType.getValue());
|
sysFiles.setStoreType("temp");
|
if (size > 0) {
|
sysFiles.setFileSize(Double.parseDouble(String.valueOf(size)));
|
}
|
String defaultValue = "0";
|
sysFiles.setIzStar(defaultValue);
|
sysFiles.setIzFolder(defaultValue);
|
sysFiles.setIzRootFolder(defaultValue);
|
sysFiles.setDelFlag(defaultValue);
|
String fileId = String.valueOf(IdWorker.getId());
|
sysFiles.setId(fileId);
|
sysFilesMapper.insert(sysFiles);
|
|
//保存至 SysFormFile
|
String tableName = SYS_FORM_FILE_TABLE_NAME;
|
String tableDataId = request.getParameter("commentId");
|
SysFormFile sysFormFile = new SysFormFile();
|
sysFormFile.setTableName(tableName);
|
sysFormFile.setFileType(fileType.getValue());
|
sysFormFile.setTableDataId(tableDataId);
|
sysFormFile.setFileId(fileId);
|
sysFormFileMapper.insert(sysFormFile);
|
|
}
|
|
@Override
|
public List<SysCommentFileVo> queryFormFileList(String tableName, String formDataId) {
|
List<SysCommentFileVo> list = baseMapper.queryFormFileList(tableName, formDataId);
|
return list;
|
}
|
|
@Override
|
public String saveOne(SysComment sysComment) {
|
this.save(sysComment);
|
//发送系统消息
|
String content = sysComment.getCommentContent();
|
if (content.indexOf("@") >= 0) {
|
Set<String> set = getCommentUsername(content);
|
if (set.size() > 0) {
|
String users = String.join(",", set);
|
MessageDTO md = new MessageDTO();
|
md.setTitle("有人在表单评论中提到了你");
|
md.setContent(content);
|
md.setToAll(false);
|
md.setToUser(users);
|
md.setFromUser("system");
|
md.setType(MessageTypeEnum.XT.getType());
|
sysBaseApi.sendTemplateMessage(md);
|
}
|
}
|
return sysComment.getId();
|
}
|
|
@Override
|
public void deleteOne(String id) {
|
this.removeById(id);
|
//还要删除关联文件
|
LambdaQueryWrapper<SysFormFile> query = new LambdaQueryWrapper<SysFormFile>()
|
.eq(SysFormFile::getTableDataId, id)
|
.eq(SysFormFile::getTableName, SYS_FORM_FILE_TABLE_NAME);
|
this.sysFormFileMapper.delete(query);
|
}
|
|
/**
|
* 通过正则获取评论中的用户账号
|
*
|
* @return
|
*/
|
private Set<String> getCommentUsername(String content) {
|
Set<String> set = new HashSet<String>(3);
|
String reg = "(@(.*?)\\[(.*?)\\])";
|
Pattern p = Pattern.compile(reg);
|
Matcher m = p.matcher(content);
|
while (m.find()) {
|
if (m.groupCount() == 3) {
|
String username = m.group(3);
|
set.add(username);
|
}
|
}
|
return set;
|
}
|
|
|
/**
|
* 本地文件上传
|
*
|
* @param mf 文件
|
* @param bizPath 自定义路径
|
* @return
|
*/
|
private String uploadLocal(MultipartFile mf, String bizPath) {
|
//LOWCOD-2580 sys/common/upload接口存在任意文件上传漏洞
|
if (oConvertUtils.isNotEmpty(bizPath) && (bizPath.contains("../") || bizPath.contains("..\\"))) {
|
throw new JeecgBootException("上传目录bizPath，格式非法！");
|
}
|
try {
|
String ctxPath = uploadpath;
|
String fileName = null;
|
File file = new File(ctxPath + File.separator + bizPath + File.separator);
|
if (!file.exists()) {
|
file.mkdirs();// 创建文件根目录
|
}
|
String orgName = mf.getOriginalFilename();// 获取文件名
|
orgName = CommonUtils.getFileName(orgName);
|
if (orgName.indexOf(".") != -1) {
|
fileName = orgName.substring(0, orgName.lastIndexOf(".")) + "_" + System.currentTimeMillis() + orgName.substring(orgName.indexOf("."));
|
} else {
|
fileName = orgName + "_" + System.currentTimeMillis();
|
}
|
String savePath = file.getPath() + File.separator + fileName;
|
File savefile = new File(savePath);
|
FileCopyUtils.copy(mf.getBytes(), savefile);
|
String dbpath = null;
|
if (oConvertUtils.isNotEmpty(bizPath)) {
|
dbpath = bizPath + File.separator + fileName;
|
} else {
|
dbpath = fileName;
|
}
|
if (dbpath.contains("\\")) {
|
dbpath = dbpath.replace("\\", "/");
|
}
|
return dbpath;
|
} catch (IOException e) {
|
log.error(e.getMessage(), e);
|
}
|
return "";
|
}
|
|
/**
|
* 查询用户信息
|
* @param idSet
|
* @return
|
*/
|
private Map<String, UserAvatar> queryUserAvatar(Set<String> idSet){
|
List<UserAvatar> list = this.baseMapper.queryUserAvatarList(idSet);
|
Map<String, UserAvatar> map = new HashMap<>();
|
if(list!=null && list.size()>0){
|
for(UserAvatar user: list){
|
map.put(user.getId(), user);
|
}
|
}
|
return map;
|
}
|
|
}
|
