有个需求 流程明细中 申领礼品,礼品支持重复 多行申领
类似下图:
解决思路 目前是 获取明细表中所有相同的物品 然后申领数量相加 然后判断库存数
所以这里想的是 存个map集合搞
大致代码如下:
RecordSet rs = new RecordSet();
rs.executeQuery(getQuerySql(billTableName), requestId);
RecordSet model = new RecordSet();
Multimap<String, Integer> multiMap = ArrayListMultimap.create();
while (rs.next()) {
String dataId = rs.getString(1);
String sl = Util.null2String(rs.getString(2), "0");
multiMap.put(dataId, Integer.parseInt(sl));
}
Map<String, String> sum = getSum(multiMap);
for (Map.Entry<String, String> entry : sum.entrySet()) {
String dataId = entry.getKey();
String sl = entry.getValue();
model.executeQuery(" select kc from " + TABLENAME + " where id =? and kc >=" + sl + " ", dataId);
if (!model.next()) {
logger.info("礼品库数据不足!requestId:{}" + requestId + "dataId:" + dataId);
throw new MyWorkflowException(" 领用明细表中 " + getDataName(dataId) + " 礼品的库存不足!");
}
}
//传入guava 的Multimap 然后循环求出 每个key 合计的领用数量
//使用stream 比较简洁
public Map<String, String> getSum(Multimap<String, Integer> maps) {
checkNotNull(maps);
Map<String, String> resultMap = Maps.newHashMap();
Map map = maps.asMap();
Iterator it = map.keySet().iterator();
while (it.hasNext()) {
String key = (String) it.next();
List<Integer> value = (List<Integer>) map.get(key);
Integer sum = value.stream().reduce(0, (a, b) -> a + b);
resultMap.put(key, sum.toString());
}
return resultMap;
}
目前这是第一版方案,后续有优化 会更新!