261 lines
11 KiB
Java
261 lines
11 KiB
Java
package shkd.repc.repmd.template;
|
|
|
|
import kd.bos.context.RequestContext;
|
|
import kd.bos.dataentity.entity.DynamicObject;
|
|
import kd.bos.dataentity.entity.DynamicObjectCollection;
|
|
import kd.bos.dataentity.utils.StringUtils;
|
|
import kd.bos.entity.ExtendedDataEntity;
|
|
import kd.bos.entity.plugin.AbstractOperationServicePlugIn;
|
|
import kd.bos.entity.plugin.AddValidatorsEventArgs;
|
|
import kd.bos.entity.plugin.args.AfterOperationArgs;
|
|
import kd.bos.entity.validate.AbstractValidator;
|
|
import kd.bos.form.*;
|
|
import kd.bos.form.control.Control;
|
|
import kd.bos.form.control.EntryGrid;
|
|
import kd.bos.form.control.events.ItemClickEvent;
|
|
import kd.bos.form.events.MessageBoxClosedEvent;
|
|
import kd.bos.form.plugin.AbstractFormPlugin;
|
|
import kd.bos.orm.query.QCP;
|
|
import kd.bos.orm.query.QFilter;
|
|
import kd.bos.servicehelper.BusinessDataServiceHelper;
|
|
import kd.sdk.plugin.Plugin;
|
|
import org.apache.poi.xssf.usermodel.XSSFCell;
|
|
import org.apache.poi.xssf.usermodel.XSSFRow;
|
|
import org.apache.poi.xssf.usermodel.XSSFSheet;
|
|
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
|
|
import shkd.repc.repmd.template.util.ExportTemplateUtil;
|
|
import shkd.repc.repmd.template.util.Template;
|
|
|
|
import java.math.BigDecimal;
|
|
import java.util.*;
|
|
|
|
/**
|
|
* 面积数据导出-二开插件
|
|
*/
|
|
public class AreaDataExportPlugin extends AbstractFormPlugin implements Plugin {
|
|
|
|
|
|
@Override
|
|
public void registerListener(EventObject e) {
|
|
super.registerListener(e);
|
|
this.addItemClickListeners("qeug_advcontoolbarap");
|
|
}
|
|
|
|
@Override
|
|
public void itemClick(ItemClickEvent evt) {
|
|
super.itemClick(evt);
|
|
if (StringUtils.equals("qeug_areadataexport", evt.getItemKey())) {
|
|
DynamicObjectCollection productEntry = this.getModel().getEntryEntity("productentry");
|
|
if (null==productEntry||productEntry.size()==0){
|
|
this.getView().showMessage("暂无可导出数据！");
|
|
return;
|
|
}
|
|
EntryGrid entryGrid = this.getControl("productentry");
|
|
int[] selectRows = entryGrid.getSelectRows();
|
|
String top="导出确认提示";
|
|
String body;
|
|
//默认导出
|
|
if (selectRows.length==0){
|
|
body = "默认导出所有【产品构成】数据？";
|
|
this.showExportMessage(top,body,"default");
|
|
}else {
|
|
body = "导出 " + selectRows.length + " 行【产品构成】数据？";
|
|
this.showExportMessage(top,body,"select");
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void confirmCallBack(MessageBoxClosedEvent evt) {
|
|
super.confirmCallBack(evt);
|
|
String callBackId = evt.getCallBackId();
|
|
StringBuilder tipsBuffer = new StringBuilder();
|
|
|
|
if (StringUtils.equalsIgnoreCase("export_area", callBackId) && evt.getResult().getValue() == 6) {
|
|
String customValue = evt.getCustomVaule();
|
|
boolean isDefaultExport = "default".equals(customValue);
|
|
|
|
// 获取模板
|
|
DynamicObject templateObj = getTemplate();
|
|
if (templateObj == null) {
|
|
return;
|
|
}
|
|
|
|
// 读取系统模板
|
|
Control productView = this.getView().getControl("productentry");
|
|
Control areaView = this.getView().getControl("qeug_subentryentity");
|
|
List<Template> templateList = ExportTemplateUtil.readSysTemplate(templateObj, productView, areaView, tipsBuffer);
|
|
if (templateList.isEmpty()) {
|
|
return;
|
|
}
|
|
|
|
// 创建 Excel
|
|
XSSFWorkbook workbook = new XSSFWorkbook();
|
|
XSSFSheet sheet = createExcelSheet(workbook, templateList, areaView);
|
|
|
|
// 获取表头信息（第三行）
|
|
List<String> columnHeaders = getColumnHeaders(sheet);
|
|
|
|
// 获取数据
|
|
DynamicObjectCollection productEntry = this.getModel().getEntryEntity("productentry");
|
|
List<DynamicObject> exportData = isDefaultExport ? new ArrayList<>(productEntry) : getSelectedData(productEntry);
|
|
|
|
// 填充数据
|
|
fillExcelData(sheet, exportData, columnHeaders);
|
|
|
|
// 上传文件并提供下载链接
|
|
exportAndDownload(workbook, customValue);
|
|
|
|
tipsBuffer.append("导出成功");
|
|
this.getView().showSuccessNotification(tipsBuffer.toString());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取模板对象
|
|
*/
|
|
private DynamicObject getTemplate() {
|
|
QFilter numberFilter = new QFilter("number", QCP.equals, "repmd_projectbill_IMPT_ENTRY");
|
|
return BusinessDataServiceHelper.loadSingle("bos_importentry_template", numberFilter.toArray());
|
|
}
|
|
|
|
/**
|
|
* 创建 Excel Sheet 并填充标题、描述、表头等
|
|
*/
|
|
private XSSFSheet createExcelSheet(XSSFWorkbook workbook, List<Template> templateList, Control areaView) {
|
|
XSSFSheet sheet = workbook.createSheet("产品构成 # productentry");
|
|
ExportTemplateUtil.createTitleRow(sheet, workbook);
|
|
ExportTemplateUtil.createDescriptionRow(sheet, workbook);
|
|
ExportTemplateUtil.createTemplateHeaders(sheet, workbook, templateList);
|
|
ExportTemplateUtil.otherStyleSet(sheet, areaView);
|
|
return sheet;
|
|
}
|
|
|
|
/**
|
|
* 获取第三行的表头信息
|
|
*/
|
|
private List<String> getColumnHeaders(XSSFSheet sheet) {
|
|
List<String> columnHeaders = new ArrayList<>();
|
|
XSSFRow thirdRow = sheet.getRow(2);
|
|
if (thirdRow != null) {
|
|
for (int cellIndex = 0; cellIndex < thirdRow.getLastCellNum(); cellIndex++) {
|
|
XSSFCell cell = thirdRow.getCell(cellIndex);
|
|
columnHeaders.add(cell != null ? cell.getStringCellValue().trim() : ""); // 避免 null
|
|
}
|
|
}
|
|
return columnHeaders;
|
|
}
|
|
|
|
/**
|
|
* 获取选中的数据
|
|
*/
|
|
private List<DynamicObject> getSelectedData(DynamicObjectCollection productEntry) {
|
|
List<DynamicObject> selectedData = new ArrayList<>();
|
|
EntryGrid entryGrid = this.getControl("productentry");
|
|
int[] selectedRows = entryGrid.getSelectRows();
|
|
for (int row : selectedRows) {
|
|
selectedData.add(productEntry.get(row));
|
|
}
|
|
return selectedData;
|
|
}
|
|
|
|
/**
|
|
* 填充 Excel 数据
|
|
*/
|
|
private void fillExcelData(XSSFSheet sheet, List<DynamicObject> productEntries, List<String> columnHeaders) {
|
|
int rowIndex = 4; // 从 Excel 第 5 行开始填充
|
|
for (DynamicObject productObj : productEntries) {
|
|
if (productObj == null) continue;
|
|
|
|
String productTypeNumber = productObj.getString("productentry_producttype.number");
|
|
String productTypeName = productObj.getString("productentry_producttype.name");
|
|
DynamicObjectCollection areaEntries = productObj.getDynamicObjectCollection("qeug_subentryentity");
|
|
|
|
if (areaEntries != null && !areaEntries.isEmpty()) {
|
|
for (DynamicObject areaObj : areaEntries) {
|
|
XSSFRow dataRow = sheet.createRow(rowIndex);
|
|
int cellIndex = 0;
|
|
|
|
for (String header : columnHeaders) {
|
|
XSSFCell cell = dataRow.createCell(cellIndex);
|
|
|
|
if ("productentry_producttype.number".equals(header)) {
|
|
cell.setCellValue(productTypeNumber);
|
|
} else if ("productentry_producttype.name".equals(header)) {
|
|
cell.setCellValue(productTypeName);
|
|
} else {
|
|
Object value = areaObj.get(header);
|
|
setCellValue(cell, value);
|
|
}
|
|
cellIndex++;
|
|
}
|
|
rowIndex++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 设置 Excel 单元格值
|
|
*/
|
|
private void setCellValue(XSSFCell cell, Object value) {
|
|
if (value instanceof String) {
|
|
cell.setCellValue((String) value);
|
|
} else if (value instanceof BigDecimal) {
|
|
cell.setCellValue(((BigDecimal) value).doubleValue());
|
|
} else if (value instanceof Integer) {
|
|
cell.setCellValue((Integer) value);
|
|
} else if (value instanceof Double) {
|
|
cell.setCellValue((Double) value);
|
|
} else if (value != null) {
|
|
cell.setCellValue(value.toString());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 导出 Excel 并提供下载链接
|
|
*/
|
|
private void exportAndDownload(XSSFWorkbook workbook, String fileName) {
|
|
String uploadedFilePath = ExportTemplateUtil.uploadExcel(workbook, fileName);
|
|
String url = RequestContext.get().getClientFullContextPath() + "/attachment/download.do?path=" + uploadedFilePath;
|
|
this.getView().openUrl(url);
|
|
}
|
|
|
|
private void showExportMessage(String top,String body,String customValue){
|
|
// 基本信息
|
|
// 备注: \r\n 可换行
|
|
//String msg = "消息头\r\n message header...";
|
|
// 详细消息
|
|
//String detail = "消息体\r\nmessage body...";
|
|
// 消息框按钮类型
|
|
// None:不启用 Toast:简短提示,几秒后消失
|
|
MessageBoxOptions options = MessageBoxOptions.OKCancel;
|
|
// 确认提示类型
|
|
// Default:默认提示 Save:保存提交类提示 Delete:删除类提示 Wait:等待类提示 Fail:失败类提示
|
|
ConfirmTypes confirmTypes = ConfirmTypes.Default;
|
|
// 确认框回调
|
|
ConfirmCallBackListener callBack = new ConfirmCallBackListener("export_area", this);
|
|
// 按钮名称
|
|
// 注意: 按钮名称参数Map的key值必须和 options 参数相对应, 否则按钮名称修改不生效!
|
|
// 例: options 使用 MessageBoxOptions.OKCancel, 则 按钮名称参数Map的key必须为2 or 6
|
|
// options 使用 MessageBoxOptions.AbortRetryIgnore, 则 按钮名称参数Map的key必须为3 or 4 or 5
|
|
// none-0; ok-1; cancel-2; abort-3; retry-4; ignore-5; yes-6; no-7; custom-8
|
|
// -----------------------------------------------------------------
|
|
// options | btnNameMaps的key的取值
|
|
// MessageBoxOptions.None | 2
|
|
// MessageBoxOptions.OK | 2
|
|
// MessageBoxOptions.OKCancel | 2 & 6
|
|
// MessageBoxOptions.AbortRetryIgnore | 3 & 4 & 5
|
|
// MessageBoxOptions.YesNoCancel | 6 & 7 & 2
|
|
// MessageBoxOptions.YesNo | 6 & 7
|
|
// MessageBoxOptions.RetryCancel | 4 & 2
|
|
// MessageBoxOptions.Toast | 2
|
|
// -----------------------------------------------------------------
|
|
Map<Integer, String> btnNameMaps = new HashMap<>();
|
|
btnNameMaps.put(2, "取消");
|
|
btnNameMaps.put(6, "确认");
|
|
// 用户自定义参数,前端会在afterConfirm中返回
|
|
//String customValue = "allowExport";
|
|
this.getView().showConfirm(top, body, options, confirmTypes, callBack, btnNameMaps,customValue);
|
|
}
|
|
} |