新增/修改数据时校验字段唯一性

假定有实体类RuleAssign.java,对应如下表:

id type rule
1 a 1001
2 b 1002
3 c 1003

一、单字段校验

前端传过来一批数据List datas,需要对rule字段校验唯一性。这批数据既有新增数据,也有修改数据。

第一步,判断datas自身是否有重复rule字段。

Set 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 insert = datas.stream().filter(data -> data.getId == null).collect(Collectors.toList());

List update = datas.stream().filter(data -> data.getId != null).collect(Collectors.toList());

第二步,处理新增数据,判断insert中的rule在数据库中是否有重复值。将rule字段提取成一个List,然后使用sql查询where in语法判断即可。


List rules = insert.stream().map(RuleAssign::getRule).collect(Collectors.toList());

第三步,处理修改数据,判断update中的rule在数据库中是否有重复,对于每条数据有两种情况。A:该条数据未修改rule字段,此时不需要判断rule唯一性,忽略该条数据。B:该条数据修改了rule字段,此时需要判断唯一性。下面我们进行两次TreeSet去重即可验证重复。

  1. 将数据库中所有数据查出来List origin,然后将update添加进去,此时origin中有原数据和修改后的数据。
origin.addAll(update);
  1. 将 A 情况的数据过滤掉,idrule相同的数据即是未修改rule字段的数据。

Set set = new TreeSet<>(Comparator.comparing(RuleAssign::getRule)
                                          .thenComparing(RuleAssign::getId));

  1. 此时的origin中有原数据和修改过rule字段的数据,再按照第一步的方法校验originrule的唯一性即可完成校验。
Set set = new TreeSet<>(Comparator.comparing(RuleAssign::getRule));

set.addAll(datas);
if (set.size() < data.size()){
   throw new Exception("rule在DB中不唯一");
}

二、多字段联合校验

判断流程和单字段一样,只需要在上述过程中的Comparator后面添加需要判断的字段即可,例如联合校验ruletype

Set set = new TreeSet<>(Comparator.comparing(RuleAssign::getRule)
                                          .thenComparing(RuleAssign::getType));

合理利用Comparator构造器即可完成各种情况下的重复判断。

你可能感兴趣的:(新增/修改数据时校验字段唯一性)