一个WMS的运输物流管理系统

package com.ken.wms.common.util; import org.apache.commons.lang3.math.NumberUtils; import org.apache.poi.hssf.usermodel.HSSFDateUtil; import org.apache.poi.ss.usermodel.*; import org.apache.poi.xssf.usermodel.XSSFFont; import org.apache.poi.xssf.usermodel.XSSFWorkbook; import org.springframework.core.io.ClassPathResource; import org.w3c.dom.Document; import org.w3c.dom.Element; import org.w3c.dom.Node; import org.w3c.dom.NodeList; import org.xml.sax.SAXException; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; import javax.xml.parsers.ParserConfigurationException; import java.io.File; import java.io.FileInputStream; import java.io.FileOutputStream; import java.io.IOException; import java.lang.reflect.Field; import java.lang.reflect.Method; import java.util.*; /** * Excel -JavaBean 转换器 * * @author Ken * @since 2017/3/27. */ public class EJConvertor { /** * 默认配置文件名 */ private static final String DEFAULT_CONFIG_FILE_NAME = "EJConvertorConfig.xml"; /** * Entity 节点名称 */ private static final String ENTITY_ELEMENT = "entity"; /** * Property 节点名称 */ private static final String PROPERTY_ELEMENT = "property"; /** * Field 节点信息 */ private static final String FIELD_ELEMENT = "field"; /** * Value 节点信息 */ private static final String VALUE_ELEMENT = "value"; /** * class 属性 */ private static final String CLASS_ATTRIBUTE = "class"; /** * sheetName 属性 */ private static final String SHEET_NAME_ATTRIBUTE = "sheetName"; /** * boldHeading 属性 */ private static final String BOLD_HEADING_ATTRIBUTE = "boldHeading"; /** * JavaBean的映射信息 */ private Map<String, MappingInfo> excelJavaBeanMap; public EJConvertor() { init(DEFAULT_CONFIG_FILE_NAME); } public EJConvertor(String filePath) { init(filePath); } /** * 初始化映射信息 * * @param fileLocation 配置文件路径 */ private void init(String fileLocation) { try { // 读取配置文件 File configFile = new ClassPathResource(fileLocation).getFile(); DocumentBuilder documentBuilder = DocumentBuilderFactory.newInstance().newDocumentBuilder(); Document doc = documentBuilder.parse(configFile); // 解析配置文件 this.excelJavaBeanMap = parseMappingInfo(doc); } catch (ParserConfigurationException | SAXException | IOException e) { e.printStackTrace(); } } /** * 解析 Document root 下配置的所有 excel-javaBean 映射信息 * * @param root Document 根节点 * @return 返回 excel-javaBean 映射信息 */ private Map<String, MappingInfo> parseMappingInfo(Document root) { // 获取 root 下的所有 entity 节点 NodeList entities = root.getElementsByTagName(ENTITY_ELEMENT); // 创建承载 MappingInfo 信息的Map Map<String, MappingInfo> mappingInfoMap = new HashMap<>(entities.getLength()); // 解析 entity 节点 for (int index = 0; index < entities.getLength(); index++) { // 创建 mappingInfo MappingInfo mappingInfo = new MappingInfo(); // 解析节点信息 Node entityNode = entities.item(index); if (entityNode.getNodeType() == Node.ELEMENT_NODE) { Element entityElement = (Element) entityNode; parseEntityElement(entityElement, mappingInfo); } // 保存节点信息 mappingInfoMap.put(mappingInfo.getClassName(), mappingInfo); } return mappingInfoMap; } /** * 解析 entity 节点信息 * * @param entityElement entity 节点 * @param mappingInfo 本 entity 节点包含的映射信息 */ private void parseEntityElement(Element entityElement, MappingInfo mappingInfo) { // 解析 entity 的 class 属性 String className = entityElement.getAttribute(CLASS_ATTRIBUTE); mappingInfo.setClassName(className); // 解析 entity 的 sheetName 属性 if (entityElement.hasAttribute(SHEET_NAME_ATTRIBUTE)) mappingInfo.setSheetName(entityElement.getAttribute(SHEET_NAME_ATTRIBUTE)); // 解析 entity 的 boldHeading 属性 if (entityElement.hasAttribute(BOLD_HEADING_ATTRIBUTE)) { String isBoldHeading = entityElement.getAttribute(BOLD_HEADING_ATTRIBUTE); mappingInfo.setBoldHeading(isBoldHeading.equals("true")); } // 读取并解析 property 节点 NodeList properties = entityElement.getElementsByTagName(PROPERTY_ELEMENT); for (int index = 0; index < properties.getLength(); index++) { Node propertyNode = properties.item(index); if (propertyNode.getNodeType() == Node.ELEMENT_NODE) { Element propertyElement = (Element) propertyNode; parsePropertyElement(propertyElement, mappingInfo); } } } /** * 解析 property 节点信息 * * @param propertyElement property 节点 * @param mappingInfo 承载映射信息 */ private void parsePropertyElement(Element propertyElement, MappingInfo mappingInfo) { NodeList infoNodes = propertyElement.getChildNodes(); String field = null; String value = null; for (int infoNode_index = 0; infoNode_index < infoNodes.getLength(); infoNode_index++) { Node infoNode = infoNodes.item(infoNode_index); if (infoNode.getNodeName().equals(FIELD_ELEMENT)) field = infoNode.getTextContent(); if (infoNode.getNodeName().equals(VALUE_ELEMENT)) value = infoNode.getTextContent(); } // 添加到映射信息中 if (field != null && value != null) { mappingInfo.addFieldValueMapping(field, value); mappingInfo.addValueFieldMapping(value, field); } } /** * 讀取 Excel 文件中的内容 Excel 文件中的每一行代表了一个对象实例，而行中各列的属性值对应为对象中的各个属性值 * 读取时，需要指定读取目标对象的类型以获得相关的映射信息，并且要求该对象已在配置文件中注册 * * @param javaBeanClass 目标对象的类型 * @param file 数据来源的 Excel 文件 * @return 包含若干个目标对象实例的 List */ public <T> List<T> excelReader(Class<T> javaBeanClass, File file) { // 参数检查 if (file == null || javaBeanClass == null) return null; // 初始化存放读取结果的 List List<T> javaBeans = new ArrayList<>(); // 获取类名和映射信息 String className = javaBeanClass.getName(); MappingInfo mappingInfo = excelJavaBeanMap.get(className); if (mappingInfo == null) return null; // 读取 Excel 文件 try (Workbook workbook = new XSSFWorkbook(new FileInputStream(file))) { Sheet dataSheet = workbook.getSheetAt(0); Row row; Cell cell; Iterator<Row> rowIterator = dataSheet.iterator(); Iterator<Cell> cellIterator; // 读取第一行表头信息 if (!rowIterator.hasNext()) return null; String fieldName; Field fieldInstance; Class<?> fieldClass; List<String> fieldNameList = new ArrayList<>();// 目标�