人员工资附件excel读取
This commit is contained in:
zhangzhiguo
parent
5e97ee80f7
commit
01dbec485c
|
|
@ -0,0 +1,388 @@
|
|||
package zcgj.zcdev.zcdev.fs.plugin.form;
|
||||
|
||||
import com.alibaba.fastjson.JSONArray;
|
||||
import com.alibaba.fastjson.JSONObject;
|
||||
import kd.bos.bill.AbstractBillPlugIn;
|
||||
import kd.bos.bill.BillShowParameter;
|
||||
import kd.bos.bill.OperationStatus;
|
||||
import kd.bos.cache.CacheFactory;
|
||||
import kd.bos.cache.tempfile.TempFileCacheDownloadable;
|
||||
import kd.bos.dataentity.OperateOption;
|
||||
import kd.bos.dataentity.entity.DynamicObject;
|
||||
import kd.bos.dataentity.entity.DynamicObjectCollection;
|
||||
import kd.bos.dataentity.utils.StringUtils;
|
||||
import kd.bos.db.DB;
|
||||
import kd.bos.db.DBRoute;
|
||||
import kd.bos.db.ResultSetHandler;
|
||||
import kd.bos.entity.datamodel.events.ChangeData;
|
||||
import kd.bos.entity.datamodel.events.PropertyChangedArgs;
|
||||
import kd.bos.entity.operate.result.OperationResult;
|
||||
import kd.bos.fileservice.FileService;
|
||||
import kd.bos.fileservice.FileServiceFactory;
|
||||
import kd.bos.form.control.AttachmentPanel;
|
||||
import kd.bos.form.control.Control;
|
||||
import kd.bos.form.control.EntryGrid;
|
||||
import kd.bos.form.control.Toolbar;
|
||||
import kd.bos.form.control.events.ClickListener;
|
||||
import kd.bos.form.control.events.UploadEvent;
|
||||
import kd.bos.form.control.events.UploadListener;
|
||||
import kd.bos.form.events.AfterDoOperationEventArgs;
|
||||
import kd.bos.form.events.ClientCallBackEvent;
|
||||
import kd.bos.id.ID;
|
||||
import kd.bos.logging.Log;
|
||||
import kd.bos.logging.LogFactory;
|
||||
import kd.bos.servicehelper.AttachmentServiceHelper;
|
||||
import kd.bos.servicehelper.operation.OperationServiceHelper;
|
||||
import kd.bos.servicehelper.operation.SaveServiceHelper;
|
||||
import kd.sdk.plugin.Plugin;
|
||||
import org.apache.poi.ss.usermodel.*;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.math.BigDecimal;
|
||||
import java.net.URL;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.security.MessageDigest;
|
||||
import java.util.*;
|
||||
|
||||
/**
|
||||
* 人员工资附件excel读取，导入插件
|
||||
*/
|
||||
public class SalaryPlugin extends AbstractBillPlugIn implements Plugin, UploadListener {
|
||||
private static final Log log = LogFactory.getLog(SalaryPlugin.class);
|
||||
|
||||
private static final String[] COLUMNKEYA = new String[]{"zcgj_name","zcgj_number","zcgj_personno","zcgj_persontype","zcgj_cbstype" ,"zcgj_yingfa" ,"zcgj_shebao" ,"zcgj_gongjijin" ,"zcgj_nianjin" ,"zcgj_geshui" ,"zcgj_shifa","zcgj_remark"};
|
||||
|
||||
private static final Set<String> storSet = new HashSet<String>();
|
||||
static {
|
||||
storSet.add("zcgj_yingfa");
|
||||
storSet.add("zcgj_shebao");
|
||||
storSet.add("zcgj_gongjijin");
|
||||
storSet.add("zcgj_nianjin");
|
||||
storSet.add("zcgj_geshui");
|
||||
storSet.add("zcgj_shifa");
|
||||
}
|
||||
|
||||
public void registerListener(EventObject e) {
|
||||
super.registerListener(e);
|
||||
AttachmentPanel toolbar = this.getControl("attachmentpanel");
|
||||
toolbar.addUploadListener(this);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void afterUpload(UploadEvent evt) {
|
||||
UploadListener.super.afterUpload(evt);
|
||||
OperateOption option= OperateOption.create();
|
||||
Object[] urls = evt.getUrls();
|
||||
String url = (String)((Map)urls[0]).get("url");
|
||||
OperationResult result = OperationServiceHelper.executeOperate("save", "zcgj_salary", new DynamicObject[]{ this.getModel().getDataEntity(true)}, option);
|
||||
loadTextFileString(url);
|
||||
sumData();
|
||||
this.getView().addClientCallBack("auto_save",0);//附件上传后自动保存
|
||||
}
|
||||
|
||||
@Override
|
||||
public void clientCallBack(ClientCallBackEvent e) {
|
||||
super.clientCallBack(e);
|
||||
if(e.getName().equals("auto_save")) {
|
||||
AttachmentPanel attachmentPanel = getControl("attachmentpanel");
|
||||
List<Map<String, Object>> data = attachmentPanel.getAttachmentData();
|
||||
if(data !=null && !data.isEmpty()) {
|
||||
this.getView().invokeOperation("save");
|
||||
this.getView().invokeOperation("refresh");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void afterRemove(UploadEvent evt) {
|
||||
UploadListener.super.afterRemove(evt);
|
||||
this.getView().invokeOperation("save");
|
||||
this.getExcelData();
|
||||
sumData();
|
||||
this.getView().invokeOperation("refresh");
|
||||
}
|
||||
|
||||
/**
|
||||
* 从临时文件缓存中获取文件，转为字符流导入
|
||||
*/
|
||||
private void loadTextFileString(String fileUrl){
|
||||
|
||||
long billId = this.getModel().getDataEntity().getLong("id");
|
||||
|
||||
TempFileCacheDownloadable downLoad = (TempFileCacheDownloadable) CacheFactory.getCommonCacheFactory().getTempFileCache();
|
||||
|
||||
InputStream inStream = null;
|
||||
try {
|
||||
String[] queryParams = new URL(fileUrl).getQuery().split("&");
|
||||
Map<String, String> downloadFileParams = new HashMap<>();
|
||||
for (String queryParam : queryParams) {
|
||||
String[] p = queryParam.split("=");
|
||||
downloadFileParams.put(p[0], p[1]);
|
||||
}
|
||||
TempFileCacheDownloadable.Content content = downLoad.get(downloadFileParams.get("configKey"), downloadFileParams.get("id"));
|
||||
inStream = content.getInputStream();
|
||||
|
||||
//自动识别 XLS / XLSX，无需自己判断格式
|
||||
Workbook workbook = WorkbookFactory.create(inStream);
|
||||
//getSheetJsonArray，请保持不变
|
||||
JSONArray sheetJsonArray = getSheetJsonArray(
|
||||
stringArrayToMd5(COLUMNKEYA),
|
||||
workbook,
|
||||
0
|
||||
);
|
||||
List<Object[]> params = new ArrayList<>();
|
||||
for (Object data : sheetJsonArray) {
|
||||
System.out.println();
|
||||
List<Object> insertData = new ArrayList<>();
|
||||
long id = ID.genLongId();
|
||||
insertData.add(billId);//fid
|
||||
insertData.add(id);//fentryid
|
||||
for (int i=0;i<12;i++){
|
||||
String columnKey = ((JSONObject) ((JSONArray) data).get(i)).getString("columnKey");
|
||||
if (storSet.contains(columnKey)) {
|
||||
insertData.add( ((JSONObject) ((JSONArray) data).get(i)).getBigDecimal("value"));
|
||||
}else{
|
||||
insertData.add( ((JSONObject) ((JSONArray) data).get(i)).getString("value"));
|
||||
}
|
||||
}
|
||||
params.add(insertData.toArray());
|
||||
}
|
||||
DBRoute dbr = DBRoute.of("fi");
|
||||
if(!params.isEmpty()){
|
||||
String sql = "insert into tk_zcgj_salaryentry " +
|
||||
"(fid,fentryid," +
|
||||
"fk_zcgj_name,fk_zcgj_number,fk_zcgj_personno,fk_zcgj_persontype,fk_zcgj_cbstype,fk_zcgj_yingfa," +
|
||||
"fk_zcgj_shebao,fk_zcgj_gongjijin,fk_zcgj_nianjin,fk_zcgj_geshui,fk_zcgj_shifa,fk_zcgj_remark) " +
|
||||
"values(?,?,?,?,?,?,?,?,?,?,?,?,?,?)";
|
||||
DB.executeBatch(dbr,sql, params);
|
||||
}
|
||||
}catch(Exception e){
|
||||
log.info("SalaryPlugin:"+e.getMessage());
|
||||
}finally{
|
||||
try {
|
||||
if (inStream != null){
|
||||
inStream.close();
|
||||
}
|
||||
} catch (IOException e) {
|
||||
log.info("SalaryPlugin:"+e.getMessage());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 从附件面板中获取附件导入
|
||||
*/
|
||||
public void getExcelData(){
|
||||
DBRoute dbr = DBRoute.of("fi");
|
||||
|
||||
BillShowParameter bsp=(BillShowParameter)this.getView().getFormShowParameter();
|
||||
long billId = this.getModel().getDataEntity().getLong("id");
|
||||
if(billId != 0){
|
||||
List<Object[]> paramsdel = new ArrayList<>();
|
||||
paramsdel.add(new Object[]{billId});
|
||||
String delsql = "delete from tk_zcgj_salaryentry where fid = ?";
|
||||
int[] ints = DB.executeBatch(dbr, delsql, paramsdel);
|
||||
|
||||
String[] COLUMNKEYA = new String[]{"zcgj_name","zcgj_number","zcgj_personno","zcgj_persontype","zcgj_cbstype" ,"zcgj_yingfa" ,"zcgj_shebao" ,"zcgj_gongjijin" ,"zcgj_nianjin" ,"zcgj_geshui" ,"zcgj_shifa","zcgj_remark"};
|
||||
List<Map<String, Object>> attachments = AttachmentServiceHelper.getAttachments("zcgj_salary",billId, "attachmentpanel");
|
||||
for (Map<String, Object> attachment : attachments) {
|
||||
String url = (String)attachment.get("relativeUrl");
|
||||
// JSONArray objects = POIHelperFast.newInputExcel(url, POIHelperFast.stringArrayToHash(COLUMNKEYA));
|
||||
FileService service = FileServiceFactory.getAttachmentFileService();
|
||||
InputStream in = service.getInputStream(url);
|
||||
try {
|
||||
//自动识别 XLS / XLSX，无需自己判断格式
|
||||
Workbook workbook = WorkbookFactory.create(in);
|
||||
//getSheetJsonArray，请保持不变
|
||||
JSONArray sheetJsonArray = getSheetJsonArray(
|
||||
stringArrayToMd5(COLUMNKEYA),
|
||||
workbook,
|
||||
0
|
||||
);
|
||||
List<Object[]> params = new ArrayList<>();
|
||||
for (Object data : sheetJsonArray) {
|
||||
System.out.println();
|
||||
List<Object> insertData = new ArrayList<>();
|
||||
long id = ID.genLongId();
|
||||
insertData.add(billId);//fid
|
||||
insertData.add(id);//fentryid
|
||||
for (int i=0;i<12;i++){
|
||||
String columnKey = ((JSONObject) ((JSONArray) data).get(i)).getString("columnKey");
|
||||
if (storSet.contains(columnKey)) {
|
||||
insertData.add( ((JSONObject) ((JSONArray) data).get(i)).getBigDecimal("value"));
|
||||
}else{
|
||||
insertData.add( ((JSONObject) ((JSONArray) data).get(i)).getString("value"));
|
||||
}
|
||||
}
|
||||
params.add(insertData.toArray());
|
||||
}
|
||||
|
||||
if(!params.isEmpty()){
|
||||
String sql = "insert into tk_zcgj_salaryentry " +
|
||||
"(fid,fentryid," +
|
||||
"fk_zcgj_name,fk_zcgj_number,fk_zcgj_personno,fk_zcgj_persontype,fk_zcgj_cbstype,fk_zcgj_yingfa," +
|
||||
"fk_zcgj_shebao,fk_zcgj_gongjijin,fk_zcgj_nianjin,fk_zcgj_geshui,fk_zcgj_shifa,fk_zcgj_remark) " +
|
||||
"values(?,?,?,?,?,?,?,?,?,?,?,?,?,?)";
|
||||
DB.executeBatch(dbr,sql, params);
|
||||
}
|
||||
} catch (IOException ee) {
|
||||
ee.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public void sumData(){
|
||||
DBRoute dbr = DBRoute.of("fi");
|
||||
long billId = this.getModel().getDataEntity().getLong("id");
|
||||
if(billId != 0){
|
||||
String sql = "select fk_zcgj_persontype persontype,fk_zcgj_cbstype cbstype,sum(fk_zcgj_yingfa) yingfa,"+
|
||||
" sum(fk_zcgj_shebao) shebao,sum(fk_zcgj_gongjijin) gongjijin,sum(fk_zcgj_nianjin) nianjin," +
|
||||
"sum(fk_zcgj_geshui) geshui,sum(fk_zcgj_shifa) shifa" +
|
||||
" from tk_zcgj_salaryentry"+
|
||||
" where fid = "+billId+
|
||||
" group by fk_zcgj_persontype,fk_zcgj_cbstype";
|
||||
List<Map<String, Object>> resultData = DB.query(dbr, sql,resultSet -> {
|
||||
List<Map<String, Object>> rows = new ArrayList<>();
|
||||
while (resultSet.next()) {
|
||||
Map<String, Object> row = new HashMap<>();
|
||||
row.put("persontype", resultSet.getString("persontype"));
|
||||
row.put("cbstype", resultSet.getString("cbstype"));
|
||||
row.put("yingfa", resultSet.getBigDecimal("yingfa"));
|
||||
row.put("shebao", resultSet.getBigDecimal("shebao"));
|
||||
row.put("gongjijin", resultSet.getBigDecimal("gongjijin"));
|
||||
row.put("nianjin", resultSet.getBigDecimal("nianjin"));
|
||||
row.put("geshui", resultSet.getBigDecimal("geshui"));
|
||||
row.put("shifa", resultSet.getBigDecimal("shifa"));
|
||||
rows.add(row);
|
||||
}
|
||||
|
||||
return rows;
|
||||
});
|
||||
|
||||
DynamicObjectCollection dynamicObjectCollection = this.getModel().getDataEntity(true).getDynamicObjectCollection("zcgj_entryentity");
|
||||
dynamicObjectCollection.clear();
|
||||
if(resultData != null){
|
||||
for (Map<String, Object> resultDatum : resultData) {
|
||||
DynamicObject dynamicObject = dynamicObjectCollection.addNew();
|
||||
dynamicObject.set("zcgj_persontype2", resultDatum.get("persontype"));
|
||||
dynamicObject.set("zcgj_cbstype2", resultDatum.get("cbstype"));
|
||||
dynamicObject.set("zcgj_yingfa2", resultDatum.get("yingfa"));
|
||||
dynamicObject.set("zcgj_shebao2", resultDatum.get("shebao"));
|
||||
dynamicObject.set("zcgj_gongjijin2", resultDatum.get("gongjijin"));
|
||||
dynamicObject.set("zcgj_nianjin2", resultDatum.get("nianjin"));
|
||||
dynamicObject.set("zcgj_geshui2", resultDatum.get("geshui"));
|
||||
dynamicObject.set("zcgj_shifa2", resultDatum.get("shifa"));
|
||||
}
|
||||
}
|
||||
this.getView().updateView("zcgj_entryentity");
|
||||
}
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
private static JSONArray getSheetJsonArray(String columnHashCode, Workbook wb, int index) {
|
||||
|
||||
JSONArray jsonArray = new JSONArray();
|
||||
Sheet sheet = wb.getSheetAt(index);
|
||||
int rowNum = sheet.getLastRowNum();
|
||||
|
||||
Row columnKeyRow = sheet.getRow(1);
|
||||
Row headerRow = sheet.getRow(2);
|
||||
Row columnKeyHashRow = sheet.getRow(1);
|
||||
|
||||
if (columnKeyHashRow != null && columnKeyHashRow.getCell(0) != null) {
|
||||
|
||||
List<String> excelData = new ArrayList<String>();
|
||||
for (int i = 0; i < 12; i++) {
|
||||
columnKeyHashRow.getCell(i).setCellType(CellType.STRING);
|
||||
excelData.add(columnKeyHashRow.getCell(i).getStringCellValue());
|
||||
}
|
||||
//columnKeyHashRow.getCell(0).setCellType(CellType.STRING);
|
||||
String key = stringArrayToMd5(excelData.toArray(new String[0]));
|
||||
//String columnKeyHashCellValue = columnKeyHashRow.getCell(0).getStringCellValue();
|
||||
if (!StringUtils.equals(key, columnHashCode)) {
|
||||
return jsonArray;
|
||||
}
|
||||
|
||||
int columnNum = columnKeyRow.getLastCellNum();
|
||||
|
||||
// 计算真正的列数
|
||||
for (int i = 0; i < columnNum; ++i) {
|
||||
Cell cell = headerRow.getCell(i);
|
||||
if (cell == null) break;
|
||||
cell.setCellType(CellType.STRING);
|
||||
String value = cell.getStringCellValue();
|
||||
if (org.apache.commons.lang3.StringUtils.isEmpty(value)) {
|
||||
columnNum = i;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// 解析数据行
|
||||
for (int j = 3; j <= rowNum; ++j) {
|
||||
JSONArray jsonRow = new JSONArray();
|
||||
boolean isBlankRow = true;
|
||||
|
||||
Row irow = sheet.getRow(j);
|
||||
if (irow == null) break;
|
||||
|
||||
for (int i = 0; i < columnNum; ++i) {
|
||||
|
||||
JSONObject jsonObject = new JSONObject();
|
||||
|
||||
// columnKey
|
||||
Cell keyCell = columnKeyRow.getCell(i);
|
||||
keyCell.setCellType(CellType.STRING);
|
||||
|
||||
// name
|
||||
Cell nameCell = headerRow.getCell(i);
|
||||
nameCell.setCellType(CellType.STRING);
|
||||
|
||||
jsonObject.put("columnKey", keyCell.getStringCellValue());
|
||||
jsonObject.put("name", nameCell.getStringCellValue());
|
||||
|
||||
String value = "";
|
||||
Cell cell = irow.getCell(i);
|
||||
if (cell != null) {
|
||||
cell.setCellType(CellType.STRING);
|
||||
value = cell.getStringCellValue();
|
||||
}
|
||||
|
||||
jsonObject.put("value", value);
|
||||
isBlankRow = isBlankRow && org.apache.commons.lang3.StringUtils.isEmpty(value);
|
||||
|
||||
jsonRow.add(jsonObject);
|
||||
}
|
||||
|
||||
if (isBlankRow) break;
|
||||
|
||||
jsonArray.add(jsonRow);
|
||||
}
|
||||
}
|
||||
|
||||
return jsonArray;
|
||||
}
|
||||
|
||||
public static String stringArrayToMd5(String[] array) {
|
||||
try {
|
||||
MessageDigest md = MessageDigest.getInstance("MD5");
|
||||
md.update(String.join(",", array).getBytes(StandardCharsets.UTF_8));
|
||||
byte[] digest = md.digest();
|
||||
|
||||
StringBuilder sb = new StringBuilder();
|
||||
for (byte b : digest) {
|
||||
sb.append(String.format("%02x", b));
|
||||
}
|
||||
return sb.toString();
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue