java基于xml配置的通用excel单表数据导入组件(四、DAO主处理类)

package XXXXX.manage.importexcel;

import java.beans.IntrospectionException;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.SQLException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.logging.Logger; import org.apache.commons.betwixt.io.BeanReader;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.xml.sax.SAXException; import XXXXX.cache.UserSession;
import xxxxxxxx.manage.importerror.BN_ImportError;
import xxxxxxxx.manage.importerror.DAO_ImportError;
import xxxxxxxx.supp.BN_Row;
import xxxxxxxx.supp.DAORunner;
import xxxxxxxx.supp.TransactionHandler;
import xxxxxxxx.util.ObjectUtil;
import xxxxxxxx.util.PathUtil;
import xxxxxxxx.util.StringUtil; public class DAO_ImportExcel { private String tableImport; //导入的临时表名
private String tableTarget; //导入的目标表名
private File excelFile; //excel数据文件
private int rows; //总行数
private int columns; //总列数
private Map<String, BN_Column> columnMap = new HashMap();// 列序号、字段定义映射
private String[] columnName; // excel列名 private BN_ImportExcel xmlBean;
private Map<String, BN_Column> excelColumn; // 列名称、字段定义映射
private Map<String, BN_Column> systemColumn;// 系统内置字段定义 Map<String, Object> resultMap = new HashMap<String, Object>();// 定义返回结果map private static Logger logger = Logger.getLogger(DAO_ImportExcel.class
.getName()); public DAO_ImportExcel(String tableTarget, File excelFile) {
this.tableTarget = tableTarget;
this.tableImport = tableTarget + "_import"; //导入临时表,固定以"_import"结尾
this.excelFile = excelFile;
loadXmlBean();
} private void loadXmlBean() {
BeanReader beanReader = new BeanReader();
// 配置BeanReader实例
beanReader.getXMLIntrospector().getConfiguration()
.setAttributesForPrimitives(false);
beanReader.getBindingConfiguration().setMapIDs(false); // 不自动生成ID
// 注册要转换对象的类,并指定根节点名称
BufferedReader br = null;
try {
beanReader.registerBeanClass("importExcel", BN_ImportExcel.class);
br = new BufferedReader(new InputStreamReader(
new FileInputStream(new File(
PathUtil.getPath("/WEB-INF/excel/" + tableImport + ".xml"))), "GBK"));
// 将XML解析Java Object
xmlBean = (BN_ImportExcel) beanReader.parse(br);
excelColumn = xmlBean.getExcelColumn();
systemColumn = xmlBean.getSystemColumn();
} catch (UnsupportedEncodingException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IntrospectionException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (SAXException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} finally {
if (br != null){
try {
br.close();
} catch (IOException e) {
}
}
}
} public Map doImport() {
Workbook workBook = null;
try {
workBook = new XSSFWorkbook(new FileInputStream(excelFile));
} catch (Exception ex) {
try {
workBook = new HSSFWorkbook(new FileInputStream(excelFile));
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
if (workBook == null){
resultMap.put("result", false);
resultMap.put("message", "导入失败!无法加载该数据文件,请检查文件格式是否正确.");
return resultMap;
}
Sheet sheet = workBook.getSheetAt(0);
rows = sheet.getPhysicalNumberOfRows();
columns = sheet.getRow(0).getPhysicalNumberOfCells();
logger.info("rows=" + rows + " columns=" + columns);
if (rows <= 1 || columns == 0 ){
resultMap.put("result", true);
resultMap.put("error", false);
resultMap.put("message", "系统没有检测到需要导入的数据内容!");
return resultMap;
}
columnName = new String[columns]; //执行数据操作之前的初始化脚本initSQL
boolean initResult = runXMLSQL(xmlBean.getInitSQL());
if (!initResult){
resultMap.put("result", false);
resultMap.put("message", "系统初始化失败,请检查配置文件'"+tableImport+".xm'的initSQL是否正确!");
return resultMap;
} // 解析第一行表头列名
Iterator cellIterator = sheet.getRow(0).cellIterator();
StringBuffer memo = new StringBuffer();
while (cellIterator.hasNext()) {
Cell cell = (Cell) cellIterator.next();
// 读取当前单元格的值
String cellValue = cell.getStringCellValue();
logger.info("cell[0," + cell.getColumnIndex() + "]=" + cellValue);
columnName[cell.getColumnIndex()] = cellValue;
if (excelColumn.containsKey(cellValue)) {
columnMap.put(cell.getColumnIndex() + "",
(BN_Column) excelColumn.get(cellValue));
} else {
memo.append("列名[").append(cellValue).append("] ");
logger.info("cell[0," + cell.getColumnIndex() + "]列名["
+ cellValue + "]" + "未在xml的<excelColumn>节点内定义!无法解析导入");
}
}
if (memo.length() > 0) {// 有未在xml的<excelColumn>节点内定义的列
BN_ImportError errorBean = new BN_ImportError();
errorBean.setOp_id(StringUtil.getUUID());
errorBean.setOp_name(tableTarget);
errorBean.setRow_no("1");
errorBean.setMemo(memo.toString()+" 未在xml配置文件中定义,无法解析导入!");
errorBean.setCreate_id(UserSession.getOpId());
errorBean.setCreate_name(UserSession.getOpName());
errorBean.setCreate_time(BN_Row.getNowInput());
DAO_ImportError.doInsert(errorBean, null);
} // 解析数据行(从第2行开始)
for (int rowNum = 1; rowNum < rows; rowNum++) {
cellIterator = sheet.getRow(rowNum).cellIterator();
memo = new StringBuffer();
BN_Row rowBean = new BN_Row();
// 处理表格数据
while (cellIterator.hasNext()) {
Cell cell = (Cell) cellIterator.next();
String cellValue = getCellValue(cell);
logger.info("cell[" + rowNum + "," + cell.getColumnIndex() + "]=" + cellValue);
if (columnMap.containsKey(cell.getColumnIndex() + "")) {// 列名有在xml中有定义
BN_Column columnBean = columnMap.get(cell.getColumnIndex() + "");
Object[] parseResult = parseCell(columnBean, rowNum,
columnName[cell.getColumnIndex()], cellValue);
if (parseResult[0] != null) {
rowBean.setColumnValue(columnBean.getColumn(), parseResult[0]);
} else {
if (parseResult[1] != null) {
memo.append(parseResult[1]);
}
}
}
}
if (memo.length() > 0) {// 有不符合内容格式要求的字段
logger.info("第["+(rowNum + 1)+"行] "+memo.toString());
BN_ImportError errorBean = new BN_ImportError();
errorBean.setOp_id(StringUtil.getUUID());
errorBean.setOp_name(tableTarget);
errorBean.setRow_no((rowNum + 1) + "");
errorBean.setMemo(memo.toString());
errorBean.setCreate_id(UserSession.getOpId());
errorBean.setCreate_name(UserSession.getOpName());
errorBean.setCreate_time(BN_Row.getNowInput());
DAO_ImportError.doInsert(errorBean, null);
} else {
// 系统级字段
Iterator iterator = systemColumn.keySet().iterator();
while (iterator.hasNext()) {
String key = iterator.next().toString();
BN_Column sysColumnBean = systemColumn.get(key);
Object result = transfer(sysColumnBean.getDataType(), "",
sysColumnBean.getClassName(),
sysColumnBean.getMethod(), sysColumnBean.getParam());
if (result != null) {
rowBean.setColumnValue(sysColumnBean.getColumn(),
result);
}
}
int result = DAORunner.insertRow(tableImport, rowBean);
if (result == -1){//保存临时表失败,登记错误信息
BN_ImportError errorBean = new BN_ImportError();
errorBean.setOp_id(StringUtil.getUUID());
errorBean.setOp_name(tableTarget);
errorBean.setRow_no((rowNum + 1) + "");
errorBean.setMemo("数据保存失败");
errorBean.setCreate_id(UserSession.getOpId());
errorBean.setCreate_name(UserSession.getOpName());
errorBean.setCreate_time(BN_Row.getNowInput());
DAO_ImportError.doInsert(errorBean, null);
}
}
}
//计算自己导入成功的数据条数
int dataSuccNum = DAORunner.count("select count(*) from "+tableImport+" where create_id=?",
new Object[]{UserSession.getOpId()});
//计算自己文件解析失败的数据条数,排除首行列名解析问题
int dataFailNum = DAORunner.count("select count(*) from t_import_error where create_id=? and row_no != ?",
new Object[]{UserSession.getOpId(), 1});
int columnFailNum = DAORunner.count("select count(*) from t_import_error where create_id=? and row_no = ?",
new Object[]{UserSession.getOpId(), 1});
boolean result = DAORunner.runTransaction(new TransactionHandler() {
public void process(Connection connections) throws SQLException {
//执行数据转移前的脚本 beforeSaveSQL
if (!ObjectUtil.isNull(xmlBean.getBeforeSaveSQL())){
String[] sqlArr = xmlBean.getBeforeSaveSQL().split(";");
for (int i = 0 ; i < sqlArr.length ; i ++){
DAORunner.runUpdate(sqlArr[i].replaceAll("\\$\\{userId}", UserSession.getOpId()),
new Object[]{}, connections);
}
} //转移数据至tableTarget
DAORunner.runUpdate("insert into "+tableTarget+" select * from "+tableImport+" where create_id=?",
new Object[]{UserSession.getOpId()}, connections); //执行数据转移后的脚本 afterSaveSQL
if (!ObjectUtil.isNull(xmlBean.getBeforeSaveSQL())){
String[] sqlArr = xmlBean.getAfterSaveSQL().split(";");
for (int i = 0 ; i < sqlArr.length ; i ++){
DAORunner.runUpdate(sqlArr[i].replaceAll("\\$\\{userId}", UserSession.getOpId()),
new Object[]{}, connections);
}
}
}
});
if (result){
resultMap.put("result", true);
if ((dataFailNum + columnFailNum) > 0){//excel出现问题数据
resultMap.put("error", true);
}
StringBuffer message = new StringBuffer();
message.append("已成功导入").append(dataSuccNum).append("条数据!");
if (dataFailNum > 0){
message.append("有").append(dataFailNum).append("条数据因格式内容有问题无法导入!");
}
if (columnFailNum > 0){
message.append("个别列名未在xml配置文件中定义无法导入!");
}
resultMap.put("message", message.toString());
return resultMap;
}else{
resultMap.put("result", false);
resultMap.put("message", "数据导入失败,请重新尝试!");
}
return resultMap;
} private String getCellValue(Cell cell) {
int type = cell.getCellType();
switch (type) {
case Cell.CELL_TYPE_NUMERIC:// 数值、日期类型
double d = cell.getNumericCellValue();
if (HSSFDateUtil.isCellDateFormatted(cell)) {// 日期类型
Date date = HSSFDateUtil.getJavaDate(d);
return new SimpleDateFormat("yyyyMMdd").format(date);
} else {// 数值类型
return d + "";
}
case Cell.CELL_TYPE_BLANK:// 空白单元格
return null;
case Cell.CELL_TYPE_STRING:// 字符类型
return cell.getStringCellValue();
case Cell.CELL_TYPE_BOOLEAN:// 布尔类型
return cell.getBooleanCellValue() + "";
default:
logger.info("未处理类型["+type+"]");
break;
}
return null;
} private Object transfer(String dataType, String cellValue,
String className, String method, String param) {
try {
Class[] classArr = null;
Object[] valueArr = null;
if (ObjectUtil.isNull(param)) {
classArr = new Class[] {};
valueArr = new Object[] {};
} else {
String[] paramArr = param.split(",");
classArr = new Class[paramArr.length];
valueArr = new Object[paramArr.length];
for (int i = 0; i < paramArr.length; i++) {
if ("${self}".equals(paramArr[i])) {
if ("varchar".equals(dataType)
|| "date".equals(dataType)) {
classArr[i] = String.class;
valueArr[i] = cellValue;
} else if ("int".equals(param)) {
classArr[i] = int.class;
valueArr[i] = new Integer(cellValue);
} else if ("float".equals(param)) {
classArr[i] = float.class;
valueArr[i] = new Float(cellValue);
}
} else {
String typeStr = paramArr[i].split(":")[0];
String valueStr = paramArr[i].split(":")[1];
if ("varchar".equals(typeStr)) {
classArr[i] = String.class;
valueArr[i] = valueStr;
} else if ("int".equals(typeStr)) {
classArr[i] = int.class;
valueArr[i] = new Integer(valueStr);
} else if ("float".equals(typeStr)) {
classArr[i] = float.class;
valueArr[i] = new Float(valueStr);
}
}
}
}
Class tClass = Class.forName(className);
Object classInstance = tClass.newInstance();
Method classMethod = tClass.getMethod(method, classArr);
Object result = classMethod.invoke(classInstance, valueArr);
return result;
} catch (SecurityException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IllegalArgumentException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (ClassNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (InstantiationException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IllegalAccessException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (NoSuchMethodException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (InvocationTargetException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return null;
} private Object[] parseCell(BN_Column columnBean, int rowNum,
String columnName, String cellValue) {
Object[] objectArr = new Object[2];
String dataType = columnBean.getDataType();
int length = columnBean.getLength();
boolean required = columnBean.isRequired();
boolean transfer = columnBean.isTransfer(); if (ObjectUtil.isNull(cellValue)) {
if (required) {
objectArr[0] = null;
objectArr[1] = "\"" + columnName + "\"不能为空! ";
return objectArr;
} else {
objectArr[0] = null;
objectArr[1] = null;
return objectArr;
}
} if (transfer) {// 需要转换内容
String className = columnBean.getClassName();
String method = columnBean.getMethod();
String param = columnBean.getParam();
Object result = transfer(dataType, cellValue, className, method,
param);
logger.info("[" + rowNum + "," + columnName + "]转换结果=" + result);
objectArr[0] = result;
objectArr[1] = null;
} else {// 不需要转换内容
// 根据数据类型检查长度和内容是否符合字段要求
if ("varchar".equals(dataType) || "date".equals(dataType)) {
if (length != 0 && cellValue.length() > length) {
objectArr[0] = null;
objectArr[1] = "\"" + columnName + "\"内容超过最大长度" + length + "的限制! ";
} else {
objectArr[0] = cellValue;
objectArr[1] = null;
}
} else if ("int".equals(dataType)) {
if (!StringUtil.isInteger(cellValue)) {
objectArr[0] = null;
objectArr[1] = "\"" + columnName + "\"内容应为整数! ";
} else {
objectArr[0] = new Integer(cellValue);
objectArr[1] = null;
}
} else if ("float".equals(dataType)) {
if (!StringUtil.isDouble(cellValue)) {
objectArr[0] = null;
objectArr[1] = "\"" + columnName + "\"内容应为浮点数! ";
} else {
objectArr[0] = new Float(cellValue);
objectArr[1] = null;
}
}
} return objectArr;
} private boolean runXMLSQL(final String sqlStr){
if (ObjectUtil.isNull(sqlStr)){
return true;
}
return DAORunner.runTransaction(new TransactionHandler() {
public void process(Connection connections) throws SQLException {
String[] sqlArr = sqlStr.split(";");
for (int i = 0 ; i < sqlArr.length ; i ++){
DAORunner.runUpdate(sqlArr[i].replaceAll("\\$\\{userId}", UserSession.getOpId()),
new Object[]{}, connections);
}
}
});
} public static void main(String[] args) throws Exception {
}
}
上一篇:javascript 一些需要知道的东西


下一篇:appium+python做移动端自动化测试