package com.whyc.util;
|
|
import com.github.pagehelper.PageInfo;
|
import com.whyc.constant.Com;
|
import com.whyc.dto.paramter.AlarmPar;
|
import com.whyc.dto.result.AlarmRes;
|
import com.whyc.mapper.BattalarmDataHistoryMapper;
|
import com.whyc.mapper.CommonMapper;
|
import com.whyc.mapper.UserLogMapper;
|
import com.whyc.pojo.UserLog;
|
import org.springframework.stereotype.Component;
|
|
import javax.annotation.Resource;
|
import java.text.ParseException;
|
import java.util.*;
|
|
/**
|
* 数据量极大的表,按照年份分表,分页查询时获取数据
|
*/
|
@Component
|
public class SubTablePageInfoUtils {
|
|
@Resource
|
private CommonMapper commonMapper;
|
|
@Resource
|
private UserLogMapper userLogMapper;
|
|
@Resource
|
private BattalarmDataHistoryMapper battAlarmDataHistoryMapper;
|
|
public PageInfo<Object> getPageInfo(int pageNum,int pageSize,
|
Date startTime,Date endTime,
|
String dbName,String tablePrefix,
|
Object pojo) throws ParseException {
|
Map<String, List<Date>> queryTimeForSubTables = DateUtil.getQueryTimeForSubTablesDesc(startTime, endTime);
|
//查询分表是否存在,存在则查询结果
|
Map<String,Integer> queryCountMap = new LinkedHashMap<>();
|
Set<String> tableYearKeySet = queryTimeForSubTables.keySet();
|
for (String tableYear : tableYearKeySet) {
|
List<Date> queryTime = queryTimeForSubTables.get(tableYear);
|
|
String tableName;
|
if(!tableYear.equals("default")){
|
//数值
|
tableName = tablePrefix+"_"+tableYear;
|
String existTableName = commonMapper.existTable(dbName, tableName);
|
if(existTableName == null){
|
continue;
|
}
|
}else{
|
tableName = tablePrefix;
|
}
|
//====== 根据不同类型类型对象对应调整 ======
|
if(pojo instanceof UserLog) {
|
UserLog userLog = (UserLog) pojo;
|
userLog.setOperationTime(queryTime.get(0));
|
userLog.setOperationTime2(queryTime.get(1));
|
userLog.setRecordYear(tableName);
|
int currentCount = userLogMapper.getCount(userLog);
|
queryCountMap.put(tableYear,currentCount);
|
}else if(pojo instanceof AlarmPar){ //电池告警历史
|
AlarmPar param = (AlarmPar) pojo;
|
param.setAlmStartTime(queryTime.get(0));
|
param.setAlmStartTime1(queryTime.get(1));
|
param.setRecordYear(tableName);
|
int currentCount = battAlarmDataHistoryMapper.getCount(param);
|
queryCountMap.put(tableYear,currentCount);
|
}
|
}
|
|
//分页信息
|
//确认总页数,总记录数
|
PageInfo<Object> pageInfo = new PageInfo<>();
|
|
int total = 0;
|
Set<String> queryKeySet = queryCountMap.keySet();
|
for (String queryKey : queryKeySet) {
|
int size = queryCountMap.get(queryKey);
|
total+=size;
|
}
|
int pages = (int) Math.ceil(Float.parseFloat(String.valueOf(total))/pageSize);
|
pageInfo.setTotal(total);
|
pageInfo.setPages(pages);
|
pageInfo.setPageNum(pageNum);
|
pageInfo.setPageSize(pageSize);
|
//根据当前页所需记录,查询当前页记录
|
int startNum = (pageNum-1)*pageSize+1;
|
int endNum = pageNum*pageSize;
|
|
//最后一个算法:上面不应该先查询所有记录,应该取count. 这后面定位到哪个表或哪几张表后,采取limit获取当前页记录数;
|
//格式:{表名,[limit 2,20]}
|
Map<String,List<Integer>> tableAndLimitMap = MathUtil.getQueryTableAndLimit(startNum,endNum,pageSize,queryCountMap);
|
Set<String> keySet = tableAndLimitMap.keySet();
|
List<Object> dataList = new LinkedList<>();
|
for (String key : keySet) {
|
List<Date> queryTime = queryTimeForSubTables.get(key);
|
//====== 根据不同类型类型对象对应调整 ======
|
String recordYear = key.equals("default") ? tablePrefix : tablePrefix + "_" + key;
|
if(pojo instanceof UserLog) {
|
UserLog userLog = (UserLog) pojo;
|
userLog.setOperationTime(queryTime.get(0));
|
userLog.setOperationTime2(queryTime.get(1));
|
|
List<Integer> limitList = tableAndLimitMap.get(key);
|
userLog.setLimitStart(limitList.get(0));
|
userLog.setLimitEnd(limitList.get(1));
|
userLog.setRecordYear(recordYear);
|
List<UserLog> list = userLogMapper.getList2(userLog);
|
dataList.addAll(list);
|
} else if (pojo instanceof AlarmPar) {
|
AlarmPar data = (AlarmPar) pojo;
|
data.setAlmStartTime(queryTime.get(0));
|
data.setAlmStartTime1(queryTime.get(1));
|
|
List<Integer> limitList = tableAndLimitMap.get(key);
|
data.setLimitStart(limitList.get(0));
|
data.setLimitEnd(limitList.get(1));
|
data.setRecordYear(recordYear);
|
List<AlarmRes> list = battAlarmDataHistoryMapper.getList2(data);
|
dataList.addAll(list);
|
}
|
}
|
pageInfo.setList(dataList);
|
return pageInfo;
|
}
|
}
|
