in backend/src/main/java/org/apache/iotdb/admin/service/impl/IotDBServiceImpl.java [4414:4463]
private <T> CountDTO executeQuery(
Class<T> clazz,
SessionPool sessionPool,
String sql,
Integer pageSize,
Integer pageNum,
String keyword)
throws BaseException {
SessionDataSetWrapper sessionDataSetWrapper = null;
try {
sessionDataSetWrapper = sessionPool.executeQueryStatement(sql);
List<T> results = new ArrayList<>();
int batchSize = sessionDataSetWrapper.getBatchSize();
int count = 0;
if (batchSize > 0) {
while (sessionDataSetWrapper.hasNext()) {
RowRecord rowRecord = sessionDataSetWrapper.next();
count++;
if (count >= pageSize * (pageNum - 1) + 1 && count <= pageSize * pageNum) {
T t = clazz.newInstance();
List<org.apache.iotdb.tsfile.read.common.Field> fields = rowRecord.getFields();
List<String> columnNames = sessionDataSetWrapper.getColumnNames();
for (int i = 0; i < fields.size(); i++) {
Field field = clazz.getDeclaredField(columnNames.get(i).replaceAll(" ", ""));
field.setAccessible(true);
field.set(t, fields.get(i).toString());
}
results.add(t);
}
}
}
CountDTO countDTO = new CountDTO();
countDTO.setObjects(results);
countDTO.setTotalCount(count);
Integer totalPage = count % pageSize == 0 ? count / pageSize : count / pageSize + 1;
countDTO.setTotalPage(totalPage);
return countDTO;
} catch (IoTDBConnectionException e) {
logger.error(e.getMessage());
throw new BaseException(ErrorCode.GET_MSM_FAIL, ErrorCode.GET_MSM_FAIL_MSG);
} catch (StatementExecutionException e) {
logger.error(e.getMessage());
throw new BaseException(ErrorCode.GET_MSM_FAIL, ErrorCode.GET_MSM_FAIL_MSG);
} catch (Exception e) {
logger.error(e.getMessage());
throw new BaseException(ErrorCode.GET_MSM_FAIL, ErrorCode.GET_MSM_FAIL_MSG);
} finally {
closeResultSet(sessionDataSetWrapper);
}
}