package com.yqzx.generator.service;
|
|
import cn.hutool.core.util.StrUtil;
|
import com.yqzx.generator.config.DataSourceConfig;
|
import com.yqzx.generator.action.model.DbColumnType;
|
import com.yqzx.generator.action.model.Table;
|
import com.yqzx.generator.action.model.TableInfo;
|
import com.yqzx.generator.action.model.Template;
|
import lombok.extern.slf4j.Slf4j;
|
|
import java.sql.Connection;
|
import java.sql.ResultSet;
|
import java.sql.SQLException;
|
import java.sql.Statement;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* @description: 表工具类
|
* @author: chaoyapeng
|
* @time: 2020/8/18 16:13
|
*/
|
@Slf4j
|
public class GeneratorService {
|
|
private DataSourceConfig dataSourceConfig;
|
|
public GeneratorService setDataSourceConfig(DataSourceConfig dataSourceConfig) {
|
this.dataSourceConfig = dataSourceConfig;
|
return this;
|
}
|
|
public GeneratorService setDataSourceConfig(String driverName, String username, String password, String url) {
|
DataSourceConfig dataSourceConfig = new DataSourceConfig();
|
dataSourceConfig.setDriverName(driverName);
|
dataSourceConfig.setUsername(username);
|
dataSourceConfig.setPassword(password);
|
dataSourceConfig.setUrl(url);
|
this.dataSourceConfig = dataSourceConfig;
|
return this;
|
}
|
|
/**
|
* 查询数据库表信息
|
*
|
* @param schema
|
* @return
|
*/
|
public List<Table> getTable(String schema) {
|
{
|
// 获取数据库连接，查询表结构信息
|
Connection conn = dataSourceConfig.getConn();
|
Statement sm = null;
|
List<Table> tables = new ArrayList<>();
|
try {
|
sm = conn.createStatement();
|
StringBuffer sql = new StringBuffer();
|
sql.append("select table_name as tablename,table_comment as tablecomment from information_schema.`tables` where table_schema = '");
|
sql.append(schema);
|
sql.append("'");
|
/*String sql = "select table_name as tablename,table_comment as tablecomment from information_schema.`tables` where table_schema = '"
|
+ schema + "'";*/
|
ResultSet rs = sm.executeQuery(sql.toString());
|
while (rs.next()) {
|
Table table = new Table();
|
table.setName(rs.getString(1));
|
table.setComment(rs.getString(2));
|
table.setLabel(table.getName() + "-" + table.getComment());
|
tables.add(table);
|
}
|
} catch (SQLException e) {
|
log.error(e.getMessage());
|
}
|
if (sm != null) {
|
try {
|
sm.close();
|
} catch (SQLException e) {
|
log.error(e.getMessage());
|
}
|
}
|
if (conn != null) {
|
try {
|
conn.close();
|
} catch (SQLException e) {
|
log.error(e.getMessage());
|
}
|
}
|
return tables;
|
}
|
}
|
|
/**
|
* 查询表结构信息
|
*
|
* @param tableName
|
* @return
|
*/
|
public List<TableInfo> getTableInfo(String tableName) {
|
Connection conn = dataSourceConfig.getConn();
|
Statement sm = null;
|
List<TableInfo> tableInfos = new ArrayList<>();
|
try {
|
sm = conn.createStatement();
|
StringBuffer sql = new StringBuffer();
|
sql.append("select column_name, data_type, column_comment from information_schema.columns where table_schema = (select database()) and table_name = '");
|
sql.append(tableName);
|
sql.append("'");
|
/*String sql = "select column_name, data_type, column_comment from information_schema.columns where table_schema = (select database()) and table_name = '"
|
+ tableName + "'";*/
|
ResultSet rs = sm.executeQuery(sql.toString());
|
while (rs.next()) {
|
TableInfo tableInfo = new TableInfo();
|
tableInfo.setName(rs.getString(1));
|
DbColumnType column = DbColumnType.get(rs.getString(2));
|
tableInfo.setType(column.getType());
|
tableInfo.setJdbcType(column.getJdbcType());
|
tableInfo.setComment(rs.getString(3));
|
tableInfo.setPropertyName(StrUtil.toCamelCase(tableInfo.getName()));
|
tableInfos.add(tableInfo);
|
}
|
if (tableInfos.size() > 0) {
|
TableInfo t = tableInfos.get(tableInfos.size() - 1);
|
tableInfos.remove(tableInfos.size() - 1);
|
t.setLast(true);
|
tableInfos.add(t);
|
}
|
} catch (SQLException e) {
|
log.error(e.getMessage());
|
}
|
if (sm != null) {
|
try {
|
sm.close();
|
} catch (SQLException e) {
|
log.error(e.getMessage());
|
}
|
}
|
if (conn != null) {
|
try {
|
conn.close();
|
} catch (SQLException e) {
|
log.error(e.getMessage());
|
}
|
}
|
return tableInfos;
|
}
|
|
/**
|
* 获取模板内容
|
*
|
* @return
|
*/
|
public List<Template> getTemplate() {
|
List<Template> templates = new ArrayList<>();
|
templates.add(create("controllerSwitch", "controller-模板"));
|
templates.add(create("serviceSwitch", "service-模板（包含Impl）"));
|
templates.add(create("mapperSwitch", "mapper-模板（包含xml和继承xml）"));
|
templates.add(create("modelSwitch", "model-模板"));
|
templates.add(create("querySwitch", "query-模板（列表查询参数，默认全部model内容）"));
|
return templates;
|
}
|
|
private static Template create(String name, String comment) {
|
Template template = new Template();
|
template.setName(name);
|
template.setComment(comment);
|
return template;
|
}
|
|
}
|
