package com.qsk.utils;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Date;
public class GenEntityUtil {
private final String myProjectPath = "";//空为当前项目路径（也可以自己指定绝对路径）
//数据库连接
private static final String DBNAME = ""; //数据库多连接名小写（不生成数据库连接就不用管）
private static final String URL ="";
private static final String NAME = "";
private static final String PASS = "";
private static final String DRIVER ="com.microsoft.sqlserver.jdbc.SQLServerDriver";
private final String tablename = "";//表名
private final String entityName = ""; //.java文件名
private final String authorName = "神转子"; //作者名字
private final String baseJavaUrl = "/src/main/java/";
private final String baseResourceUrl = "/src/main/resources/";
private final String COMMPATH = "com.qsk."; //项目总包路径
private final String requestMappingName = "/lisreport/"; //controller层跳转路径前缀
private final String ymlUrl = "config"; //配置路径，空就表示在resources目录下，否则请填 config
private final String mapperOutPath = "mappers"; //指定mapper.xml生成所在地，这里就放到resource下了
private final String entityPath = COMMPATH + "entity"; //指定实体生成所在包的路径
private final String controllerPath = COMMPATH + "controller"; //指定表现层生成包所在的路径
private final String servicePath = COMMPATH + "service"; //service层地址
private final String serviceImplPath = servicePath + ".impl"; //serviceImpl层地址
private final String mapperPath = COMMPATH + "mapper"; //mapper层地址
private final String configPath = COMMPATH + "core.config"; //配置文件地址
private final String resultPath = COMMPATH + "common.result";
private final String resultName = initUp("DataGridResultInfo"); //公共返回类名（首字母大写）
private final String connConfigPath = ".conndb"; //数据库配置路径（在配置文件地址下）
private final String dataName = ""; //表名注释
private final String divisionSymbol = "_"; //根据某个字符串进行分割
private boolean f_yml = true; //是否生成连接数据库的 yml 文件（建议多数据库连接时使用）生成完之后请关闭，否则会一直生成
private boolean f_dataSource = true; //是否在java中生成连接配置
private boolean f_util = true; //是否需要导入包java.util.*
private boolean f_sql = false; //是否需要导入包java.sql.*
private boolean f_data = true; //是否需要使用 @Data注解
private boolean f_divisionStr = true; //是否替换字段分割符号
private boolean f_generateMapper = true; //是否生成 mapper
private boolean f_service = true; //是否生成 service 服务层
private boolean f_controller = true; //是否生成 Controller 层
private boolean f_layuiHtml = true; //是否生成 layui 的html
private final String htmlPath = ""; //默认是在 templates 下
private final String layuiUrl =
"/coststatic/frame/layui/"; //配置 layui 文件路径，没有请从官网下载
private final String jsUrl = "/coststatic/js/"; //js生成路径（默认static目录下）
private boolean f_sql_select = true; //是否需要查询（若不生成layui的html就不用管）
private String[] colnames; //列名数组
private String[] colTypes; //列名类型数组
private int[] colSizes; //列名大小数组
private String dbType;
private final File file = new File("");
private final String projectPath = file.getAbsolutePath();
/*
* g构造函数
*/
public GenEntityUtil(){
//创建连接
Connection con;
//查要生成实体类的表
String sql = "select * from " + tablename;
Statement pStemt = null;
try {
try {
//加载驱动
Class.forName(DRIVER);
} catch (ClassNotFoundException e1) {
e1.printStackTrace();
}
//链接数据库
con = DriverManager.getConnection(URL,NAME,PASS);
DatabaseMetaData metdata = con.getMetaData();
this.dbType = metdata.getDatabaseProductName();
pStemt = (Statement) con.createStatement();
ResultSet rs = pStemt.executeQuery(sql);
ResultSetMetaData rsmd = rs.getMetaData();
int size = rsmd.getColumnCount(); //统计列
colnames = new String[size];
colTypes = new String[size];
colSizes = new int[size];
for (int i = 0; i < size; i++) {
colnames[i] = rsmd.getColumnName(i + 1);
colTypes[i] = rsmd.getColumnTypeName(i + 1);
if(colTypes[i].equalsIgnoreCase("date") || colTypes[i].equalsIgnoreCase("timestamp")){
f_util = true;
}
if(colTypes[i].equalsIgnoreCase("blob") || colTypes[i].equalsIgnoreCase("char")){
f_sql = true;
}
colSizes[i] = rsmd.getColumnDisplaySize(i + 1);
}
System.out.println("自动生成中...");
init();
System.out.println("生成成功！");
} catch (SQLException e) {
e.printStackTrace();
} finally{
// try {
// con.close();
// } catch (SQLException e) {
// e.printStackTrace();
// }
}
}
private void init() {
if(f_yml) {
initYml();
}
if(f_dataSource) {
initDataSource();
}
InitEntity(colnames,colTypes,colSizes); //生成实体类
//是否生成 mapper
if(f_generateMapper) {
InitMapperService();
InitMapper(colnames,colTypes,colSizes); //生成 mapper.xml
}
if(f_service) {
InitService();
}
if(f_controller) {
InitController();
}
if(f_layuiHtml) {
String path = this.resultPath.replace(".", "/") + "/" + this.resultName;
if(!isFileExists(path)) {
initResult();
}
initLayuiHtml();
}
}
private void initLayuiHtml() {
// TODO Auto-generated method stub
String content = parseLayuiHtml();
String outputPath = this.baseResourceUrl + "/templates/" + this.htmlPath + "/" + initLower(this.entityName) + "/index.html";
commWrite("layuiHtml", content, outputPath);
content = parseLayuiJsHtml();
outputPath = this.baseResourceUrl + "/static/" + this.jsUrl + "/" + initLower(this.entityName) + "/index.js";
commWrite("layuiHtml", content, outputPath);
}
private String parseLayuiJsHtml() {
// TODO Auto-generated method stub
StringBuffer sb = new StringBuffer();
sb.append("var table;\r\n");
sb.append("var layer;\r\n");
sb.append("var tableId = \"" + initLower(this.entityName) + "Table\";\r\n");
sb.append("layui.use([ 'layer', 'table', 'jquery' ], function() {\r\n");
sb.append(" table = layui.table\r\n");
sb.append(" ,layer = layui.layer\r\n");
sb.append(" $ = layui.$\r\n");
sb.append(" var cols = [ [ // 表头\r\n");
String colname = "";
for (int i = 0; i < colnames.length; i++) {
if(i == 0) {
sb.append(" {\r\n");
}else {
sb.append("{\r\n");
}
colname = colnames[i].toLowerCase();
if(f_divisionStr) {
colname = replaceSymbol(colname);
}
sb.append(" field : '" + colname + "',\r\n");
sb.append(" title : '字段" + (i+1) + "',\r\n");
sb.append(" align : 'center',\r\n");
sb.append(" templet : function(data){\r\n");
sb.append(" var value = \"\";\r\n");
sb.append(" if(data." + colname + " != null ){\r\n");
sb.append(" value = data." + colname + "\r\n");
sb.append(" }\r\n");
if(sqlType2JavaType(colTypes[i]).equals("int")) {
sb.append(" return \"\" + value + \"
\";\r\n");
}else {
sb.append(" return \"\" + value + \"
\";\r\n");
}
sb.append(" }\r\n");
sb.append(" }, ");
}
sb.append("] ];\r\n\r\n");
sb.append(" myTable = table.render({\r\n");
sb.append(" elem : '#" + initLower(this.entityName) + "Table',\r\n");
sb.append(" id : \"" + initLower(this.entityName) + "Table\",\r\n");
sb.append(" url : '"+ this.requestMappingName +"selectList',\r\n");
sb.append(" page : true, // 开启分页\r\n");
sb.append(" parseData:function(res){\r\n");
sb.append(" return {\r\n");
sb.append(" \"code\" : 0,\r\n");
sb.append(" data : res.data,\r\n");
sb.append(" \"count\" : res.total,\r\n");
sb.append(" }\r\n");
sb.append(" },\r\n");
sb.append(" cols : cols\r\n");
sb.append(" });\r\n\r\n");
sb.append(" window.select = function(){\r\n");
sb.append(" var keyword = $(\"#keyword\").val();\r\n");
sb.append(" table.reload('" + initLower(this.entityName) + "Table', {\r\n");
sb.append(" where : {\r\n");
sb.append(" keyword : keyword\r\n");
sb.append(" },\r\n");
sb.append(" page : {\r\n");
sb.append(" curr : 1\r\n");
sb.append(" }\r\n");
sb.append(" });\r\n");
sb.append(" };\r\n\r\n");
sb.append("});\r\n");
return sb.toString();
}
private String parseLayuiHtml() {
// TODO Auto-generated method stub
StringBuffer sb = new StringBuffer();
sb.append("\r\n");
sb.append("\r\n");
sb.append("\r\n");
sb.append(" 科室基本资料 \r\n");
sb.append(" \r\n");
sb.append(" \r\n");
processAllResult(sb);
sb.append(" \r\n");
processAllSql(sb);
sb.append(" \r\n");
return sb.toString();
}
/**
* 生成公共返回类
* @return
*/
private String parseResult() {
StringBuffer sb = new StringBuffer();
//导入包名
sb.append("package " + this.resultPath + ";\r\n\r\n");
sb.append("import java.util.List;\r\n");
sb.append("import java.util.Map;\r\n\r\n");
sb.append("import lombok.Data;\r\n\r\n");
sb.append("@Data\r\n");
sb.append("public class " + this.resultName + " {\r\n\r\n");
sb.append(" private volatile static " + resultName + " " + initLower(resultName) + ";\r\n\r\n");
sb.append(" private " + resultName + "() {}\r\n\r\n");
sb.append(" public static " + this.resultName + " get" + this.resultName + "() {\r\n");
sb.append(" if(" + initLower(this.resultName) + " == null) {\r\n");
sb.append(" synchronized (" + resultName + ".class) {\r\n");
sb.append(" if(" + initLower(this.resultName) + " == null)\r\n");
sb.append(" " + initLower(this.resultName) + " = new " + this.resultName + "();\r\n");
sb.append(" }\r\n");
sb.append(" }\r\n");
sb.append(" return " + initLower(this.resultName) + ";\r\n");
sb.append(" }\r\n\r\n");
sb.append(" private int total;\r\n");
sb.append(" private List rows;\r\n");
sb.append(" private List data;\r\n");
sb.append(" private Object objData;\r\n");
sb.append(" private boolean success;\r\n");
sb.append(" private boolean error;\r\n");
sb.append(" private String msg;\r\n");
sb.append(" private Map dataMap;\r\n");
sb.append(" private String code;\r\n\r\n");
sb.append("}\r\n");
return sb.toString();
}
/**
* 生成所有的 sql 语句
* @param sb
*/
private void processAllSql(StringBuffer sb) {
// 生成查询语句
if(f_sql_select) {
selectList(sb);
}
}
private void selectList(StringBuffer sb) {
sb.append(" \r\n");
sb.append(" SELECT\r\n");
sb.append(" ");
for (int i = 0; i < colnames.length; i++) {
// 是否替换
if(i == 0) {
sb.append(colnames[i].toUpperCase());
} else if(i % 5 == 0){
sb.append(",\r\n " + colnames[i].toUpperCase());
} else {
sb.append(", " + colnames[i].toUpperCase());
}
}
sb.append("\r\n FROM " + tablename + "\r\n");
sb.append(" \r\n");
sb.append(" \r\n");
sb.append(" \r\n\r\n");
}
/**
* g功能：生成所有result
* @param sb
*/
private void processAllResult(StringBuffer sb) {
String colname = "";
for (int i = 0; i < colnames.length; i++) {
sb.append("
