|
|
@@ -0,0 +1,139 @@
|
|
|
+package com.etotem.cfc.service;
|
|
|
+
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
|
+import com.etotem.cfc.entity.DanKnowledgeBase;
|
|
|
+import com.etotem.cfc.entity.DanKnowledgeBaseDimension;
|
|
|
+import com.etotem.cfc.entity.DanKnowledgeBaseTag;
|
|
|
+import com.etotem.cfc.mapper.DanKnowledgeBaseDimensionMapper;
|
|
|
+import com.etotem.cfc.mapper.DanKnowledgeBaseMapper;
|
|
|
+import com.etotem.cfc.mapper.DanKnowledgeBaseTagMapper;
|
|
|
+import org.springframework.stereotype.Service;
|
|
|
+import org.springframework.transaction.annotation.Transactional;
|
|
|
+
|
|
|
+import javax.annotation.Resource;
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+import java.util.stream.Collectors;
|
|
|
+
|
|
|
+@Service
|
|
|
+public class DanKnowledgeBaseService extends ServiceImpl<DanKnowledgeBaseMapper, DanKnowledgeBase> {
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private DanKnowledgeBaseDimensionMapper danKnowledgeBaseDimensionMapper;
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private DanKnowledgeBaseTagMapper danKnowledgeBaseTagMapper;
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private DifySyncService difySyncService;
|
|
|
+
|
|
|
+ public Map<String, Object> list(String keyword, Integer status, int page, int size) {
|
|
|
+ Page<DanKnowledgeBase> p = new Page<>(page, size);
|
|
|
+ LambdaQueryWrapper<DanKnowledgeBase> wrapper = new LambdaQueryWrapper<>();
|
|
|
+ if (keyword != null && !keyword.isEmpty()) {
|
|
|
+ wrapper.and(w -> w.like(DanKnowledgeBase::getTitle, keyword)
|
|
|
+ .or().like(DanKnowledgeBase::getContent, keyword));
|
|
|
+ }
|
|
|
+ if (status != null) {
|
|
|
+ wrapper.eq(DanKnowledgeBase::getStatus, status);
|
|
|
+ }
|
|
|
+ wrapper.orderByAsc(DanKnowledgeBase::getSort);
|
|
|
+ this.page(p, wrapper);
|
|
|
+ Map<String, Object> data = new HashMap<>();
|
|
|
+ data.put("records", p.getRecords());
|
|
|
+ data.put("total", p.getTotal());
|
|
|
+ data.put("page", p.getCurrent());
|
|
|
+ data.put("size", p.getSize());
|
|
|
+ return data;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Transactional(rollbackFor = Exception.class)
|
|
|
+ public DanKnowledgeBase saveKnowledge(DanKnowledgeBase knowledge) {
|
|
|
+ if (knowledge.getId() != null) {
|
|
|
+ this.updateById(knowledge);
|
|
|
+ } else {
|
|
|
+ this.save(knowledge);
|
|
|
+ }
|
|
|
+ return knowledge;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Transactional(rollbackFor = Exception.class)
|
|
|
+ public DanKnowledgeBase saveWithAssociations(DanKnowledgeBase knowledge,
|
|
|
+ List<Long> dimensionIds,
|
|
|
+ List<Long> tagIds) {
|
|
|
+ if (knowledge.getId() != null) {
|
|
|
+ this.updateById(knowledge);
|
|
|
+ } else {
|
|
|
+ this.save(knowledge);
|
|
|
+ }
|
|
|
+ if (dimensionIds != null) {
|
|
|
+ bindDimensions(knowledge.getId(), dimensionIds);
|
|
|
+ }
|
|
|
+ if (tagIds != null) {
|
|
|
+ bindTags(knowledge.getId(), tagIds);
|
|
|
+ }
|
|
|
+ return knowledge;
|
|
|
+ }
|
|
|
+
|
|
|
+ public Map<String, Object> getWithAssociations(Long id) {
|
|
|
+ DanKnowledgeBase knowledge = this.getById(id);
|
|
|
+ if (knowledge == null) {
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+ List<Long> dimensionIds = danKnowledgeBaseDimensionMapper.selectList(
|
|
|
+ new LambdaQueryWrapper<DanKnowledgeBaseDimension>()
|
|
|
+ .eq(DanKnowledgeBaseDimension::getKnowledgeId, id))
|
|
|
+ .stream().map(DanKnowledgeBaseDimension::getDimensionId)
|
|
|
+ .collect(Collectors.toList());
|
|
|
+ List<Long> tagIds = danKnowledgeBaseTagMapper.selectList(
|
|
|
+ new LambdaQueryWrapper<DanKnowledgeBaseTag>()
|
|
|
+ .eq(DanKnowledgeBaseTag::getKnowledgeId, id))
|
|
|
+ .stream().map(DanKnowledgeBaseTag::getTagId)
|
|
|
+ .collect(Collectors.toList());
|
|
|
+ Map<String, Object> result = new HashMap<>();
|
|
|
+ result.put("knowledge", knowledge);
|
|
|
+ result.put("dimensionIds", dimensionIds);
|
|
|
+ result.put("tagIds", tagIds);
|
|
|
+ return result;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Transactional(rollbackFor = Exception.class)
|
|
|
+ public boolean deleteKnowledge(Long id) {
|
|
|
+ danKnowledgeBaseDimensionMapper.delete(
|
|
|
+ new LambdaQueryWrapper<DanKnowledgeBaseDimension>()
|
|
|
+ .eq(DanKnowledgeBaseDimension::getKnowledgeId, id));
|
|
|
+ danKnowledgeBaseTagMapper.delete(
|
|
|
+ new LambdaQueryWrapper<DanKnowledgeBaseTag>()
|
|
|
+ .eq(DanKnowledgeBaseTag::getKnowledgeId, id));
|
|
|
+ return this.removeById(id);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Transactional(rollbackFor = Exception.class)
|
|
|
+ public void bindDimensions(Long knowledgeId, List<Long> dimensionIds) {
|
|
|
+ danKnowledgeBaseDimensionMapper.delete(
|
|
|
+ new LambdaQueryWrapper<DanKnowledgeBaseDimension>()
|
|
|
+ .eq(DanKnowledgeBaseDimension::getKnowledgeId, knowledgeId));
|
|
|
+ for (Long dimensionId : dimensionIds) {
|
|
|
+ DanKnowledgeBaseDimension rel = new DanKnowledgeBaseDimension();
|
|
|
+ rel.setKnowledgeId(knowledgeId);
|
|
|
+ rel.setDimensionId(dimensionId);
|
|
|
+ danKnowledgeBaseDimensionMapper.insert(rel);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ @Transactional(rollbackFor = Exception.class)
|
|
|
+ public void bindTags(Long knowledgeId, List<Long> tagIds) {
|
|
|
+ danKnowledgeBaseTagMapper.delete(
|
|
|
+ new LambdaQueryWrapper<DanKnowledgeBaseTag>()
|
|
|
+ .eq(DanKnowledgeBaseTag::getKnowledgeId, knowledgeId));
|
|
|
+ for (Long tagId : tagIds) {
|
|
|
+ DanKnowledgeBaseTag rel = new DanKnowledgeBaseTag();
|
|
|
+ rel.setKnowledgeId(knowledgeId);
|
|
|
+ rel.setTagId(tagId);
|
|
|
+ danKnowledgeBaseTagMapper.insert(rel);
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|