diff --git a/src/main/java/com/huawei/datashow/service/HandleDataSourceServiceImpl.java b/src/main/java/com/huawei/datashow/service/HandleDataSourceServiceImpl.java index e96116fa8abae04b00a76f4ebabcf21942a257d3..ad71da24e71ebe29c39f505395697f41a01bc2d6 100644 --- a/src/main/java/com/huawei/datashow/service/HandleDataSourceServiceImpl.java +++ b/src/main/java/com/huawei/datashow/service/HandleDataSourceServiceImpl.java @@ -30,19 +30,17 @@ public class HandleDataSourceServiceImpl implements HandleDataSourceService { int count = (int) arrayList.get(0).get("count"); int batch = count / 1000; - sql = sql + " limit %d, %d"; - if (batch > 0) { for (int i = 0; i < batch; i++) { - String jsonListMap = openGaussDataBaseServiceImpl.getSourceData(pollName, String.format(sql, i * 1000, 1000)); + String jsonListMap = openGaussDataBaseServiceImpl.getSourceData(pollName, sql + String.format(" limit %d, %d", i * 1000, 1000)); List listMap = JSON.parseObject(jsonListMap, ArrayList.class); CSVUtil.writeCSVFile(dataSourceName, listMap, i != 0); } - String jsonListMap = openGaussDataBaseServiceImpl.getSourceData(pollName, String.format(sql, batch * 1000, count - batch * 1000 + 1)); + String jsonListMap = openGaussDataBaseServiceImpl.getSourceData(pollName, sql + String.format(" limit %d, %d", batch * 1000, count - batch * 1000 + 1)); List listMap = JSON.parseObject(jsonListMap, ArrayList.class); CSVUtil.writeCSVFile(dataSourceName, listMap, true); } else { - String jsonListMap = openGaussDataBaseServiceImpl.getSourceData(pollName, String.format(sql, 0, count)); + String jsonListMap = openGaussDataBaseServiceImpl.getSourceData(pollName, sql + String.format(" limit %d, %d", 0, count)); List listMap = JSON.parseObject(jsonListMap, ArrayList.class); CSVUtil.writeCSVFile(dataSourceName, listMap, false); }