统计分析-导出Excel并生成图表;
This commit is contained in:
yu
parent
4d53e9541f
commit
4ed42b06fb
@ -0,0 +1,14 @@
|
||||
package com.das.common.constant;
|
||||
|
||||
/**
|
||||
* 统计分析
|
||||
*/
|
||||
public interface StatisticalAnalysisConstant {
|
||||
|
||||
|
||||
String TREND_ANALYSE = "趋势分析";
|
||||
|
||||
String POWER_CURVE = "功率曲线";
|
||||
|
||||
String TREND_CONTRAST = "趋势对比";
|
||||
}
|
||||
@ -5,6 +5,7 @@ import cn.hutool.core.util.RandomUtil;
|
||||
import cn.hutool.poi.excel.ExcelUtil;
|
||||
import cn.hutool.poi.excel.ExcelWriter;
|
||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||
import com.das.common.constant.StatisticalAnalysisConstant;
|
||||
import com.das.common.exceptions.ServiceException;
|
||||
import com.das.common.utils.BeanCopyUtils;
|
||||
import com.das.modules.curve.domain.entity.CurveItemEntity;
|
||||
@ -20,9 +21,13 @@ import com.das.modules.page.service.StatisticalAnalysisService;
|
||||
import jakarta.servlet.ServletOutputStream;
|
||||
import jakarta.servlet.http.HttpServletRequest;
|
||||
import jakarta.servlet.http.HttpServletResponse;
|
||||
import org.apache.poi.ss.usermodel.ClientAnchor;
|
||||
import org.apache.poi.ss.usermodel.HorizontalAlignment;
|
||||
import org.apache.poi.ss.usermodel.VerticalAlignment;
|
||||
import org.apache.poi.ss.usermodel.Workbook;
|
||||
import org.apache.poi.ss.util.CellRangeAddress;
|
||||
import org.apache.poi.xddf.usermodel.chart.*;
|
||||
import org.apache.poi.xssf.usermodel.XSSFChart;
|
||||
import org.apache.poi.xssf.usermodel.XSSFClientAnchor;
|
||||
import org.apache.poi.xssf.usermodel.XSSFDrawing;
|
||||
import org.apache.poi.xssf.usermodel.XSSFSheet;
|
||||
@ -82,19 +87,22 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
//获取Excel的列
|
||||
LinkedHashMap<String, String> map = getTrendColumnName(param);
|
||||
List<Map<String, Object>> dataList = new ArrayList<>();
|
||||
//图表数据集
|
||||
DefaultCategoryDataset dataset = new DefaultCategoryDataset();
|
||||
// 遍历数据，填充Excel和图表数据集
|
||||
setTrendAnalyseExcelValue(mapsList, dataList, map, dataset);
|
||||
// 遍历数据，填充Excel数据集
|
||||
setTrendAnalyseExcelValue(mapsList, dataList);
|
||||
//获取图表类别集
|
||||
List<String> chartKey = getCharKey(map);
|
||||
ExcelWriter writer = ExcelUtil.getWriter(RandomUtil.randomInt(100, 1000) + "statistics" + ".xlsx");
|
||||
//设置Excel样式
|
||||
setExcelStyle(writer, map, dataList);
|
||||
// 使用JFreeChart生成折线图
|
||||
createChart(dataList, writer, dataset, "趋势分析");
|
||||
//生成折线图
|
||||
addChartToExcel(writer,dataList,StatisticalAnalysisConstant.TREND_ANALYSE,chartKey);
|
||||
//下载Excel
|
||||
downloadExcel(response, writer,"趋势分析");
|
||||
downloadExcel(response, writer,StatisticalAnalysisConstant.TREND_ANALYSE);
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* 功率曲线Excel导出
|
||||
*
|
||||
@ -114,15 +122,15 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
setPowerCurveExcelValue(resultMap, dataList, curveItemEntitieList);
|
||||
//获取功率曲线的列
|
||||
LinkedHashMap<String, String> map = getPowerCurveColumnName();
|
||||
//获取图表数据集
|
||||
DefaultCategoryDataset dataset = getDefaultCategoryDataset(dataList);
|
||||
//获取图表类别集
|
||||
List<String> chartKey = getCharKey(map);
|
||||
ExcelWriter writer = ExcelUtil.getWriter(RandomUtil.randomInt(100, 1000) + "statistics" + ".xlsx");
|
||||
//设置Excel样式
|
||||
setExcelStyle(writer, map, dataList);
|
||||
//使用JFreeChart生成图表
|
||||
createChart(dataList, writer, dataset, "功率曲线分析");
|
||||
//生成折线图
|
||||
addChartToExcel(writer,dataList,StatisticalAnalysisConstant.POWER_CURVE,chartKey);
|
||||
//下载Excel
|
||||
downloadExcel(response, writer,"功率曲线分析");
|
||||
downloadExcel(response, writer,StatisticalAnalysisConstant.POWER_CURVE);
|
||||
}
|
||||
|
||||
|
||||
@ -140,44 +148,19 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
//自定义别名 别名的key和实体类中的名称要对应上！
|
||||
LinkedHashMap<String, String> map = gettrendContrastColumnName(param);
|
||||
List<Map<String, Object>> dataList = new ArrayList<>();
|
||||
//图表数据集
|
||||
DefaultCategoryDataset dataset = new DefaultCategoryDataset();
|
||||
//图表类别集
|
||||
List<String> charKey = getCharKey(map);
|
||||
// 遍历数据，将数据添加到dataList中
|
||||
setTrendContrastExcelValue(maps, dataList, map, dataset);
|
||||
setTrendContrastExcelValue(maps, dataList);
|
||||
ExcelWriter writer = ExcelUtil.getWriter(RandomUtil.randomInt(100, 1000) + "statistics" + ".xlsx");
|
||||
//设置Excel样式
|
||||
setExcelStyle(writer, map, dataList);
|
||||
// 使用JFreeChart生成折线图
|
||||
createChart(dataList, writer, dataset, "趋势对比");
|
||||
//生成折线图
|
||||
addChartToExcel(writer,dataList,StatisticalAnalysisConstant.TREND_CONTRAST,charKey);
|
||||
//下载Excel
|
||||
downloadExcel(response, writer,"趋势对比");
|
||||
downloadExcel(response, writer,StatisticalAnalysisConstant.TREND_CONTRAST);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取图表数据集
|
||||
*
|
||||
* @param dataList 数据集
|
||||
* @return 图表数据集
|
||||
*/
|
||||
private DefaultCategoryDataset getDefaultCategoryDataset(List<Map<String, Object>> dataList) {
|
||||
DefaultCategoryDataset dataset = new DefaultCategoryDataset();
|
||||
for (Map<String, Object> mapData : dataList) {
|
||||
Object time = mapData.get("time");
|
||||
Object speed = mapData.get("iWindSpeed");
|
||||
Object power = mapData.get("iGenPower");
|
||||
Object theorySpeed = mapData.get("theoryIWindSpeed");
|
||||
Object theoryPower = mapData.get("theoryIGenPower");
|
||||
if (speed != null && power != null) {
|
||||
dataset.addValue((Float) speed, "风速实际值", (String) time);
|
||||
dataset.addValue((Float) power, "功率实际值", (String) time);
|
||||
}
|
||||
if (theorySpeed != null && theoryPower != null) {
|
||||
dataset.addValue((Float) theorySpeed, "风速理论值", (String) time);
|
||||
dataset.addValue((Float) theoryPower, "功率理论值", (String) time);
|
||||
}
|
||||
}
|
||||
return dataset;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
@ -200,15 +183,11 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
|
||||
/**
|
||||
* 趋势分析-遍历数据，填充Excel和图表数据集
|
||||
*
|
||||
* @param mapsList 测点历史数据
|
||||
* @param dataList Excel数据集
|
||||
* @param map 表格的列
|
||||
* @param dataset 图表数据集
|
||||
*/
|
||||
private void setTrendAnalyseExcelValue(List<Map<String, Map<String, Map<String, Object>>>> mapsList,
|
||||
List<Map<String, Object>> dataList,
|
||||
LinkedHashMap<String, String> map, DefaultCategoryDataset dataset) {
|
||||
List<Map<String, Object>> dataList) {
|
||||
for (int i = 0; i < mapsList.size(); i++) {
|
||||
List<String> timesListstr = new ArrayList<>();
|
||||
List<Double> valuesList = new ArrayList<>();
|
||||
@ -237,10 +216,6 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
}
|
||||
String pointNameKey = pointName + num;
|
||||
String timeKey = "time" + num;
|
||||
//添加图表的数据集
|
||||
for (int j = 0; j < timesListstr.size(); j++) {
|
||||
dataset.addValue(valuesList.get(j), map.get(pointNameKey), timesListstr.get(j));
|
||||
}
|
||||
if (i == 0) {
|
||||
for (int j = 0; j < timesListstr.size(); j++) {
|
||||
Map<String, Object> dataMap = new HashMap<>();
|
||||
@ -258,16 +233,31 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取图表类别集
|
||||
* @param map Excel的列
|
||||
* @return 图表类别集
|
||||
*/
|
||||
private List<String> getCharKey(LinkedHashMap<String, String> map) {
|
||||
//获取图表类别集
|
||||
List<String> chartKey = new ArrayList<>();
|
||||
for (Map.Entry<String, String> stringStringEntry : map.entrySet()) {
|
||||
String value = stringStringEntry.getValue();
|
||||
if (!value.contains("时间")){
|
||||
chartKey.add(value);
|
||||
}
|
||||
}
|
||||
return chartKey;
|
||||
}
|
||||
|
||||
/**
|
||||
* 趋势对比填充Excel数据集
|
||||
*
|
||||
* @param dataList Excel数据集
|
||||
* @param map 表格的列
|
||||
* @param dataset Excel数据集
|
||||
* @param maps 测点数据
|
||||
*/
|
||||
private void setTrendContrastExcelValue(Map<String, Map<String, Map<String, Object>>> maps,
|
||||
List<Map<String, Object>> dataList,
|
||||
LinkedHashMap<String, String> map, DefaultCategoryDataset dataset) {
|
||||
List<Map<String, Object>> dataList) {
|
||||
for (Map.Entry<String, Map<String, Map<String, Object>>> stringMapEntry : maps.entrySet()) {
|
||||
int flagNum = 0;
|
||||
for (Map.Entry<String, Map<String, Object>> mapEntry : stringMapEntry.getValue().entrySet()) {
|
||||
@ -303,19 +293,12 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
}
|
||||
}
|
||||
flagNum++;
|
||||
//生成图表的数据集
|
||||
for (int j = 0; j < timesListstr.size(); j++) {
|
||||
dataset.addValue(valuesList.get(j), map.get(key), timesListstr.get(j));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 趋势-获取表格的列
|
||||
* *
|
||||
*
|
||||
* @param param 查询条件
|
||||
* @return 表格的列
|
||||
*/
|
||||
@ -360,8 +343,7 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
|
||||
/**
|
||||
* 获取功率曲线的列
|
||||
*
|
||||
* @return
|
||||
* @return Excel的列
|
||||
*/
|
||||
private LinkedHashMap<String, String> getPowerCurveColumnName() {
|
||||
LinkedHashMap<String, String> map = new LinkedHashMap<>();
|
||||
@ -458,6 +440,7 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
*
|
||||
* @param response 响应对象
|
||||
* @param writer Excel对象
|
||||
* @param title 标题
|
||||
*/
|
||||
private void downloadExcel(HttpServletResponse response, ExcelWriter writer,String title) {
|
||||
response.setCharacterEncoding(StandardCharsets.UTF_8.name());
|
||||
@ -468,7 +451,7 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
response.setDateHeader("Expires", 0);
|
||||
try {
|
||||
// 设置请求头属性
|
||||
response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(title+".xls", StandardCharsets.UTF_8));
|
||||
response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(title+"-"+System.currentTimeMillis()+".xls", StandardCharsets.UTF_8));
|
||||
ServletOutputStream out = response.getOutputStream();
|
||||
// 写出到文件
|
||||
writer.flush(out, true);
|
||||
@ -482,8 +465,7 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
}
|
||||
|
||||
/**
|
||||
* 使用JFreeChart生成折线图
|
||||
*
|
||||
* 使用JFreeChart生成折线图(图片),已弃用
|
||||
* @param data excel数据集
|
||||
* @param writer excel对象
|
||||
* @param dataset 图表数据集
|
||||
@ -554,6 +536,95 @@ public class StatisticalAnalysisServiceImpl implements StatisticalAnalysisServic
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成折线图,并插入到Excel中
|
||||
* @param writer Sheet对象
|
||||
* @param data 测点数据集合
|
||||
* @param titleText 标题
|
||||
* @param chartKey 类别数据集
|
||||
*/
|
||||
private static void addChartToExcel(ExcelWriter writer, List<Map<String, Object>> data,
|
||||
String titleText,List<String> chartKey) {
|
||||
try {
|
||||
// 获取Sheet对象
|
||||
XSSFSheet sheet = (XSSFSheet) writer.getSheet();
|
||||
// 创建绘图区域
|
||||
XSSFDrawing drawing = sheet.createDrawingPatriarch();
|
||||
int columnCount = sheet.getRow(0).getPhysicalNumberOfCells() + 1;
|
||||
//图表位置,插入到最后一列+1
|
||||
ClientAnchor anchor = drawing.createAnchor(0, 0, 0, 0, columnCount, 0, 15, 15);
|
||||
// 创建图表
|
||||
XSSFChart xssfChart = drawing.createChart(anchor);
|
||||
// 设置图表标题
|
||||
xssfChart.setTitleText(titleText);
|
||||
// 创建数据系列
|
||||
XDDFChartLegend legend = xssfChart.getOrAddLegend();
|
||||
legend.setPosition(LegendPosition.TOP_RIGHT);
|
||||
XDDFCategoryAxis bottomAxis = xssfChart.createCategoryAxis(AxisPosition.BOTTOM);
|
||||
XDDFValueAxis leftAxis = xssfChart.createValueAxis(AxisPosition.LEFT);
|
||||
leftAxis.setCrossBetween(AxisCrossBetween.BETWEEN);
|
||||
XDDFLineChartData chartData = (XDDFLineChartData) xssfChart.createData(ChartTypes.LINE, bottomAxis, leftAxis);
|
||||
// 遍历每个系列并添加数据
|
||||
setChartData(data, titleText, chartKey, chartData, sheet);
|
||||
xssfChart.plot(chartData);
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 给每个类别,添加数据
|
||||
* @param data 测点数据
|
||||
* @param titleText 标题
|
||||
* @param chartKey 类别key
|
||||
* @param chartData 图表数据
|
||||
* @param sheet sheet对象
|
||||
*/
|
||||
private static void setChartData(List<Map<String, Object>> data, String titleText, List<String> chartKey, XDDFLineChartData chartData, XSSFSheet sheet) {
|
||||
if (titleText.equals(StatisticalAnalysisConstant.TREND_ANALYSE)) {
|
||||
int strStartRow = 0;
|
||||
int strEndCol = 0;
|
||||
int numStartRow = 1;
|
||||
int numEndCol = 1;
|
||||
for (int i = 0; i < chartKey.size(); i++) {
|
||||
XDDFLineChartData.Series series = (XDDFLineChartData.Series) chartData.addSeries(
|
||||
createStringDataReference(sheet, 1, data.size(), strStartRow, strEndCol), // 类别（X轴）
|
||||
createNumericDataReference(sheet, 1, data.size(), numStartRow, numEndCol) // 值（Y轴）
|
||||
);
|
||||
series.setTitle(chartKey.get(i), null);
|
||||
strStartRow += 2;
|
||||
strEndCol += 2;
|
||||
numStartRow += 2;
|
||||
numEndCol += 2;
|
||||
}
|
||||
} else {
|
||||
for (int i = 0; i < chartKey.size(); i++) {
|
||||
XDDFLineChartData.Series series = (XDDFLineChartData.Series) chartData.addSeries(
|
||||
createStringDataReference(sheet, 1, data.size(), 0, 0), // 类别（X轴）
|
||||
createNumericDataReference(sheet, 1, data.size(), i + 1, i + 1) // 值（Y轴）
|
||||
);
|
||||
series.setTitle(chartKey.get(i), null);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建字符串数据引用
|
||||
* @return XDDFDataSource<String>
|
||||
*/
|
||||
private static XDDFDataSource<String> createStringDataReference(XSSFSheet sheet, int startRow, int endRow, int startCol, int endCol) {
|
||||
CellRangeAddress ref = new CellRangeAddress(startRow, endRow, startCol, endCol);
|
||||
return XDDFDataSourcesFactory.fromStringCellRange(sheet, ref);
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建值数据引用
|
||||
* @return XDDFNumericalDataSource<Double>
|
||||
*/
|
||||
private static XDDFNumericalDataSource<Double> createNumericDataReference(XSSFSheet sheet, int startRow, int endRow, int startCol, int endCol) {
|
||||
CellRangeAddress ref = new CellRangeAddress(startRow, endRow, startCol, endCol);
|
||||
return XDDFDataSourcesFactory.fromNumericCellRange(sheet, ref);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user