将集合划分成指定大小的小集合
最近做项目遇到某些接口对点位数量有限制(最大1000个),因此,在获取点位数量超过1000个时,需要对接口进行分段调用,集合分段方法如下:
public static <T> List<List<T>> split(List<T> resList, int count) {
if (resList == null || count < 1) {
return null;
}
List<List<T>> ret = new ArrayList<List<T>>();
int size = resList.size();
if (size <= count) {
// 数据量不足count指定的大小
ret.add(resList);
} else {
int pre = size / count;
int last = size % count;
// 前面pre个集合,每个大小都是count个元素
for (int i = 0; i < pre; i++) {
List<T> itemList = new ArrayList<T>();
for (int j = 0; j < count; j++) {
itemList.add(resList.get(i * count + j));
}
ret.add(itemList);
}
// last的进行处理
if (last > 0) {
List<T> itemList = new ArrayList<T>();
for (int i = 0; i < last; i++) {
itemList.add(resList.get(pre * count + i));
}
ret.add(itemList);
}
}
return ret;
}
实际应用用法
/**
* 获取监控点的所有tags,适用于监控点数量无限,监控点个数最大1000,超过1000需要分段
*/
public List<CameraDTOV3> getCameraInfoHasTags(List<String> indexCodes) {
List<CameraDTOV3> result = new ArrayList<>();
if (CollectionUtils.isEmpty(indexCodes)) {
return result;
}
//如果监控点编号数量<1000,不用分段,直接调用,获取所有的监控点信息
int maxCount = 1000;
int size = indexCodes.size();
if (size <= maxCount) {
//不用分段,直接获取所有监控点的tags
result = irdsClient.cameraInfoV3(indexCodes);
} else {
List<List<String>> ret = split(indexCodes, maxCount);
for (int i = 0; i < ret.size(); i++) {
List<CameraDTOV3> cameras = irdsClient.cameraInfoV3(ret.get(i));
for (CameraDTOV3 cameraDTOV3 : cameras) {
result.add(cameraDTOV3);
}
}
}
return result;
}