R fail(Integer code, String message, T data) {
+ return new R<>(code, message, false, data);
+ }
+
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/common/ResultCode.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/common/ResultCode.java
new file mode 100644
index 0000000..3de00ad
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/common/ResultCode.java
@@ -0,0 +1,44 @@
+package com.xkcoding.codegen.common;
+
+import lombok.Getter;
+
+/**
+ *
+ * 通用状态枚举
+ *
+ *
+ * @package: com.xkcoding.codegen.common
+ * @description: 通用状态枚举
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 10:13
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+@Getter
+public enum ResultCode implements IResultCode {
+ /**
+ * 成功
+ */
+ OK(200, "成功"),
+ /**
+ * 失败
+ */
+ ERROR(500, "失败");
+
+ /**
+ * 返回码
+ */
+ private Integer code;
+
+ /**
+ * 返回消息
+ */
+ private String message;
+
+ ResultCode(Integer code, String message) {
+ this.code = code;
+ this.message = message;
+ }
+
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/constants/GenConstants.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/constants/GenConstants.java
new file mode 100644
index 0000000..8602bc2
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/constants/GenConstants.java
@@ -0,0 +1,26 @@
+package com.xkcoding.codegen.constants;
+
+/**
+ *
+ * 常量池
+ *
+ *
+ * @package: com.xkcoding.codegen.constants
+ * @description: 常量池
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 10:04
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+public interface GenConstants {
+ /**
+ * 签名
+ */
+ String SIGNATURE = "xkcoding";
+
+ /**
+ * JDBC连接串前缀
+ */
+ String JDBC_URL_PREFIX = "jdbc:mysql://";
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/controller/CodeGenController.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/controller/CodeGenController.java
new file mode 100755
index 0000000..841ea4d
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/controller/CodeGenController.java
@@ -0,0 +1,60 @@
+package com.xkcoding.codegen.controller;
+
+import cn.hutool.core.io.IoUtil;
+import com.xkcoding.codegen.common.R;
+import com.xkcoding.codegen.entity.GenConfig;
+import com.xkcoding.codegen.entity.TableRequest;
+import com.xkcoding.codegen.service.CodeGenService;
+import lombok.AllArgsConstructor;
+import lombok.SneakyThrows;
+import org.springframework.http.HttpHeaders;
+import org.springframework.web.bind.annotation.*;
+
+import javax.servlet.http.HttpServletResponse;
+
+/**
+ *
+ * 代码生成器
+ *
+ *
+ * @package: com.xkcoding.codegen.controller
+ * @description: 代码生成器
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 10:11
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+@RestController
+@AllArgsConstructor
+@RequestMapping("/generator")
+public class CodeGenController {
+ private final CodeGenService codeGenService;
+
+ /**
+ * 列表
+ *
+ * @param request 参数集
+ * @return 数据库表
+ */
+ @GetMapping("/table")
+ public R listTables(TableRequest request) {
+ return R.success(codeGenService.listTables(request));
+ }
+
+ /**
+ * 生成代码
+ */
+ @SneakyThrows
+ @PostMapping("")
+ public void generatorCode(@RequestBody GenConfig genConfig, HttpServletResponse response) {
+ byte[] data = codeGenService.generatorCode(genConfig);
+
+ response.reset();
+ response.setHeader(HttpHeaders.CONTENT_DISPOSITION, String.format("attachment; filename=%s.zip", genConfig.getTableName()));
+ response.addHeader(HttpHeaders.CONTENT_LENGTH, String.valueOf(data.length));
+ response.setContentType("application/octet-stream; charset=UTF-8");
+
+ IoUtil.write(response.getOutputStream(), Boolean.TRUE, data);
+ }
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/ColumnEntity.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/ColumnEntity.java
new file mode 100755
index 0000000..d106fb1
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/ColumnEntity.java
@@ -0,0 +1,48 @@
+package com.xkcoding.codegen.entity;
+
+import lombok.Data;
+
+/**
+ *
+ * 列属性： https://blog.csdn.net/lkforce/article/details/79557482
+ *
+ *
+ * @package: com.xkcoding.codegen.entity
+ * @description: 列属性： https://blog.csdn.net/lkforce/article/details/79557482
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 09:46
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+@Data
+public class ColumnEntity {
+ /**
+ * 列表
+ */
+ private String columnName;
+ /**
+ * 数据类型
+ */
+ private String dataType;
+ /**
+ * 备注
+ */
+ private String comments;
+ /**
+ * 驼峰属性
+ */
+ private String caseAttrName;
+ /**
+ * 普通属性
+ */
+ private String lowerAttrName;
+ /**
+ * 属性类型
+ */
+ private String attrType;
+ /**
+ * 其他信息
+ */
+ private String extra;
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/GenConfig.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/GenConfig.java
new file mode 100644
index 0000000..7e1cd0c
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/GenConfig.java
@@ -0,0 +1,48 @@
+package com.xkcoding.codegen.entity;
+
+import lombok.Data;
+
+/**
+ *
+ * 生成配置
+ *
+ *
+ * @package: com.xkcoding.codegen.entity
+ * @description: 生成配置
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 09:47
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+@Data
+public class GenConfig {
+ /**
+ * 请求参数
+ */
+ private TableRequest request;
+ /**
+ * 包名
+ */
+ private String packageName;
+ /**
+ * 作者
+ */
+ private String author;
+ /**
+ * 模块名称
+ */
+ private String moduleName;
+ /**
+ * 表前缀
+ */
+ private String tablePrefix;
+ /**
+ * 表名称
+ */
+ private String tableName;
+ /**
+ * 表备注
+ */
+ private String comments;
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/TableEntity.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/TableEntity.java
new file mode 100755
index 0000000..4786726
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/TableEntity.java
@@ -0,0 +1,46 @@
+package com.xkcoding.codegen.entity;
+
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ *
+ * 表属性： https://blog.csdn.net/lkforce/article/details/79557482
+ *
+ *
+ * @package: com.xkcoding.codegen.entity
+ * @description: 表属性： https://blog.csdn.net/lkforce/article/details/79557482
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 09:47
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+@Data
+public class TableEntity {
+ /**
+ * 名称
+ */
+ private String tableName;
+ /**
+ * 备注
+ */
+ private String comments;
+ /**
+ * 主键
+ */
+ private ColumnEntity pk;
+ /**
+ * 列名
+ */
+ private List columns;
+ /**
+ * 驼峰类型
+ */
+ private String caseClassName;
+ /**
+ * 普通类型
+ */
+ private String lowerClassName;
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/TableRequest.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/TableRequest.java
new file mode 100644
index 0000000..411faf9
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/entity/TableRequest.java
@@ -0,0 +1,44 @@
+package com.xkcoding.codegen.entity;
+
+import lombok.Data;
+
+/**
+ *
+ * 表格请求参数
+ *
+ *
+ * @package: com.xkcoding.codegen.entity
+ * @description: 表格请求参数
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 10:24
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+@Data
+public class TableRequest {
+ /**
+ * 当前页
+ */
+ private Integer currentPage;
+ /**
+ * 每页条数
+ */
+ private Integer pageSize;
+ /**
+ * jdbc-url
+ */
+ private String url;
+ /**
+ * 用户名
+ */
+ private String username;
+ /**
+ * 密码
+ */
+ private String password;
+ /**
+ * 表名
+ */
+ private String tableName;
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/service/CodeGenService.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/service/CodeGenService.java
new file mode 100644
index 0000000..7123db8
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/service/CodeGenService.java
@@ -0,0 +1,37 @@
+package com.xkcoding.codegen.service;
+
+import cn.hutool.db.Entity;
+import com.xkcoding.codegen.common.PageResult;
+import com.xkcoding.codegen.entity.GenConfig;
+import com.xkcoding.codegen.entity.TableRequest;
+
+/**
+ *
+ * 代码生成器
+ *
+ *
+ * @package: com.xkcoding.codegen.service
+ * @description: 代码生成器
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 10:15
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+public interface CodeGenService {
+ /**
+ * 生成代码
+ *
+ * @param genConfig 生成配置
+ * @return 代码压缩文件
+ */
+ byte[] generatorCode(GenConfig genConfig);
+
+ /**
+ * 分页查询表信息
+ *
+ * @param request 请求参数
+ * @return 表名分页信息
+ */
+ PageResult listTables(TableRequest request);
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/service/impl/CodeGenServiceImpl.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/service/impl/CodeGenServiceImpl.java
new file mode 100755
index 0000000..9ce59b8
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/service/impl/CodeGenServiceImpl.java
@@ -0,0 +1,128 @@
+package com.xkcoding.codegen.service.impl;
+
+import cn.hutool.core.io.IoUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.db.Db;
+import cn.hutool.db.Entity;
+import cn.hutool.db.Page;
+import com.xkcoding.codegen.common.PageResult;
+import com.xkcoding.codegen.entity.GenConfig;
+import com.xkcoding.codegen.entity.TableRequest;
+import com.xkcoding.codegen.service.CodeGenService;
+import com.xkcoding.codegen.utils.CodeGenUtil;
+import com.xkcoding.codegen.utils.DbUtil;
+import com.zaxxer.hikari.HikariDataSource;
+import lombok.AllArgsConstructor;
+import lombok.SneakyThrows;
+import org.springframework.stereotype.Service;
+
+import java.io.ByteArrayOutputStream;
+import java.math.BigDecimal;
+import java.util.List;
+import java.util.zip.ZipOutputStream;
+
+/**
+ * 代码生成器
+ *
+ * @author lengleng
+ * @date 2018-07-30
+ */
+@Service
+@AllArgsConstructor
+public class CodeGenServiceImpl implements CodeGenService {
+ private final String TABLE_SQL_TEMPLATE = "select table_name tableName, engine, table_comment tableComment, create_time createTime from information_schema.tables where table_schema = (select database()) %s order by create_time desc";
+
+ private final String COLUMN_SQL_TEMPLATE = "select column_name columnName, data_type dataType, column_comment columnComment, column_key columnKey, extra from information_schema.columns where table_name = ? and table_schema = (select database()) order by ordinal_position";
+
+ private final String COUNT_SQL_TEMPLATE = "select count(1) from (%s)tmp";
+
+ private final String PAGE_SQL_TEMPLATE = " limit ?,?";
+
+ /**
+ * 分页查询表信息
+ *
+ * @param request 请求参数
+ * @return 表名分页信息
+ */
+ @Override
+ @SneakyThrows
+ public PageResult listTables(TableRequest request) {
+ HikariDataSource dataSource = DbUtil.buildFromTableRequest(request);
+ Db db = new Db(dataSource);
+
+ Page page = new Page(request.getCurrentPage(), request.getPageSize());
+ int start = page.getStartPosition();
+ int pageSize = page.getPageSize();
+
+ String paramSql = StrUtil.EMPTY;
+ if (StrUtil.isNotBlank(request.getTableName())) {
+ paramSql = "and table_name like concat('%', ?, '%')";
+ }
+ String sql = String.format(TABLE_SQL_TEMPLATE, paramSql);
+ String countSql = String.format(COUNT_SQL_TEMPLATE, sql);
+
+ List query;
+ BigDecimal count;
+ if (StrUtil.isNotBlank(request.getTableName())) {
+ query = db.query(sql + PAGE_SQL_TEMPLATE, request.getTableName(), start, pageSize);
+ count = (BigDecimal) db.queryNumber(countSql, request.getTableName());
+ } else {
+ query = db.query(sql + PAGE_SQL_TEMPLATE, start, pageSize);
+ count = (BigDecimal) db.queryNumber(countSql);
+ }
+
+ PageResult pageResult = new PageResult<>(count.longValue(), page.getPageNumber(), page.getPageSize(), query);
+
+ dataSource.close();
+ return pageResult;
+ }
+
+ /**
+ * 生成代码
+ *
+ * @param genConfig 生成配置
+ * @return 代码压缩文件
+ */
+ @Override
+ public byte[] generatorCode(GenConfig genConfig) {
+ ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
+ ZipOutputStream zip = new ZipOutputStream(outputStream);
+
+ //查询表信息
+ Entity table = queryTable(genConfig.getRequest());
+ //查询列信息
+ List columns = queryColumns(genConfig.getRequest());
+ //生成代码
+ CodeGenUtil.generatorCode(genConfig, table, columns, zip);
+ IoUtil.close(zip);
+ return outputStream.toByteArray();
+ }
+
+ @SneakyThrows
+ private Entity queryTable(TableRequest request) {
+ HikariDataSource dataSource = DbUtil.buildFromTableRequest(request);
+ Db db = new Db(dataSource);
+
+ String paramSql = StrUtil.EMPTY;
+ if (StrUtil.isNotBlank(request.getTableName())) {
+ paramSql = "and table_name = ?";
+ }
+ String sql = String.format(TABLE_SQL_TEMPLATE, paramSql);
+ Entity entity = db.queryOne(sql, request.getTableName());
+
+ dataSource.close();
+ return entity;
+ }
+
+ @SneakyThrows
+ private List queryColumns(TableRequest request) {
+ HikariDataSource dataSource = DbUtil.buildFromTableRequest(request);
+ Db db = new Db(dataSource);
+
+ List query = db.query(COLUMN_SQL_TEMPLATE, request.getTableName());
+
+ dataSource.close();
+ return query;
+ }
+
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/utils/CodeGenUtil.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/utils/CodeGenUtil.java
new file mode 100644
index 0000000..3c9bfca
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/utils/CodeGenUtil.java
@@ -0,0 +1,267 @@
+package com.xkcoding.codegen.utils;
+
+import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.io.IoUtil;
+import cn.hutool.core.util.CharsetUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.db.Entity;
+import cn.hutool.setting.dialect.Props;
+import com.google.common.collect.Lists;
+import com.xkcoding.codegen.constants.GenConstants;
+import com.xkcoding.codegen.entity.ColumnEntity;
+import com.xkcoding.codegen.entity.GenConfig;
+import com.xkcoding.codegen.entity.TableEntity;
+import lombok.experimental.UtilityClass;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.text.WordUtils;
+import org.apache.velocity.Template;
+import org.apache.velocity.VelocityContext;
+import org.apache.velocity.app.Velocity;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.StringWriter;
+import java.nio.charset.StandardCharsets;
+import java.util.*;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
+
+/**
+ *
+ * 代码生成器 工具类
+ *
+ *
+ * @package: com.xkcoding.codegen.utils
+ * @description: 代码生成器 工具类
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 09:27
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+@Slf4j
+@UtilityClass
+public class CodeGenUtil {
+
+ private final String ENTITY_JAVA_VM = "Entity.java.vm";
+ private final String MAPPER_JAVA_VM = "Mapper.java.vm";
+ private final String SERVICE_JAVA_VM = "Service.java.vm";
+ private final String SERVICE_IMPL_JAVA_VM = "ServiceImpl.java.vm";
+ private final String CONTROLLER_JAVA_VM = "Controller.java.vm";
+ private final String MAPPER_XML_VM = "Mapper.xml.vm";
+ private final String API_JS_VM = "api.js.vm";
+
+ private List getTemplates() {
+ List templates = new ArrayList<>();
+ templates.add("template/Entity.java.vm");
+ templates.add("template/Mapper.java.vm");
+ templates.add("template/Mapper.xml.vm");
+ templates.add("template/Service.java.vm");
+ templates.add("template/ServiceImpl.java.vm");
+ templates.add("template/Controller.java.vm");
+
+ templates.add("template/api.js.vm");
+ return templates;
+ }
+
+ /**
+ * 生成代码
+ */
+ public void generatorCode(GenConfig genConfig, Entity table, List columns, ZipOutputStream zip) {
+ //配置信息
+ Props props = getConfig();
+ boolean hasBigDecimal = false;
+ //表信息
+ TableEntity tableEntity = new TableEntity();
+ tableEntity.setTableName(table.getStr("tableName"));
+
+ if (StrUtil.isNotBlank(genConfig.getComments())) {
+ tableEntity.setComments(genConfig.getComments());
+ } else {
+ tableEntity.setComments(table.getStr("tableComment"));
+ }
+
+ String tablePrefix;
+ if (StrUtil.isNotBlank(genConfig.getTablePrefix())) {
+ tablePrefix = genConfig.getTablePrefix();
+ } else {
+ tablePrefix = props.getStr("tablePrefix");
+ }
+
+ //表名转换成Java类名
+ String className = tableToJava(tableEntity.getTableName(), tablePrefix);
+ tableEntity.setCaseClassName(className);
+ tableEntity.setLowerClassName(StrUtil.lowerFirst(className));
+
+ //列信息
+ List columnList = Lists.newArrayList();
+ for (Entity column : columns) {
+ ColumnEntity columnEntity = new ColumnEntity();
+ columnEntity.setColumnName(column.getStr("columnName"));
+ columnEntity.setDataType(column.getStr("dataType"));
+ columnEntity.setComments(column.getStr("columnComment"));
+ columnEntity.setExtra(column.getStr("extra"));
+
+ //列名转换成Java属性名
+ String attrName = columnToJava(columnEntity.getColumnName());
+ columnEntity.setCaseAttrName(attrName);
+ columnEntity.setLowerAttrName(StrUtil.lowerFirst(attrName));
+
+ //列的数据类型，转换成Java类型
+ String attrType = props.getStr(columnEntity.getDataType(), "unknownType");
+ columnEntity.setAttrType(attrType);
+ if (!hasBigDecimal && "BigDecimal".equals(attrType)) {
+ hasBigDecimal = true;
+ }
+ //是否主键
+ if ("PRI".equalsIgnoreCase(column.getStr("columnKey")) && tableEntity.getPk() == null) {
+ tableEntity.setPk(columnEntity);
+ }
+
+ columnList.add(columnEntity);
+ }
+ tableEntity.setColumns(columnList);
+
+ //没主键，则第一个字段为主键
+ if (tableEntity.getPk() == null) {
+ tableEntity.setPk(tableEntity.getColumns().get(0));
+ }
+
+ //设置velocity资源加载器
+ Properties prop = new Properties();
+ prop.put("file.resource.loader.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader");
+ Velocity.init(prop);
+ //封装模板数据
+ Map map = new HashMap<>(16);
+ map.put("tableName", tableEntity.getTableName());
+ map.put("pk", tableEntity.getPk());
+ map.put("className", tableEntity.getCaseClassName());
+ map.put("classname", tableEntity.getLowerClassName());
+ map.put("pathName", tableEntity.getLowerClassName().toLowerCase());
+ map.put("columns", tableEntity.getColumns());
+ map.put("hasBigDecimal", hasBigDecimal);
+ map.put("datetime", DateUtil.now());
+ map.put("year", DateUtil.year(new Date()));
+
+ if (StrUtil.isNotBlank(genConfig.getComments())) {
+ map.put("comments", genConfig.getComments());
+ } else {
+ map.put("comments", tableEntity.getComments());
+ }
+
+ if (StrUtil.isNotBlank(genConfig.getAuthor())) {
+ map.put("author", genConfig.getAuthor());
+ } else {
+ map.put("author", props.getStr("author"));
+ }
+
+ if (StrUtil.isNotBlank(genConfig.getModuleName())) {
+ map.put("moduleName", genConfig.getModuleName());
+ } else {
+ map.put("moduleName", props.getStr("moduleName"));
+ }
+
+ if (StrUtil.isNotBlank(genConfig.getPackageName())) {
+ map.put("package", genConfig.getPackageName());
+ map.put("mainPath", genConfig.getPackageName());
+ } else {
+ map.put("package", props.getStr("package"));
+ map.put("mainPath", props.getStr("mainPath"));
+ }
+ VelocityContext context = new VelocityContext(map);
+
+ //获取模板列表
+ List templates = getTemplates();
+ for (String template : templates) {
+ //渲染模板
+ StringWriter sw = new StringWriter();
+ Template tpl = Velocity.getTemplate(template, CharsetUtil.UTF_8);
+ tpl.merge(context, sw);
+
+ try {
+ //添加到zip
+ zip.putNextEntry(new ZipEntry(Objects.requireNonNull(getFileName(template, tableEntity.getCaseClassName(), map
+ .get("package")
+ .toString(), map.get("moduleName").toString()))));
+ IoUtil.write(zip, StandardCharsets.UTF_8, false, sw.toString());
+ IoUtil.close(sw);
+ zip.closeEntry();
+ } catch (IOException e) {
+ throw new RuntimeException("渲染模板失败，表名：" + tableEntity.getTableName(), e);
+ }
+ }
+ }
+
+
+ /**
+ * 列名转换成Java属性名
+ */
+ private String columnToJava(String columnName) {
+ return WordUtils.capitalizeFully(columnName, new char[]{'_'}).replace("_", "");
+ }
+
+ /**
+ * 表名转换成Java类名
+ */
+ private String tableToJava(String tableName, String tablePrefix) {
+ if (StrUtil.isNotBlank(tablePrefix)) {
+ tableName = tableName.replaceFirst(tablePrefix, "");
+ }
+ return columnToJava(tableName);
+ }
+
+ /**
+ * 获取配置信息
+ */
+ private Props getConfig() {
+ Props props = new Props("generator.properties");
+ props.autoLoad(true);
+ return props;
+ }
+
+ /**
+ * 获取文件名
+ */
+ private String getFileName(String template, String className, String packageName, String moduleName) {
+ // 包路径
+ String packagePath = GenConstants.SIGNATURE + File.separator + "src" + File.separator + "main" + File.separator + "java" + File.separator;
+ // 资源路径
+ String resourcePath = GenConstants.SIGNATURE + File.separator + "src" + File.separator + "main" + File.separator + "resources" + File.separator;
+ // api路径
+ String apiPath = GenConstants.SIGNATURE + File.separator + "api" + File.separator;
+
+ if (StrUtil.isNotBlank(packageName)) {
+ packagePath += packageName.replace(".", File.separator) + File.separator + moduleName + File.separator;
+ }
+
+ if (template.contains(ENTITY_JAVA_VM)) {
+ return packagePath + "entity" + File.separator + className + ".java";
+ }
+
+ if (template.contains(MAPPER_JAVA_VM)) {
+ return packagePath + "mapper" + File.separator + className + "Mapper.java";
+ }
+
+ if (template.contains(SERVICE_JAVA_VM)) {
+ return packagePath + "service" + File.separator + className + "Service.java";
+ }
+
+ if (template.contains(SERVICE_IMPL_JAVA_VM)) {
+ return packagePath + "service" + File.separator + "impl" + File.separator + className + "ServiceImpl.java";
+ }
+
+ if (template.contains(CONTROLLER_JAVA_VM)) {
+ return packagePath + "controller" + File.separator + className + "Controller.java";
+ }
+
+ if (template.contains(MAPPER_XML_VM)) {
+ return resourcePath + "mapper" + File.separator + className + "Mapper.xml";
+ }
+
+ if (template.contains(API_JS_VM)) {
+ return apiPath + className.toLowerCase() + ".js";
+ }
+
+ return null;
+ }
+}
diff --git a/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/utils/DbUtil.java b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/utils/DbUtil.java
new file mode 100644
index 0000000..db6fc96
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/java/com/xkcoding/codegen/utils/DbUtil.java
@@ -0,0 +1,33 @@
+package com.xkcoding.codegen.utils;
+
+import com.xkcoding.codegen.constants.GenConstants;
+import com.xkcoding.codegen.entity.TableRequest;
+import com.zaxxer.hikari.HikariDataSource;
+import lombok.experimental.UtilityClass;
+import lombok.extern.slf4j.Slf4j;
+
+/**
+ *
+ * 数据库工具类
+ *
+ *
+ * @package: com.xkcoding.codegen.utils
+ * @description: 数据库工具类
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 10:26
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+@Slf4j
+@UtilityClass
+public class DbUtil {
+ public HikariDataSource buildFromTableRequest(TableRequest request) {
+ HikariDataSource dataSource = new HikariDataSource();
+ dataSource.setJdbcUrl(GenConstants.JDBC_URL_PREFIX + request.getUrl());
+ dataSource.setUsername(request.getUsername());
+ dataSource.setPassword(request.getPassword());
+ return dataSource;
+ }
+
+}
diff --git a/spring-boot-demo-codegen/src/main/resources/application.yml b/spring-boot-demo-codegen/src/main/resources/application.yml
new file mode 100644
index 0000000..a02fbde
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/resources/application.yml
@@ -0,0 +1,4 @@
+server:
+ port: 8080
+ servlet:
+ context-path: /demo
\ No newline at end of file
diff --git a/spring-boot-demo-codegen/src/main/resources/generator.properties b/spring-boot-demo-codegen/src/main/resources/generator.properties
new file mode 100755
index 0000000..0ef3933
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/resources/generator.properties
@@ -0,0 +1,29 @@
+#\u4EE3\u7801\u751F\u6210\u5668\uFF0C\u914D\u7F6E\u4FE1\u606F
+mainPath=com.xkcoding
+#\u5305\u540D
+package=com.xkcoding
+moduleName=generator
+#\u4F5C\u8005
+author=Yangkai.Shen
+#\u8868\u524D\u7F00(\u7C7B\u540D\u4E0D\u4F1A\u5305\u542B\u8868\u524D\u7F00)
+tablePrefix=tb_
+#\u7C7B\u578B\u8F6C\u6362\uFF0C\u914D\u7F6E\u4FE1\u606F
+tinyint=Integer
+smallint=Integer
+mediumint=Integer
+int=Integer
+integer=Integer
+bigint=Long
+float=Float
+double=Double
+decimal=BigDecimal
+bit=Boolean
+char=String
+varchar=String
+tinytext=String
+text=String
+mediumtext=String
+longtext=String
+date=LocalDateTime
+datetime=LocalDateTime
+timestamp=LocalDateTime
diff --git a/spring-boot-demo-codegen/src/main/resources/static/index.html b/spring-boot-demo-codegen/src/main/resources/static/index.html
new file mode 100644
index 0000000..acb87e1
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/resources/static/index.html
@@ -0,0 +1,313 @@
+
+
+
+
+ 代码生成器
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ jdbc:mysql://
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ 查询
+
+
+
+
+
+
+
+
+
+ {{ row.tableName }}
+
+
+ 生成代码
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
生成配置
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ 取消
+ 生成代码
+
+
+