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 }}
+
+
+ 生成代码
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
生成配置
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ 取消
+ 生成代码
+
+
+
\n \n '+String(e)+"\n
\n ",render:f,onClose:i,closable:l,type:"message"}),function(){var e=u++;return function(){v.remove(""+a+e)}}()}(t.content,t.duration,e,t.onClose,t.closable,t.render)},config:function(e){(e.top||0===e.top)&&(o.top=e.top),(e.duration||0===e.duration)&&(o.duration=e.duration)},destroy:function(){var e=c();l=null,e.destroy("ivu-message")}}},function(e,t,n){"use strict";Object.defineProperty(t,"__esModule",{value:!0});var i=n(196),r=n.n(i);for(var s in i)"default"!==s&&function(e){n.d(t,e,function(){return i[e]})}(s);var a=n(479),o=(n.n(a),n(0)),l=Object(o.a)(r.a,a.render,a.staticRenderFns,!1,null,null,null);t.default=l.exports},function(e,t,n){"use strict";Object.defineProperty(t,"__esModule",{value:!0});var i=n(197),r=n.n(i);for(var s in i)"default"!==s&&function(e){n.d(t,e,function(){return i[e]})}(s);var a=n(478),o=(n.n(a),n(0)),l=Object(o.a)(r.a,a.render,a.staticRenderFns,!1,null,null,null);t.default=l.exports},function(e,t,n){"use strict";Object.defineProperty(t,"__esModule",{value:!0});t.render=function(){var e=this,t=e.$createElement,n=e._self._c||t;return n("transition",{attrs:{name:e.transitionName},on:{enter:e.handleEnter,leave:e.handleLeave}},[n("div",{class:e.classes,style:e.styles},["notice"===e.type?[n("div",{ref:"content",class:e.contentClasses,domProps:{innerHTML:e._s(e.content)}}),e._v(" "),n("div",{class:e.contentWithIcon},[n("render-cell",{attrs:{render:e.renderFunc}})],1),e._v(" "),e.closable?n("a",{class:[e.baseClass+"-close"],on:{click:e.close}},[n("i",{staticClass:"ivu-icon ivu-icon-ios-close"})]):e._e()]:e._e(),e._v(" "),"message"===e.type?[n("div",{ref:"content",class:[e.baseClass+"-content"]},[n("div",{class:[e.baseClass+"-content-text"],domProps:{innerHTML:e._s(e.content)}}),e._v(" "),n("div",{class:[e.baseClass+"-content-text"]},[n("render-cell",{attrs:{render:e.renderFunc}})],1),e._v(" "),e.closable?n("a",{class:[e.baseClass+"-close"],on:{click:e.close}},[n("i",{staticClass:"ivu-icon ivu-icon-ios-close"})]):e._e()])]:e._e()],2)])},t.staticRenderFns=[]},function(e,t,n){"use strict";Object.defineProperty(t,"__esModule",{value:!0});t.render=function(){var e=this,t=e.$createElement,n=e._self._c||t;return n("div",{class:e.classes,style:e.wrapStyles},e._l(e.notices,function(t){return n("Notice",{key:t.name,attrs:{"prefix-cls":e.prefixCls,styles:t.styles,type:t.type,content:t.content,duration:t.duration,render:t.render,"has-title":t.hasTitle,withIcon:t.withIcon,closable:t.closable,name:t.name,"transition-name":t.transitionName,"on-close":t.onClose}})}),1)},t.staticRenderFns=[]},function(e,t,n){"use strict";Object.defineProperty(t,"__esModule",{value:!0});var i=function(e){return e&&e.__esModule?e:{default:e}}(n(481));var r=void 0;function s(){var e=arguments.length>0&&void 0!==arguments[0]?arguments[0]:void 0;return r=r||i.default.newInstance({closable:!1,maskClosable:!1,footerHide:!0,render:e})}function a(e){var t=s("render"in e?e.render:void 0);e.onRemove=function(){r=null},t.show(e)}i.default.info=function(){var e=arguments.length>0&&void 0!==arguments[0]?arguments[0]:{};return e.icon="info",e.showCancel=!1,a(e)},i.default.success=function(){var e=arguments.length>0&&void 0!==arguments[0]?arguments[0]:{};return e.icon="success",e.showCancel=!1,a(e)},i.default.warning=function(){var e=arguments.length>0&&void 0!==arguments[0]?arguments[0]:{};return e.icon="warning",e.showCancel=!1,a(e)},i.default.error=function(){var e=arguments.length>0&&void 0!==arguments[0]?arguments[0]:{};return e.icon="error",e.showCancel=!1,a(e)},i.default.confirm=function(){var e=arguments.length>0&&void 0!==arguments[0]?arguments[0]:{};return e.icon="confirm",e.showCancel=!0,a(e)},i.default.remove=function(){if(!r)return!1;s().remove()},t.default=i.default},function(e,t,n){"use strict";Object.defineProperty(t,"__esModule",{value:!0});var i=u(n(12)),r=u(n(1)),s=u(n(13)),a=u(n(482)),o=u(n(24)),l=u(n(5));function u(e){return e&&e.__esModule?e:{default:e}}var d="ivu-modal-confirm";a.default.newInstance=function(e){(0,r.default)(void 0,void 0);var t=e||{},n=new s.default({mixins:[l.default],data:(0,i.default)({},t,{visible:!1,width:416,title:"",body:"",iconType:"",iconName:"",okText:void 0,cancelText:void 0,showCancel:!1,loading:!1,buttonLoading:!1,scrollable:!1,closable:!1}),render:function(e){var n=this,s=[];this.showCancel&&s.push(e(o.default,{props:{type:"text",size:"large"},on:{click:this.cancel}},this.localeCancelText)),s.push(e(o.default,{props:{type:"primary",size:"large",loading:this.buttonLoading},on:{click:this.ok}},this.localeOkText));var l=void 0;l=this.render?e("div",{attrs:{class:d+"-body "+d+"-body-render"}},[this.render(e)]):e("div",{attrs:{class:d+"-body"}},[e("div",{domProps:{innerHTML:this.body}})]);var u=void 0;return this.title&&(u=e("div",{attrs:{class:d+"-head"}},[e("div",{class:this.iconTypeCls},[e("i",{class:this.iconNameCls})]),e("div",{attrs:{class:d+"-head-title"},domProps:{innerHTML:this.title}})])),e(a.default,{props:(0,i.default)({},t,{width:this.width,scrollable:this.scrollable,closable:this.closable}),domProps:{value:this.visible},on:{input:function(e){(0,r.default)(this,n),this.visible=e}.bind(this)}},[e("div",{attrs:{class:d}},[u,l,e("div",{attrs:{class:d+"-footer"}},s)])])},computed:{iconTypeCls:function(){return[d+"-head-icon",d+"-head-icon-"+String(this.iconType)]},iconNameCls:function(){return["ivu-icon","ivu-icon-"+String(this.iconName)]},localeOkText:function(){return this.okText?this.okText:this.t("i.modal.okText")},localeCancelText:function(){return this.cancelText?this.cancelText:this.t("i.modal.cancelText")}},methods:{cancel:function(){this.$children[0].visible=!1,this.buttonLoading=!1,this.onCancel(),this.remove()},ok:function(){this.loading?this.buttonLoading=!0:(this.$children[0].visible=!1,this.remove()),this.onOk()},remove:function(){var e=this;setTimeout(function(){(0,r.default)(this,e),this.destroy()}.bind(this),300)},destroy:function(){this.$destroy(),document.body.removeChild(this.$el),this.onRemove()},onOk:function(){},onCancel:function(){},onRemove:function(){}}}),u=n.$mount();document.body.appendChild(u.$el);var c=n.$children[0];return{show:function(e){switch(c.$parent.showCancel=e.showCancel,c.$parent.iconType=e.icon,e.icon){case"info":c.$parent.iconName="ios-information-circle";break;case"success":c.$parent.iconName="ios-checkmark-circle";break;case"warning":c.$parent.iconName="ios-alert";break;case"error":c.$parent.iconName="ios-close-circle";break;case"confirm":c.$parent.iconName="ios-help-circle"}"width"in e&&(c.$parent.width=e.width),"closable"in e&&(c.$parent.closable=e.closable),"title"in e&&(c.$parent.title=e.title),"content"in e&&(c.$parent.body=e.content),"okText"in e&&(c.$parent.okText=e.okText),"cancelText"in e&&(c.$parent.cancelText=e.cancelText),"onCancel"in e&&(c.$parent.onCancel=e.onCancel),"onOk"in e&&(c.$parent.onOk=e.onOk),"loading"in e&&(c.$parent.loading=e.loading),"scrollable"in e&&(c.$parent.scrollable=e.scrollable),c.$parent.onRemove=e.onRemove,c.visible=!0},remove:function(){c.visible=!1,c.$parent.buttonLoading=!1,c.$parent.remove()},component:c}}.bind(void 0),t.default=a.default},function(e,t,n){"use strict";Object.defineProperty(t,"__esModule",{value:!0});var i=n(199),r=n.n(i);for(var s in i)"default"!==s&&function(e){n.d(t,e,function(){return i[e]})}(s);var a=n(483),o=(n.n(a),n(0)),l=Object(o.a)(r.a,a.render,a.staticRenderFns,!1,null,null,null);t.default=l.exports},function(e,t,n){"use strict";Object.defineProperty(t,"__esModule",{value:!0});t.render=function(){var e=this,t=e.$createElement,n=e._self._c||t;return n("div",{directives:[{name:"transfer-dom",rawName:"v-transfer-dom"}],attrs:{"data-transfer":e.transfer}},[n("transition",{attrs:{name:e.transitionNames[1]}},[e.showMask?n("div",{directives:[{name:"show",rawName:"v-show",value:e.visible,expression:"visible"}],class:e.maskClasses,style:e.wrapStyles,on:{click:e.handleMask}}):e._e()]),e._v(" "),n("div",{class:e.wrapClasses,style:e.wrapStyles,on:{click:e.handleWrapClick}},[n("transition",{attrs:{name:e.transitionNames[0]},on:{"after-leave":e.animationFinish}},[n("div",{directives:[{name:"show",rawName:"v-show",value:e.visible,expression:"visible"}],class:e.classes,style:e.mainStyles},[n("div",{ref:"content",class:e.contentClasses,style:e.contentStyles,on:{click:e.handleClickModal}},[e.closable?n("a",{class:[e.prefixCls+"-close"],on:{click:e.close}},[e._t("close",[n("Icon",{attrs:{type:"ios-close"}})])],2):e._e(),e._v(" "),e.showHead?n("div",{class:[e.prefixCls+"-header"],on:{mousedown:e.handleMoveStart}},[e._t("header",[n("div",{class:[e.prefixCls+"-header-inner"]},[e._v(e._s(e.title))])])],2):e._e(),e._v(" "),n("div",{class:[e.prefixCls+"-body"]},[e._t("default")],2),e._v(" "),e.footerHide?e._e():n("div",{class:[e.prefixCls+"-footer"]},[e._t("footer",[n("i-button",{attrs:{type:"text",size:"large"},nativeOn:{click:function(t){return e.cancel(t)}}},[e._v(e._s(e.localeCancelText))]),e._v(" "),n("i-button",{attrs:{type:"primary",size:"large",loading:e.buttonLoading},nativeOn:{click:function(t){return e.ok(t)}}},[e._v(e._s(e.localeOkText))])])],2)])])])],1)],1)},t.staticRenderFns=[]},function(e,t,n){"use strict";Object.defineProperty(t,"__esModule",{value:!0});var i=function(e){return e&&e.__esModule?e:{default:e}}(n(195));var r="ivu-notice",s="ivu-icon",a="ivu_notice_key_",o=24,l=4.5,u=void 0,d=1,c={info:"ios-information-circle",success:"ios-checkmark-circle",warning:"ios-alert",error:"ios-close-circle"};function f(){return u=u||i.default.newInstance({prefixCls:r,styles:{top:o+"px",right:0}})}function h(e,t){var n=t.title||"",i=t.desc||"",o=t.name||""+a+d,u=t.onClose||function(){},h=t.render,p=0===t.duration?0:t.duration||l;d++;var v=f(),m=void 0,g=void 0,b=t.render&&!n?"":i||t.render?" "+r+"-with-desc":"";if("normal"==e)g=!1,m='\n \n
'+String(n)+'
\n
'+String(i)+"
\n
\n
\n \n \n
'+String(n)+'
\n
'+String(i)+"
\n
+ * ${comments}
+ *
+ *
+ * @package: ${package}.${moduleName}.controller
+ * @description: ${comments}
+ * @author: ${author}
+ * @date: Created in ${datetime}
+ * @copyright: Copyright (c) ${year}
+ * @version: V1.0
+ * @modified: ${author}
+ */
+@RestController
+@AllArgsConstructor
+@RequestMapping("/${pathName}")
+public class ${className}Controller {
+
+ private final ${className}Service ${classname}Service;
+
+ /**
+ * 分页查询${comments}
+ * @param page 分页对象
+ * @param ${classname} ${comments}
+ * @return R
+ */
+ @GetMapping("")
+ public R list${className}(Page page, ${className} ${classname}) {
+ return new R<>(${classname}Service.page(page,Wrappers.query(${classname})));
+ }
+
+
+ /**
+ * 通过id查询${comments}
+ * @param ${pk.lowerAttrName} id
+ * @return R
+ */
+ @GetMapping("/{${pk.lowerAttrName}}")
+ public R get${className}(@PathVariable("${pk.lowerAttrName}") ${pk.attrType} ${pk.lowerAttrName}){
+ return new R<>(${classname}Service.getById(${pk.lowerAttrName}));
+ }
+
+ /**
+ * 新增${comments}
+ * @param ${classname} ${comments}
+ * @return R
+ */
+ @ApiLog("新增${comments}")
+ @PostMapping
+ public R save${className}(@RequestBody ${className} ${classname}){
+ return new R<>(${classname}Service.save(${classname}));
+ }
+
+ /**
+ * 修改${comments}
+ * @param ${pk.lowerAttrName} id
+ * @param ${classname} ${comments}
+ * @return R
+ */
+ @ApiLog("修改${comments}")
+ @PutMapping("/{${pk.lowerAttrName}}")
+ public R update${className}(@PathVariable ${pk.attrType} ${pk.lowerAttrName}, @RequestBody ${className} ${classname}){
+ return new R<>(${classname}Service.updateById(${classname}));
+ }
+
+ /**
+ * 通过id删除${comments}
+ * @param ${pk.lowerAttrName} id
+ * @return R
+ */
+ @ApiLog("删除${comments}")
+ @DeleteMapping("/{${pk.lowerAttrName}}")
+ public R delete${className}(@PathVariable ${pk.attrType} ${pk.lowerAttrName}){
+ return new R<>(${classname}Service.removeById(${pk.lowerAttrName}));
+ }
+
+}
diff --git a/spring-boot-demo-codegen/src/main/resources/template/Entity.java.vm b/spring-boot-demo-codegen/src/main/resources/template/Entity.java.vm
new file mode 100755
index 0000000..6456f4c
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/resources/template/Entity.java.vm
@@ -0,0 +1,43 @@
+package ${package}.${moduleName}.entity;
+
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import com.baomidou.mybatisplus.extension.activerecord.Model;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+#if(${hasBigDecimal})
+import java.math.BigDecimal;
+#end
+import java.io.Serializable;
+import java.time.LocalDateTime;
+
+/**
+ *
+ * ${comments}
+ *
+ *
+ * @package: ${package}.${moduleName}.entity
+ * @description: ${comments}
+ * @author: ${author}
+ * @date: Created in ${datetime}
+ * @copyright: Copyright (c) ${year}
+ * @version: V1.0
+ * @modified: ${author}
+ */
+@Data
+@TableName("${tableName}")
+@EqualsAndHashCode(callSuper = true)
+public class ${className} extends Model<${className}> {
+ private static final long serialVersionUID = 1L;
+
+ #foreach ($column in $columns)
+ /**
+ * $column.comments
+ */
+ #if($column.columnName == $pk.columnName)
+ @TableId
+ #end
+ private $column.attrType $column.lowerAttrName;
+ #end
+
+}
diff --git a/spring-boot-demo-codegen/src/main/resources/template/Mapper.java.vm b/spring-boot-demo-codegen/src/main/resources/template/Mapper.java.vm
new file mode 100755
index 0000000..7415cb8
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/resources/template/Mapper.java.vm
@@ -0,0 +1,23 @@
+package ${package}.${moduleName}.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.springframework.stereotype.Component;
+import ${package}.${moduleName}.entity.${className};
+
+/**
+ *
+ * ${comments}
+ *
+ *
+ * @package: ${package}.${moduleName}.mapper
+ * @description: ${comments}
+ * @author: ${author}
+ * @date: Created in ${datetime}
+ * @copyright: Copyright (c) ${year}
+ * @version: V1.0
+ * @modified: ${author}
+ */
+@Component
+public interface ${className}Mapper extends BaseMapper<${className}> {
+
+}
diff --git a/spring-boot-demo-codegen/src/main/resources/template/Mapper.xml.vm b/spring-boot-demo-codegen/src/main/resources/template/Mapper.xml.vm
new file mode 100755
index 0000000..d8b2fc2
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/resources/template/Mapper.xml.vm
@@ -0,0 +1,13 @@
+
+
+
+
+ #foreach($column in $columns)
+ #if($column.lowerAttrName==$pk.lowerAttrName)
+
+ #else
+
+ #end
+ #end
+
+
diff --git a/spring-boot-demo-codegen/src/main/resources/template/Service.java.vm b/spring-boot-demo-codegen/src/main/resources/template/Service.java.vm
new file mode 100755
index 0000000..028598f
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/resources/template/Service.java.vm
@@ -0,0 +1,21 @@
+package ${package}.${moduleName}.service;
+
+import com.baomidou.mybatisplus.extension.service.IService;
+import ${package}.${moduleName}.entity.${className};
+
+/**
+ *
+ * ${comments}
+ *
+ *
+ * @package: ${package}.${moduleName}.service
+ * @description: ${comments}
+ * @author: ${author}
+ * @date: Created in ${datetime}
+ * @copyright: Copyright (c) ${year}
+ * @version: V1.0
+ * @modified: ${author}
+ */
+public interface ${className}Service extends IService<${className}> {
+
+}
diff --git a/spring-boot-demo-codegen/src/main/resources/template/ServiceImpl.java.vm b/spring-boot-demo-codegen/src/main/resources/template/ServiceImpl.java.vm
new file mode 100755
index 0000000..2fa0e6c
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/resources/template/ServiceImpl.java.vm
@@ -0,0 +1,25 @@
+package ${package}.${moduleName}.service.impl;
+
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import ${package}.${moduleName}.entity.${className};
+import ${package}.${moduleName}.mapper.${className}Mapper;
+import ${package}.${moduleName}.service.${className}Service;
+import org.springframework.stereotype.Service;
+
+/**
+ *
+ * ${comments}
+ *
+ *
+ * @package: ${package}.${moduleName}.service.impl
+ * @description: ${comments}
+ * @author: ${author}
+ * @date: Created in ${datetime}
+ * @copyright: Copyright (c) ${year}
+ * @version: V1.0
+ * @modified: ${author}
+ */
+@Service
+public class ${className}ServiceImpl extends ServiceImpl<${className}Mapper, ${className}> implements ${className}Service {
+
+}
diff --git a/spring-boot-demo-codegen/src/main/resources/template/api.js.vm b/spring-boot-demo-codegen/src/main/resources/template/api.js.vm
new file mode 100644
index 0000000..a367f1e
--- /dev/null
+++ b/spring-boot-demo-codegen/src/main/resources/template/api.js.vm
@@ -0,0 +1,60 @@
+import request from '@/router/axios'
+
+/**
+ * 分页查询${comments}
+ * @param query 分页查询条件
+ */
+export function fetchList(query) {
+ return request({
+ url: '/${moduleName}/${pathName}',
+ method: 'get',
+ params: query
+ })
+}
+
+/**
+ * 新增${comments}
+ * @param obj ${comments}
+ */
+export function addObj(obj) {
+ return request({
+ url: '/${moduleName}/${pathName}',
+ method: 'post',
+ data: obj
+ })
+}
+
+/**
+ * 通过id查询${comments}
+ * @param id 主键
+ */
+export function getObj(id) {
+ return request({
+ url: '/${moduleName}/${pathName}/' + id,
+ method: 'get'
+ })
+}
+
+/**
+ * 通过id删除${comments}
+ * @param id 主键
+ */
+export function delObj(id) {
+ return request({
+ url: '/${moduleName}/${pathName}/' + id,
+ method: 'delete'
+ })
+}
+
+/**
+ * 修改${comments}
+ * @param id 主键
+ * @param obj ${comments}
+ */
+export function putObj(id, obj) {
+ return request({
+ url: '/${moduleName}/${pathName}/' + id,
+ method: 'put',
+ data: obj
+ })
+}
diff --git a/spring-boot-demo-codegen/src/test/java/com/xkcoding/codegen/CodeGenServiceTest.java b/spring-boot-demo-codegen/src/test/java/com/xkcoding/codegen/CodeGenServiceTest.java
new file mode 100644
index 0000000..145524c
--- /dev/null
+++ b/spring-boot-demo-codegen/src/test/java/com/xkcoding/codegen/CodeGenServiceTest.java
@@ -0,0 +1,77 @@
+package com.xkcoding.codegen;
+
+import cn.hutool.core.io.IoUtil;
+import cn.hutool.db.Entity;
+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 lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringRunner;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.OutputStream;
+
+/**
+ *
+ * 代码生成service测试
+ *
+ *
+ * @package: com.xkcoding.codegen
+ * @description: 代码生成service测试
+ * @author: yangkai.shen
+ * @date: Created in 2019-03-22 10:34
+ * @copyright: Copyright (c) 2019
+ * @version: V1.0
+ * @modified: yangkai.shen
+ */
+@RunWith(SpringRunner.class)
+@SpringBootTest
+@Slf4j
+public class CodeGenServiceTest {
+ @Autowired
+ private CodeGenService codeGenService;
+
+ @Test
+ public void testTablePage() {
+ TableRequest request = new TableRequest();
+ request.setCurrentPage(1);
+ request.setPageSize(10);
+ request.setUrl("jdbc:mysql://127.0.0.1:3306/spring-boot-demo");
+ request.setUsername("root");
+ request.setPassword("root");
+ request.setTableName("sec_");
+ PageResult pageResult = codeGenService.listTables(request);
+ log.info("【pageResult】= {}", pageResult);
+ }
+
+ @Test
+ @SneakyThrows
+ public void testGeneratorCode() {
+ GenConfig config = new GenConfig();
+
+ TableRequest request = new TableRequest();
+ request.setUrl("127.0.0.1:3306/spring-boot-demo");
+ request.setUsername("root");
+ request.setPassword("root");
+ request.setTableName("shiro_user");
+ config.setRequest(request);
+
+ config.setModuleName("shiro");
+ config.setAuthor("Yangkai.Shen");
+ config.setComments("用户角色信息");
+ config.setPackageName("com.xkcoding");
+ config.setTablePrefix("shiro_");
+
+ byte[] zip = codeGenService.generatorCode(config);
+ OutputStream outputStream = new FileOutputStream(new File("/Users/yangkai.shen/Desktop/" + request.getTableName() + ".zip"));
+ IoUtil.write(outputStream, true, zip);
+ }
+
+}
diff --git a/spring-boot-demo-codegen/src/test/java/com/xkcoding/codegen/SpringBootDemoCodegenApplicationTests.java b/spring-boot-demo-codegen/src/test/java/com/xkcoding/codegen/SpringBootDemoCodegenApplicationTests.java
new file mode 100644
index 0000000..b5b45e3
--- /dev/null
+++ b/spring-boot-demo-codegen/src/test/java/com/xkcoding/codegen/SpringBootDemoCodegenApplicationTests.java
@@ -0,0 +1,16 @@
+package com.xkcoding.codegen;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringRunner;
+
+@RunWith(SpringRunner.class)
+@SpringBootTest
+public class SpringBootDemoCodegenApplicationTests {
+
+ @Test
+ public void contextLoads() {
+ }
+
+}