将excel实体表格生成为JavaBean对象

package org.example; import com.alibaba.excel.EasyExcel; import com.alibaba.excel.read.listener.PageReadListener; import freemarker.template.Configuration; import freemarker.template.Template; import java.io.*; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; public class ParseModelToJavaBean { private static final String CLASS_PATH = "src/main/java/org/example/po"; private static Map<String,String> toJavaType ; // 对应数据库类型 private static Map<String,String> toDbType; static{ //初始化类型转换 toJavaType = new HashMap<>(); toJavaType.put("STRING","String"); toJavaType.put("DATETIME","Date"); toDbType = new HashMap<>(); // 自行定义类型转换 toDbType.put("STRING","DB.STRING"); toDbType.put("DATETIME","DB.DATETIME"); } public static void main(String[] args) { Configuration configuration = new Configuration(); Writer out = null; try { configuration.setDirectoryForTemplateLoading(new File(new File("").getAbsolutePath()+File.separator+"file")); Map<String, Object> dataMap = new HashMap<String, Object>(); dataMap.put("classPath", "org.example.po"); dataMap.put("className", "User"); List<PropertyEntity> properties = parseExcel("test.xlsx"); transportPropertyType(properties); dataMap.put("properties", properties); Template template = configuration.getTemplate("test.ftl"); File docFile = new File(CLASS_PATH + "\\" + "User.java"); out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(docFile))); template.process(dataMap, out); } catch (Exception e) { e.printStackTrace(); } finally { try { if (null != out) { out.flush(); } } catch (Exception e2) { e2.printStackTrace(); } } } public static List<PropertyEntity> parseExcel(String entityFileName){ List<PropertyEntity> propertyEntities = new ArrayList<>(); String fileName = new File("").getAbsolutePath()+File.separator+"file"+File.separator+"entity"+File.separator+entityFileName; System.out.println(fileName); EasyExcel.read(fileName, PropertyEntity.class, new PageReadListener<PropertyEntity>(dataList -> { for (PropertyEntity entity : dataList) { propertyEntities.add(entity); } })).sheet().doRead(); return propertyEntities; } /** * 转换属性 * @param properties */ public static void transportPropertyType(List<PropertyEntity> properties){ properties.forEach(propertyEntity -> { propertyEntity.setPropertyJavaType(toJavaType.get(propertyEntity.getPropertyType())); propertyEntity.setPropertyDbType(toDbType.get(propertyEntity.getPropertyType())); String fieldName = propertyEntity.getOrgFieldName(); propertyEntity.setPropertyName(toHump(fieldName)); }); } public static String toHump(String fieldName){ // 驼峰命名 String[] keyArray = fieldName.toLowerCase().split("_"); if (keyArray.length > 1) { StringBuilder sb = new StringBuilder(); for (int i = 0; i < keyArray.length; i++) { String ks = keyArray[i]; if (!"".equals(ks)) { if (i == 0) { sb.append(ks); } else { int c = ks.charAt(0); if (c >= 97 && c <= 122) { int v = c - 32; sb.append((char) v); if (ks.length() > 1) { sb.append(ks.substring(1)); } } else { sb.append(ks); } } } } return sb.toString(); } return fieldName; } }