1. 修复bug

master
xueqingkun 1 week ago
parent 2205b05d85
commit 52dcd6996a

@ -1,5 +1,7 @@
package com.supervision.police.domain; package com.supervision.police.domain;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.annotation.FieldFill; import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.TableField; import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId; import com.baomidou.mybatisplus.annotation.TableId;
@ -131,4 +133,29 @@ public class NotePrompt implements Serializable {
@TableField(exist = false) @TableField(exist = false)
private static final long serialVersionUID = 1L; private static final long serialVersionUID = 1L;
public String findRelationValue(){
return filterValue("relation");
}
public String findHeadEntityValue(){
return filterValue("headEntityType");
}
public String findTailEntityValue(){
return filterValue("tailEntityType");
}
private String filterValue(String templateName){
if (CollUtil.isEmpty(this.tripleList)){
return null;
}
// 匹配出关系的value
TripleInfoDTO tripleInfoDTO1 = this.tripleList.stream()
.filter(tripleInfoDTO -> StrUtil.equals(tripleInfoDTO.getTemplateName(), templateName)).findFirst().orElse(null);
if (tripleInfoDTO1 == null){
return null;
}
return tripleInfoDTO1.getValue();
}
} }

@ -158,7 +158,7 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
boolean save; boolean save;
if (StringUtils.isEmpty(prompt.getId())) { if (StringUtils.isEmpty(prompt.getId())) {
// 新增的时候,校验是否已经存在相同的三元组关系,如果已经存在了相同的三元组关系,不允许添加 // 新增的时候,校验是否已经存在相同的三元组关系,如果已经存在了相同的三元组关系,不允许添加
if (checkHasSameTriple(prompt.getStartEntityType(), prompt.getRelType(), prompt.getEndEntityType(), null)) { if (checkHasSameTriple(prompt.findHeadEntityValue(), prompt.findRelationValue(), prompt.findTailEntityValue(), null)) {
return R.fail("已存在相同的三元组关系,不允许添加"); return R.fail("已存在相同的三元组关系,不允许添加");
} }
save = notePromptService.save(prompt); save = notePromptService.save(prompt);
@ -170,7 +170,7 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
notePromptTypeRelService.save(rel); notePromptTypeRelService.save(rel);
} }
} else { } else {
if (checkHasSameTriple(prompt.getStartEntityType(), prompt.getRelType(), prompt.getEndEntityType(), prompt.getId())) { if (checkHasSameTriple(prompt.findHeadEntityValue(), prompt.findRelationValue(), prompt.findTailEntityValue(), prompt.getId())) {
return R.fail("已存在相同的三元组关系,不允许添加"); return R.fail("已存在相同的三元组关系,不允许添加");
} }
save = notePromptService.updateById(prompt); save = notePromptService.updateById(prompt);

Loading…
Cancel
Save