假定有实体类RuleAssign.java
,对应如下表:
id | type | rule |
---|---|---|
1 | a | 1001 |
2 | b | 1002 |
3 | c | 1003 |
一、单字段校验
前端传过来一批数据List<ruleAssign> datas
,需要对rule字段校验唯一性。这批数据既有新增数据,也有修改数据。
第一步,判断datas
自身是否有重复rule字段。
Set<RuleAssign> set = new TreeSet<>(Comparator.comparing(RuleAssign::getRule));
set.addAll(datas);
if (set.size() < datas.size()){
throw new Exception("rule在list中不唯一");
}
这里用到了TreeSet有序集合,通过Comparator定义比较字段,将datas
去重,如果去重后的集合比原集合size小,则说明有重复数据,即有重复rule字段。
然后通过判断id将数据分为 新增 和 修改 两批。
List<ruleAssign> insert = datas.stream().filter(data -> data.getId == null).collect(Collectors.toList());
List<ruleAssign> update = datas.stream().filter(data -> data.getId != null).collect(Collectors.toList());
第二步,处理新增数据,判断insert
中的rule
在数据库中是否有重复值。将rule字段提取成一个List,然后使用sql查询where in
语法判断即可。
List<String> rules = insert.stream().map(RuleAssign::getRule).collect(Collectors.toList());
第三步,处理修改数据,判断update
中的rule
在数据库中是否有重复,对于每条数据有两种情况。A:该条数据未修改rule
字段,此时不需要判断rule
唯一性,忽略该条数据。B:该条数据修改了rule
字段,此时需要判断唯一性。下面我们进行两次TreeSet去重即可验证重复。
- 将数据库中所有数据查出来
List<ruleAssign> origin
,然后将update
添加进去,此时origin
中有原数据和修改后的数据。
origin.addAll(update);
- 将 A 情况的数据过滤掉,
id
和rule
相同的数据即是未修改rule
字段的数据。
Set<RuleAssign> set = new TreeSet<>(Comparator.comparing(RuleAssign::getRule)
.thenComparing(RuleAssign::getId));
- 此时的
origin
中有原数据和修改过rule
字段的数据,再按照第一步的方法校验origin
中rule
的唯一性即可完成校验。
Set<RuleAssign> set = new TreeSet<>(Comparator.comparing(RuleAssign::getRule));
set.addAll(datas);
if (set.size() < data.size()){
throw new Exception("rule在DB中不唯一");
}
二、多字段联合校验
判断流程和单字段一样,只需要在上述过程中的Comparator
后面添加需要判断的字段即可,例如联合校验rule
和type
Set<RuleAssign> set = new TreeSet<>(Comparator.comparing(RuleAssign::getRule)
.thenComparing(RuleAssign::getType));
合理利用Comparator构造器即可完成各种情况下的重复判断。