package com.zhitan.knowledgeBase.service.impl;
|
|
import cn.hutool.core.bean.BeanUtil;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.zhitan.common.constant.MessageConstant;
|
import com.zhitan.common.exception.ServiceException;
|
import com.zhitan.knowledgeBase.domain.dto.KnowledgeBaseAddDTO;
|
import com.zhitan.knowledgeBase.domain.dto.KnowledgeBaseEditDTO;
|
import com.zhitan.knowledgeBase.domain.dto.KnowledgeBasePageDTO;
|
import com.zhitan.knowledgeBase.domain.entity.KnowledgeBase;
|
import com.zhitan.knowledgeBase.domain.entity.KnowledgeBaseFile;
|
import com.zhitan.knowledgeBase.domain.enums.EnergyTypeEnum;
|
import com.zhitan.knowledgeBase.domain.vo.KnowledgeBaseDetailVO;
|
import com.zhitan.knowledgeBase.domain.vo.KnowledgeBasePageVO;
|
import com.zhitan.knowledgeBase.mapper.KnowledgeBaseFileMapper;
|
import com.zhitan.knowledgeBase.mapper.KnowledgeBaseMapper;
|
import com.zhitan.knowledgeBase.service.IKnowledgeBaseService;
|
import org.apache.commons.lang3.ObjectUtils;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import javax.annotation.Resource;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* 知识库(KnowledgeBase)表服务实现类
|
*
|
* @author makejava
|
* @since 2025-01-10 15:05:28
|
*/
|
@Service
|
public class KnowledgeBaseServiceImpl extends ServiceImpl<KnowledgeBaseMapper, KnowledgeBase> implements IKnowledgeBaseService {
|
|
@Resource
|
private KnowledgeBaseFileMapper fileMapper;
|
|
|
/**
|
* 知识库-列表查询
|
*/
|
public Page<KnowledgeBasePageVO> getPageList(KnowledgeBasePageDTO pageDTO) {
|
Page<KnowledgeBasePageVO> voPage = new Page<>(pageDTO.getPageNum(), pageDTO.getPageSize());
|
|
// 分页查询
|
Page<KnowledgeBase> knowledgeBasePage = baseMapper.selectPage(new Page<>(pageDTO.getPageNum(), pageDTO.getPageSize()),
|
Wrappers.<KnowledgeBase>lambdaQuery()
|
.like(ObjectUtils.isNotEmpty(pageDTO.getTitle()), KnowledgeBase::getTitle, pageDTO.getTitle())
|
.eq(ObjectUtils.isNotEmpty(pageDTO.getType()), KnowledgeBase::getType, pageDTO.getType())
|
.eq(KnowledgeBase::getDelFlag, false).orderByDesc(KnowledgeBase::getCreateTime));
|
|
List<KnowledgeBase> records = knowledgeBasePage.getRecords();
|
if (records.isEmpty()){
|
return voPage;
|
}
|
// 数据转换
|
voPage.setRecords(records.stream().map(item -> {
|
KnowledgeBasePageVO pageVO = new KnowledgeBasePageVO();
|
BeanUtil.copyProperties(item, pageVO);
|
pageVO.setTypeDesc(EnergyTypeEnum.getDescByCode(item.getType()));
|
return pageVO;
|
}).collect(Collectors.toList()));
|
voPage.setTotal(knowledgeBasePage.getTotal());
|
return voPage;
|
}
|
|
/**
|
* 知识库-查询详情
|
*/
|
public KnowledgeBaseDetailVO getDetail(Long id) {
|
KnowledgeBaseDetailVO vo = new KnowledgeBaseDetailVO();
|
|
// 查询数据
|
KnowledgeBase knowledgeBase = baseMapper.selectById(id);
|
if (knowledgeBase != null){
|
// 数据转换
|
BeanUtil.copyProperties(knowledgeBase, vo);
|
vo.setUrl(fileMapper.selectList(Wrappers.<KnowledgeBaseFile>lambdaQuery()
|
.select(KnowledgeBaseFile::getUrl)
|
.eq(KnowledgeBaseFile::getKnowledgeBaseId, id))
|
.stream().map(KnowledgeBaseFile::getUrl).collect(Collectors.toList()));
|
}
|
return vo;
|
}
|
|
/**
|
* 知识库-新增
|
*/
|
@Transactional(rollbackFor = Exception.class)
|
public void add(KnowledgeBaseAddDTO addDTO) {
|
// 插入数据
|
KnowledgeBase knowledgeBase = new KnowledgeBase();
|
BeanUtil.copyProperties(addDTO, knowledgeBase);
|
baseMapper.insert(knowledgeBase);
|
|
// 插入文件
|
if (ObjectUtils.isNotEmpty(addDTO.getUrl())){
|
addDTO.getUrl().forEach(url->{
|
KnowledgeBaseFile file = new KnowledgeBaseFile();
|
file.setKnowledgeBaseId(knowledgeBase.getId());
|
file.setUrl(url);
|
fileMapper.insert(file);
|
});
|
}
|
}
|
|
/**
|
* 知识库-更新
|
*/
|
@Transactional(rollbackFor = Exception.class)
|
public void edit(KnowledgeBaseEditDTO editDTO) {
|
// 查询数据
|
KnowledgeBase knowledgeBase = baseMapper.selectById(editDTO.getId());
|
if (ObjectUtils.isEmpty(knowledgeBase)){
|
throw new ServiceException(MessageConstant.KNOWLEDGE_BASE_NOT_EXIST);
|
}
|
// 更新数据
|
BeanUtils.copyProperties(editDTO, knowledgeBase);
|
baseMapper.updateById(knowledgeBase);
|
|
// 删除文件
|
fileMapper.delete(Wrappers.<KnowledgeBaseFile>lambdaQuery()
|
.eq(KnowledgeBaseFile::getKnowledgeBaseId, knowledgeBase.getId()));
|
|
// 插入文件
|
if (ObjectUtils.isNotEmpty(editDTO.getUrl())){
|
editDTO.getUrl().forEach(url->{
|
KnowledgeBaseFile file = new KnowledgeBaseFile();
|
file.setKnowledgeBaseId(knowledgeBase.getId());
|
file.setUrl(url);
|
fileMapper.insert(file);
|
});
|
}
|
}
|
|
/**
|
* 知识库-删除
|
*/
|
@Transactional(rollbackFor = Exception.class)
|
public void delete(Long id) {
|
// 查询数据
|
KnowledgeBase knowledgeBase = baseMapper.selectById(id);
|
if (ObjectUtils.isEmpty(knowledgeBase)){
|
throw new ServiceException(MessageConstant.KNOWLEDGE_BASE_NOT_EXIST);
|
}
|
// 删除数据
|
knowledgeBase.setDelFlag(true);
|
baseMapper.updateById(knowledgeBase);
|
|
// 删除文件
|
fileMapper.delete(Wrappers.<KnowledgeBaseFile>lambdaQuery()
|
.eq(KnowledgeBaseFile::getKnowledgeBaseId, id));
|
}
|
}
|
