记录一次使用chatgpt解决实际问题的,需求是在页面添加一个订单导出excel的功能,订单编号、订单明细,相同订单编号合并单元格,模板如下
表头表尾不用说,
主要是表格内容部分,左边是订单编号,右边是订单明细,**这边难就难在一个订单编号可能存在多条明细,所以订单编号相同时需要合并单元格,**我们将需求整理清楚,下面就可以去询问chatgpt了,
首先我们试探性的问他:
他给出了回答,可以看出,他是完全知道怎么写的,接着我们就可以一步步继续深入问他了,下面就要带入自己的需求:
可以看到,我们这时将表单数据直接带进去询问,他依旧很清楚的给出了答案接下来我们继续深入询问他,创建的表单是否为父子结构,订单编号为父数据,其余为子列数据:
这时他依旧很清晰的给出了回答,并且他知道应该将父数据订单编号合并单元格。
真的很神奇,下面我们继续深入:我发现,他所有的数据都是根据我给的表单结构,自己造的,所以就询问他,如果数据是从数据库中取出来的,应该怎么写
他很快给出了回答,还是很正确的,但是这只是一个void main 测试方法,并不是我们想要的controller、service方法,于是我继续询问:如果是由页面发起请求,进行导出应该怎么写呢
这时他居然直接给出了正确的controller方法,现在已经离我们想要的代码非常接近了。
我们继续询问:让他加上表头、表尾、金额统计等等
接下来我们基本就可以进行手动测试了,测试过程中会出现各种问题,没关系。我们把发现的问题直接告诉他,让他修改就行,比如说我发现他忘记给内容添加表头了,直接就问他:
他很快会给出修改后的代码,当然,如果你代码测试中出现报错,也可以直接把代码和错误信息发过去问他:
他很快会解决,并进行说明,总之你出现问题都可以直接询问,他都是能看得懂的,最终代码如下:
@Override
public void exportBill(HttpServletResponse response, List<Integer> idList) throws Exception{
// 获取数据
List<BillProductOderVO> orders = billProductService.getAllOrders(idList);
// 创建工作簿
Workbook workbook = new XSSFWorkbook();
// 创建工作表
Sheet sheet = workbook.createSheet("Sheet1");
// 创建表头单元格样式
CellStyle headerStyle = workbook.createCellStyle();
headerStyle.setAlignment(HorizontalAlignment.CENTER);
headerStyle.setVerticalAlignment(VerticalAlignment.CENTER);
Font headerFont = workbook.createFont();
headerFont.setBold(true);
headerStyle.setFont(headerFont);
// 创建正式内容单元格样式
CellStyle contentStyle = workbook.createCellStyle();
contentStyle.setAlignment(HorizontalAlignment.CENTER);
contentStyle.setVerticalAlignment(VerticalAlignment.CENTER);
contentStyle.setBorderTop(BorderStyle.THIN);
contentStyle.setBorderBottom(BorderStyle.THIN);
contentStyle.setBorderLeft(BorderStyle.THIN);
contentStyle.setBorderRight(BorderStyle.THIN);
// 添加大表头
Row bigHeaderRow = sheet.createRow(0);
bigHeaderRow.setHeightInPoints(30);
Cell bigHeaderCell = bigHeaderRow.createCell(0);
bigHeaderCell.setCellValue("销售出库单");
bigHeaderCell.setCellStyle(headerStyle);
sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, 7));
// 添加第一行表头
Row firstHeaderRow = sheet.createRow(1);
firstHeaderRow.setHeightInPoints(20);
List<String> firstHeader = Arrays.asList("日期:", DateUtil.getNowDayForChinese(), "单据编号:", "", "客户:","", "制单人:",JWTUtil.getCurrentUser().getTokenRealName());
for (int i = 0; i < firstHeader.size(); i++) {
Cell cell = firstHeaderRow.createCell(i);
cell.setCellValue(firstHeader.get(i));
cell.setCellStyle(headerStyle);
}
// 添加第二行表头
Row secondHeaderRow = sheet.createRow(2);
secondHeaderRow.setHeightInPoints(20);
List<String> secondHeader = Arrays.asList("联系人:","", "联系电话:","", "部门:","", "仓库:","");
for (int i = 0; i < secondHeader.size(); i++) {
Cell cell = secondHeaderRow.createCell(i);
cell.setCellValue(secondHeader.get(i));
cell.setCellStyle(headerStyle);
}
// 添加正式数据表头
Row headerRow = sheet.createRow(3);
headerRow.setHeightInPoints(20);
List<String> headers = Arrays.asList("订单编号","存货名称", "规格型号", "计量单位", "数量", "单价", "金额","");
for (int i = 0; i < headers.size(); i++) {
Cell cell = headerRow.createCell(i);
cell.setCellValue(headers.get(i));
cell.setCellStyle(headerStyle);
}
// 创建数据行,并设置内容
int rowIndex = 4;
BigDecimal totalAmount = BigDecimal.ZERO;
String lastParentId = "";
Map<String, Integer> startRows = new HashMap<>();
Map<String, Integer> endRows = new HashMap<>();
for (BillProductOderVO order : orders) {
String parentId = order.getOrderNo();
if (parentId.equals(lastParentId)) {
// 当前行与上一行的订单编号相同,不需要合并单元格,更新结束行
Row row = sheet.createRow(rowIndex++);
for (int i = 0; i < headers.size(); i++) {
Cell subCell = row.createCell(i);
subCell.setCellValue(getCellValue(order, i));
subCell.setCellStyle(contentStyle);
}
endRows.put(parentId, rowIndex - 1);
totalAmount = totalAmount.add(order.getAllMoney());
} else {
// 当前行与上一行的订单编号不同,更新起始行和结束行
if (startRows.containsKey(lastParentId)) {
int startRow = startRows.get(lastParentId);
int endRow = endRows.get(lastParentId);
if (endRow - startRow > 0) {
// 只有当有两行及以上相同的订单编号时才需要合并单元格
CellRangeAddress mergedRegion = new CellRangeAddress(startRow, endRow, 0, 0);
sheet.addMergedRegion(mergedRegion);
}
}
startRows.put(parentId, rowIndex);
endRows.put(parentId, rowIndex);
Row row = sheet.createRow(rowIndex++);
for (int i = 0; i < headers.size(); i++) {
Cell subCell = row.createCell(i);
subCell.setCellValue(getCellValue(order, i));
subCell.setCellStyle(contentStyle);
}
totalAmount = totalAmount.add(order.getAllMoney());
}
lastParentId = parentId;
}
// 对于最后一个订单号,需要进行合并单元格操作
if (startRows.containsKey(lastParentId)) {
int startRow = startRows.get(lastParentId);
int endRow = endRows.get(lastParentId);
if (endRow - startRow > 0) {
// 只有当有两行及以上相同的订单编号时才需要合并单元格
CellRangeAddress mergedRegion = new CellRangeAddress(startRow, endRow, 0, 0);
sheet.addMergedRegion(mergedRegion);
}
}
// 创建总金额行
Row totalRow = sheet.createRow(rowIndex++);
List<String> totalData = Arrays.asList("合计", "", "", "", "", "", String.valueOf(totalAmount),"");
for (int i = 0; i < totalData.size(); i++) {
Cell cell = totalRow.createCell(i);
cell.setCellValue(totalData.get(i));
cell.setCellStyle(contentStyle);
}
// 添加备注行
Row remarkRow = sheet.createRow(rowIndex++);
remarkRow.setHeightInPoints(20);
Cell remarkCell = remarkRow.createCell(0);
remarkCell.setCellValue("备注:");
sheet.addMergedRegion(new CellRangeAddress(rowIndex - 1, rowIndex - 1, 0, headers.size() - 1));
// 添加友情提示行
Row promptRow = sheet.createRow(rowIndex++);
promptRow.setHeightInPoints(20);
Cell promptCell = promptRow.createCell(0);
promptCell.setCellValue("友情提示:1、商品请当场验收,如有问题可即时更换或拒收2、易变质食品请妥善保管。");
sheet.addMergedRegion(new CellRangeAddress(rowIndex - 1, rowIndex - 1, 0, headers.size() - 1));
// 添加审核人、出库人、送货人、签收人行
Row personRow = sheet.createRow(rowIndex++);
personRow.setHeightInPoints(20);
List<String> personData = Arrays.asList("审核人:", "出库人:", "送货人:", "签收人:");
for (int i = 0; i < personData.size(); i++) {
Cell cell = personRow.createCell(i * 2);
cell.setCellValue(personData.get(i));
sheet.addMergedRegion(new CellRangeAddress(rowIndex - 1, rowIndex - 1, i * 2, i * 2 + 1));
}
// 设置列宽
for (int i = 0; i < headers.size(); i++) {
if(i==1){
sheet.setColumnWidth(i,60 * 256);
}else {
sheet.setColumnWidth(i, 20 * 256);
}
}
// 设置响应头,告诉浏览器该文件为Excel格式
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
response.setHeader("Content-Disposition", "attachment; filename=\"orders.xlsx\"");
// 将工作簿写入输出流,导出Excel文件
OutputStream outputStream = response.getOutputStream();
workbook.write(outputStream);
workbook.close();
outputStream.close();
}
private String getCellValue(BillProductOderVO order, int columnIndex) {
switch (columnIndex) {
case 0:
return order.getOrderNo();
case 1:
return order.getProductName();
case 2:
return order.getSpec();
case 3:
return order.getUnit();
case 4:
return String.valueOf(order.getProductQuantity());
case 5:
return String.valueOf(order.getProductPrice());
case 6:
return String.valueOf(order.getAllMoney());
default:
return "";
}
}
不得不说,ChatGPT还是很强大的,这样的小功能如果你之前做过类似的,相信很快就能写出来,但是如果没有,一般人还是要花点时间去琢磨的。
ChatGPT的出现,真的很nice,不过所谓的程序员成功将自己卷没了这句话也并不正确,因为如果你根本是小白,连基本的代码都看不懂,哪怕给你ChatGPT也没法写功能。ChatGPT的关键就在于如何提问,对于一些小功能,使用它还是非常节省时间的。