package com.matrix.codeGeneration.ext;
|
|
import com.matrix.codeGeneration.convert.DefaultTypeHandle;
|
import com.matrix.codeGeneration.model.OutDataSource;
|
import com.matrix.codeGeneration.model.PropertyColumn;
|
import com.matrix.codeGeneration.model.TableClassModel;
|
import com.matrix.codeGeneration.plugin.ExcelImport;
|
import org.apache.commons.lang.StringUtils;
|
|
import java.io.File;
|
import java.io.IOException;
|
import java.util.*;
|
|
/**
|
* excel数据源
|
*
|
* @author jiangyouyao
|
*
|
*/
|
public class ExcelDataSource extends OutDataSource {
|
// excel文件地址
|
private String sourcePath;
|
private String author;
|
|
private String dbType;
|
|
public ExcelDataSource() {
|
}
|
|
@Override
|
public List<TableClassModel> convertToTableModel() {
|
|
Map<String, List<List<Object>>> analysisExcel = new HashMap<>();
|
// 源文件路径
|
File sourceDir = new File(sourcePath);
|
if (!sourceDir.isDirectory()) {
|
new Exception("目标目录不存在");
|
}
|
File[] files = sourceDir.listFiles();
|
if (files != null) {
|
for (File file : files) {
|
if (file.canRead()) {// 是否为一个可读文件
|
String fileName = file.getName();
|
// 检查扩展名称
|
String extension = fileName.lastIndexOf(".") == -1 ? ""
|
: fileName.substring(fileName.lastIndexOf(".") + 1);
|
if (extension.equals("xls") || extension.equals("xlsx")) {
|
System.out.println(fileName);
|
// 读取excel文件的内容
|
try {
|
analysisExcel.putAll(ExcelImport.read2007ExcelAllShell(file, null, 6));
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
} else {
|
new RuntimeException(file.getName() + "文件不可读");
|
}
|
}
|
}
|
List<TableClassModel> models = analysisExcel(analysisExcel);
|
this.setTableClassModels(models);
|
return this.getTableClassModels();
|
}
|
|
/**
|
* 解析excel文件中数据表信息
|
*
|
* @param analysisExcel
|
* @return
|
*/
|
private List<TableClassModel> analysisExcel(Map<String, List<List<Object>>> analysisExcel) {
|
List<TableClassModel> models = new ArrayList<>();
|
Set<String> keys = analysisExcel.keySet();
|
// 除了第一行，其他每一行代表一个字段，一个key是一个表
|
for (String tableName : keys) {
|
TableClassModel model = new TableClassModel();
|
List<List<Object>> data = analysisExcel.get(tableName);
|
// 设置数据表基本信息
|
// 构建表名
|
|
int nameIndex = tableName.indexOf("(");
|
model.setTableMemo(tableName.substring(nameIndex+1, tableName.length()-1));
|
model.setTableName(tableName.substring(0, nameIndex));
|
// 设置类名和变量名称
|
model.setClassName(getNameConvert().tableNameToClassName(model.getTableName()));
|
model.setClassVariableName(getNameConvert().classNameToVariableName(model.getClassName()));
|
|
// 该表需要展示到页面的字段个数
|
int showCount=0;
|
// 遍历设置字段信息
|
for (int i = 1; i < data.size(); i++) {
|
List<Object> row = data.get(i);
|
PropertyColumn p = new PropertyColumn();
|
// 列名
|
String columnName = (String) row.get(0);
|
p.setColumn(columnName);
|
// 属性名称
|
p.setProperty(getNameConvert().columnToProperty(columnName));
|
|
// 字段类型、长度
|
String type = (String) row.get(2);
|
p.setFullJdbcType(type);
|
|
// jdbc类型设置
|
int index = type.indexOf("(") > 0 ? type.indexOf("(") : type.length();
|
|
p.setJdbcType(type.substring(0, index).trim().replaceAll("2", ""));
|
System.out.println("jdbctype=" + p.getJdbcType());
|
// 基本类型
|
p.setClassType(DefaultTypeHandle.TypeMapping.get(dbType).get(p.getJdbcType()));
|
|
String isAllowNull = (String) row.get(3);
|
|
if (isAllowNull != null && isAllowNull.equals("N")) {
|
isAllowNull = " NOT NULL ";
|
} else {
|
isAllowNull = "";
|
}
|
|
p.setIsAllowNull(isAllowNull);
|
|
// 备注中 [表示界面显示字段] [表示必填字段*]
|
String memo = (String) row.get(1);
|
int begin = memo.indexOf("[");
|
int end = memo.indexOf("]");
|
|
if (begin > -1 && end - begin > 1) {
|
showCount++;
|
// 显示到页面上的字段
|
String subString1 = memo.substring(begin, end);
|
boolean isNecessary = subString1.indexOf("*") > 0;
|
if (isNecessary) {
|
p.setShowName(memo.substring(begin + 1, end - 1));
|
} else {
|
p.setShowName(memo.substring(begin + 1, end));
|
}
|
p.setIsNecessary(isNecessary);
|
p.setIsVisible(true);
|
// 截取[符号前面的字段作为数据库的备注
|
p.setMemo(memo.substring(0, begin));
|
} else {
|
p.setIsNecessary(false);
|
p.setIsVisible(false);
|
p.setMemo(memo);
|
}
|
// get，set方法名称
|
p.setMethodName(getNameConvert().propertyToMethod(p.getProperty()));
|
|
// 设置主键
|
if (p.getMemo() != null && p.getMemo().equals("主键")) {
|
p.setIsPrimaryKey(true);
|
model.setPrimaryKey(p);
|
} else {
|
p.setIsPrimaryKey(false);
|
}
|
|
// 索引
|
String indexName = (String) row.get(5);
|
if(null == indexName || "".equals(indexName)){
|
indexName = null;
|
}
|
p.setIndexName(indexName);
|
|
// 设置jsp编辑页面字段长度限制
|
String columnLengthStr = row.get(6) == null ? null : (String) row.get(6);
|
if (columnLengthStr!=null && columnLengthStr.trim().length()>0) {
|
// 从excel中读取的数字默认是double类型的，所以做如下处理去掉小数点
|
Double columnLength = Double.parseDouble(columnLengthStr);
|
p.setColumnLength(columnLength.intValue()+"");
|
}
|
|
model.getMapping().add(p);
|
}
|
// 需要加上该表有多少个显示到list列表的字段个数
|
model.setShowCount(showCount);
|
models.add(model);
|
}
|
System.out.println(analysisExcel);
|
return models;
|
}
|
|
public String getSourcePath() {
|
return sourcePath;
|
}
|
|
public void setDbType(String dbType) {
|
this.dbType = dbType;
|
}
|
|
public void setSourcePath(String sourcePath) {
|
this.sourcePath = sourcePath;
|
}
|
|
public void setAuthor(String author) {
|
this.author = author;
|
}
|
|
public String getAuthor() {
|
return author;
|
}
|
|
public String getDbType() {
|
return dbType;
|
}
|
|
}
|
