轻量级代码生成器加测试数据生成器

大忽悠爱学习

发布于 2023-02-02 15:55:09

8970

文章被收录于专栏：c++与qt学习c++与qt学习

轻量级代码生成器加测试数据生成器

Gitee项目链接

介绍

本项目是一个轻量级代码生成器,并提供多种方式来完成模拟数据的批量生产,项目架构如下:

代码生成

常用注解

@Table: 指明表名和表注释

@Retention(RetentionPolicy.RUNTIME)
@Target(value = ElementType.TYPE)
public @interface Table {
    /**
     * 不指定默认为类名小写转驼峰
     */
    String value() default "";
    /**
     * 注释
     */
    String comment() default "";
    /**
     * 是否需要添加dropTable语句
     */
    boolean dropTable() default true;
}

@Field: 指明当前表中某个字段的相关信息

@Retention(RetentionPolicy.RUNTIME)
@Target(value = ElementType.FIELD)
public @interface Filed {
    /**
     * 字段名: 不指定,默认为属性名小写转驼峰
     */
    String value() default "";
    /**
     * 字段类型,可以手动指定确切的字段类型,否则采用内定的java-sql类型转换过程
     */
    String filedType() default "";
    /**
     * 字段长度,对于varchar,char等类型来说是需要进行设置的,这里也可以不进行设置,会给予默认大小
     */
    String filedLen() default "";
    /**
     * 字段是否为notNull,默认为true
     */
    boolean notNull() default true;
    /**
     * 字段是否自增
     */
    boolean autoIncrement() default false;
    /**
     * 字段默认值
     */
    String defaultVal() default "";
    /**
     * 字段描述
     */
    String comment() default "";
    /**
     * 字段顺序,值越小优先级越大
     */
    int order() default Integer.MAX_VALUE;
}

@Key:

@Retention(RetentionPolicy.RUNTIME)
@Target(value = ElementType.FIELD)
public @interface Key {
    /**
     * 索引类型
     */
    String keyType() default "";
    /**
     * 索引名
     */
    String keyName() default  "";
    /**
     * 索引列: 默认包含当前字段,通过该属性指定其他字段---对于复合索引而言
     */
    String[] otherColumns() default {};


    //-----------外键索引----------------

    /**
     * 主表名
     */
    String masterTableName() default "";
    /**
     * 受约束的主表字段
     */
    String[] masterColumns() default {};
    /**
     * 受约束从表字段: 默认包含当前字段,通过该属性指定其他字段,对于复合索引而言
     */
    String[] otherSalveColumns() default {};
}

基本使用

@Table(comment = "学生表",value = "stu")
public class Stu {
    @Key(keyType = KeyTypeConstants.PRIMARY_KEY)
    @Filed(comment = "主键id", autoIncrement = true)
    private Integer id;
    @Filed(comment = "姓名")
    private String name;
    @Filed(comment = "年龄")
    private Integer age;
    @Filed(comment = "就读大学")
    private String university;
    @Filed(comment = "主攻编程语言")
    private String language;
    @Filed(comment = "老家城市")
    private String homeCity;
    @Filed(comment = "电话号码")
    private String phoneNumber;
    @Filed(comment = "qq邮箱号码")
    private String qqEamil;
    @Filed(comment = "出生日期")
    private LocalDateTime birthTime;
    @Filed(comment = "创建时间", defaultVal = DefaultValConstants.CURRENT_TIMESTAMP)
    private LocalDateTime createTime;
    @Filed(comment = "修改时间", defaultVal = DefaultValConstants.CURRENT_TIMESTAMP)
    private LocalDateTime updateTime;
}

测试

@SpringBootTest
public class ExampleOneTest {
    @Autowired
    private SqlManager sqlManager;

    @Test
    public void test() {
         sqlManager.addMappedObject(new Stu());
         sqlManager.build();
    }
}

生成的模板文件(如果不满意,可以自定义模板,后面会介绍):

产生的建库语句:

CREATE DATABASE IF NOT EXISTS `test`;

USE `test`;
SET NAMES utf8mb4;
SET FOREIGN_KEY_CHECKS = 0;

DROP TABLE IF EXISTS `stu`;

CREATE TABLE `stu` (
	`id` int NOT NULL AUTO_INCREMENT COMMENT '主键id',
	`name` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '姓名',
	`age` int NOT NULL COMMENT '年龄',
	`university` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '就读大学',
	`language` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '主攻编程语言',
	`home_city` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '老家城市',
	`phone_number` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '电话号码',
	`qq_eamil` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT 'qq邮箱号码',
	`birth_time` datetime NOT NULL COMMENT '出生日期',
	`create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
	`update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '修改时间',
	PRIMARY KEY (id)
) ENGINE = InnoDB AUTO_INCREMENT = 0 CHARACTER SET = 'utf8' COLLATE = 'utf8_general_ci' ROW_FORMAT = compact COMMENT '学生表';
SET FOREIGN_KEY_CHECKS = 1;

全局控制属性

通过全局属性控制,我们可以控制SqlManager的一些行为:

@ConfigurationProperties(prefix ="easy-generator.global")
@Data
@Slf4j
public class GlobalProperties {
    /**
     * sqlExecer需要执行sqlBuilder产生的sql吗? ---> 默认是不执行的
     */
    private Boolean execSql=Boolean.FALSE;
    /**
     * 需要生成模板文件吗? ---> 默认是生成的
     */
    private Boolean sqlMapper=Boolean.TRUE;
    /**
     * 是否需要生成模拟数据 ---> 默认不生成,我们需要手动开启
     */
    private Boolean mockData=Boolean.FALSE;
    /**
     * 是否将sqlBuilder生成的sql打印到文件保存 --> 默认会保存
     */
    private Boolean printSqlToFile=Boolean.TRUE;
    /**
     * sql文件输出到哪里保存呢? --> 默认是当前项目路径下
     */
    private String sqlFilePath;
    /**
     * 是否需要添加创建数据库的sql语句,如果数据库不存在则进行创建 --> 默认会添加
     */
    private Boolean createDataBase=Boolean.TRUE;
    /**
     * 是否从已经存在的表中进行信息导入 ---> 默认不开启,如果开启了从已经存在的表导入,那么将会忽略对应的JavaBean对象解析
     */
    private Boolean importFromExistTable=Boolean.FALSE;
    /**
     * 已经存在的表名,如果存在多个,用逗号分隔开来 
     */
    private String  existTableName;
    /**
     * 特性管理(后面介绍)
     */
    private Feature feature=Feature.newDefaultFeature();
}

具体每个属性什么含义,大家可以参考上面的注释,外加自己进行一番测试。

模板文件相关属性

/**
 * 最终每个文件的输出路径为: projectPath+basePackageModuleName+对应模块名+文件名
 */
@ConfigurationProperties(prefix = "easy-generator.template")
@Data
public class SqlTemplateConfig {
    /**
     * 项目路径
     */
    private String projectPath="";
    /**
     * 基础包模块: 可以不指定
     */
    private String basePackageModuleName="";
    /**
     * entity模块名: 该模块下又分为 dao,dto,vo,factory
     */
    private String entityModuleName="entity";
    /**
     * dao层下存放直接与数据库交互的模型
     */
    private String daoModuleName= ModuleUtil.merge(entityModuleName,"dao");
    /**
     * dto层下存放响应给前端的结果对象
     */
    private String dtoModuleName=ModuleUtil.merge(entityModuleName,"dto");
    /**
     * vo层下存放接收前端请求参数的对象
     */
    private String voModuleName=ModuleUtil.merge(entityModuleName,"vo");
    /**
     * factory层下存放dto,vo,dao三层转换的对象工厂
     */
    private String factoryModuleName=ModuleUtil.merge(entityModuleName,"factory");;
    /**
     * controller模块名
     */
    private String controllerModuleName="controller";
    /**
     * mapper模块名
     */
    private String mapperModuleName="mapper";
    /**
     * service模块名
     */
    private String serviceModuleName="service";
    /**
     * serviceImpl模块名
     */
    private String serviceImplModuleName=ModuleUtil.merge(serviceModuleName,"impl");
    /**
     * 是否启用默认模板配置: 会自动生成默认提供的模板代码
     */
    private Boolean defaultConfig=Boolean.TRUE;
    /**
     * 模板文件的基础目录: 相对于classPath而言
     */
    private String templateBaseDir= "template";

    public static final String DEFAULT_TEMPLATE_BASE_DIR="default-template";
}

SqlTemplateConfig主要用来控制各个模板文件的输出路径,核心公式为:

projectPath+basePackageModuleName+对应模块名+文件名

模板文件配置

默认模板文件存放地:

如果我们不想生成某个默认模板文件,可以动态移除:

sqlManager.removeDefaultTemplate(DefaultTemplateManager.CONTROLLER_TEMPLATE);

如果需要自定义模板文件也很简单,只需要三步:

编写模板文件,因为底层使用Velocity模板引擎,大家需要先去简单了解一下它的语法,即可开始使用,模板文件的编写可以参考默认的模板文件
编写对应的RenderHandler,往模板上下文中填充数据,RenderHandler返回的相对路径为: basePackageModuleName+对应模块名+文件名

public class ControllerRenderHandler implements RenderHandler<VelocityContext> {

    @Override
    public String paramPrepare(VelocityContext velocityContext, TableMapInfo tableMapInfo, SqlTemplateConfig sqlTemplateConfig, TemplateContext templateContext) {
        velocityContext.put("packagePath", sqlTemplateConfig.getControllerModuleName());
        velocityContext.put("className",tableMapInfo.getTableName());
        velocityContext.put("classNameDto",tableMapInfo.getTableName()+"Dto");
        velocityContext.put("serviceModule",sqlTemplateConfig.getServiceModuleName());
        velocityContext.put("voModule",sqlTemplateConfig.getVoModuleName());
        velocityContext.put("dtoModule",sqlTemplateConfig.getDtoModuleName());
        velocityContext.put("cacheFeature", SpringUtil.getBean(GlobalProperties.class).getFeature().getCache());
        templateContext.setVal(TemplateContext.JAVA_FILE_NAME,tableMapInfo.getTableName()+"Controller.java");

        return sqlTemplateConfig.getControllerModuleName();
    }
}

3.使用SqlManager添加自定义模板

sqlManager.addTemplate("hello.vm",new HelloRenderHandler());

4.默认会去类路径下的template目录下寻找我们的自定义模板文件,可以在yml配置进行修改

    /**
     * 模板文件的基础目录: 相对于classPath而言
     */
    private String templateBaseDir= "template";

5.我们的模板文件最终生成路径为: projectPath+RenderHandler返回的相对路径

模拟数据生成

如果我们要使用模拟数据生成的功能,我们需要配置该功能为开启状态:

easy-generator:
  global:
    mock-data: true

如何生成模拟数据,我们只使用使用两个注解提示生成器即可:

@TableMock:

/**
 * 标注在类上,指明表名,要插入的数据条数
 */
@Retention(RetentionPolicy.RUNTIME)
public @interface TableMock {
    /**
     * @return 表名,不指定默认为类名首字母小写并进行驼峰转换
     */
    String tableName() default "";
    /**
     * @return 模拟生成的数据条数
     */
    int mockNum() default 10;
}

@FieldMock

/**
 * 标注在字段上,指明模拟数据生成的规则:
 * 1.固定
 * 2.随机
 * 3.递增
 * 4.规则
 * 5.词库
 */
@Retention(RetentionPolicy.RUNTIME)
public @interface FieldMock {
    /**
     * @return 列名,默认为字段名小写并且进行驼峰到下划线的转换
     */
    String fieldName() default "";
    /**
     * @return 模拟数据生成的规则
     */
    int rule();
    /**
     * 指定额外信息: 
     * 1. 固定值,指明固定值的大小
     * 2. 随机值,指明随机类型
     * 3. 递增值,指明递增初值
     * 4. 规则值,指明正则表达式
     * 5. 词库值,指明来源于哪个词库,除非使用默认词库,否则需要指明词库文件的全路径
     */
    String extra();
}

随机值的额外信息比较丰富:

/**
 * 随机规则下的随机生成类型值选择
 */
public interface RandomRuleConstants {
    /**
     * 随机生成一个字符串
     */
     String STRING="STRING";
    /**
     * 随机生成一个整数
     */
     String INT="INT";
    /**
     * 随机生成一个小数
     */
     String FLOAT="FLOAT";
    /**
     * 随机生成一个日期
     */
     String DATE="DATE";
    /**
     * 随机生成一个时间戳
     */
     String TIMESTAMP="TIMESTAMP";
    /**
     * 随机生成一个网址
     */
     String URL="URL";
    /**
     * 随机生成一个IP
     */
     String IP="IP";
    /**
     * 随机生成一个邮箱
     */
     String EMAIL="EAMIL";
    /**
     * 随机生成一个姓名
     */
     String NAME="NAME";
    /**
     * 随机生成一个城市名
     */
     String CITY="CITY";
    /**
     * 随机生成一个大学名
     */
     String UNIVERSITY="UNIVERSITY";
    /**
     * 随机生成一个电话号码
     */
     String PHONE="PHONE";
    /**
     * 随机生成一个年龄
     */
    String AGE="AGE";
}

使用演示:

@Table(comment = "学生表")
@TableMock(mockNum = 20)
public class Stu {
    @Key(keyType = KeyTypeConstants.PRIMARY_KEY)
    @Filed(comment = "主键id", autoIncrement = true)
    private Integer id;
    @Filed(comment = "姓名")
    @FieldMock(rule = MockRuleConstants.RANDOM, extra = RandomRuleConstants.NAME)
    private String name;
    @Filed(comment = "年龄")
    @FieldMock(rule = MockRuleConstants.RANDOM, extra = RandomRuleConstants.AGE)
    private Integer age;
    @Filed(comment = "就读大学")
    @FieldMock(rule = MockRuleConstants.DICT, extra = DefaultDictConstants.UNIVERSITY_DICT)
    private String university;
    @Filed(comment = "主攻编程语言")
    @FieldMock(rule = MockRuleConstants.DICT,extra = DefaultDictConstants.LANGUAGE_DICT)
    private String language;
    @Filed(comment = "老家城市")
    @FieldMock(rule = MockRuleConstants.DICT, extra = DefaultDictConstants.CITY_DICT)
    private String homeCity;
    @Filed(comment = "电话号码")
    @FieldMock(rule = MockRuleConstants.RANDOM, extra = RandomRuleConstants.PHONE)
    private String phoneNumber;
    @Filed(comment = "qq邮箱号码")
    @FieldMock(rule = MockRuleConstants.RULE, extra = "[1-9][0-9]{8,10}\\@[q][q]\\.[c][o][m]")
    private String qqEamil;
    @Filed(comment = "出生日期")
    @FieldMock(rule = MockRuleConstants.RANDOM, extra = RandomRuleConstants.DATE)
    private LocalDateTime birthTime;
    @Filed(comment = "创建时间", defaultVal = DefaultValConstants.CURRENT_TIMESTAMP)
    private LocalDateTime createTime;
    @Filed(comment = "修改时间", defaultVal = DefaultValConstants.CURRENT_TIMESTAMP)
    private LocalDateTime updateTime;
}

测试:

@SpringBootTest
public class StuTableCreateAndMockTest {
    @Autowired
    private SqlManager sqlManager;

    @Test
    public void test(){
        sqlManager.addMockObject(new Stu());
        //如果我们不需要根据OopBuilder生成模板文件和执行建库sql,可以注释掉,前提是确保相关表已经存在
        //sqlManager.addMappedObject(new Stu());
        sqlManager.build();
    }
}

对于没有标注@FieldMock注解的字段,那么再生成对应的Insert语句时,会跳过该字段的插入:

insert into tableName(mockFiled1,mockFiled2,...) values(mockVal1,mockVal2,...);

自定义词库

如果需要引入自定义词库,需要:

    @FieldMock(rule = MockRuleConstants.DICT,extra = "词库文件的路径")
    private String language;

词库文件中词语的格式如下:

多个词语间通过英文的逗号进行分隔。

索引注意事项

通过在字段上标注@Key注解来指定索引:

    @Key(keyType = KeyTypeConstants.PRIMARY_KEY)

对于通用的索引,我们可以直接通过KeyTypeConstants完成枚举:

public interface KeyTypeConstants {
    String PRIMARY_KEY="PRIMARY KEY";
    String UNIQUE_KEY="UNIQUE";
    String KEY="KEY";
    String FOREIGN_KEY="FOREIGN KEY";
}

对于特定数据库支持的索引,例如mysql,可以根据MysqlKeyTypeConstants完成枚举:

public interface MysqlKeyTypeConstants {
    String PRIMARY_KEY="PRIMARY KEY";
    String COMPOSE_PRIMARY_KEY= "PRIMARY KEY";
    String UNIQUE_KEY="UNIQUE";
    String COMPOSE_UNIQUE_KEY_="UNIQUE";
    String KEY="KEY";
    String COMPOSE_KEY="KEY";
    String FOREIGN_KEY="FOREIGN KEY";
}

从已经存在的表完成映射,生成模板代码

如果表已经不存在了,不需要重新创建,但是此时我们需要根据已经存在的表生成对应的模板文件,那么可以进行如下配置:

easy-generator:
  template:
       project-path: C:\Users\zdh\IdeaProjects\Easy-generator\src\main\java
       base-package-module-name: com.easyCode.test
  global:
    #是否通过sqlExec执行建库语句    
    exec-sql: true
    #记录建库语句和insert插入语句到文件中保存    
    print-sql-to-file: true
    #是否开启sqlMapper生成模板文件的功能    
    sqlMapper: true
    #是否需要去解析Mock相关注解完成模拟数据的生成    
    mock-data: true
    #从已经存在的表中导入数据    
    import-from-exist-table: true
    #要映射的表名,如果存在多个表需要映射,用逗号分隔: stu,course    
    exist-table-name: stu

如果import-from-exist-table属性被设置为true,那么即使exec-sql为true,也会跳过sqlExec组件的执行,因为表已经存在了,也不会生成相关的建库语句来执行。

使用步骤

拉取项目到本地,通过maven install安装到本地仓库

2. 在个人项目中引入坐标

    <groupId>easyCode.dhy</groupId>
    <artifactId>Easy-generator</artifactId>
    <version>1.0-SNAPSHOT</version>

因为采用SpringBoot自动配置,所以个人项目只需要引入Spring相关依赖后,创建一个启动类，然后再正常写测试类，完成代码生成即可，包括模拟数据，当然还需要修改一下模板代码的输出路径。

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2023-01-19，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法

腾讯云测试服务

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

编程算法

腾讯云测试服务

登录后参与评论

0 条评论

热度