mybatis-plus code generator demo

code generator 代码生成器主程序

// 主生成程序
public class MybatisGenerator {

    public static final String url = "jdbc:mysql://localhost:3306/user?useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true";
    public static final String userName = "root";
    public static final String userPassword = "123456";

    public static final String projectPath = "C:/work/demo/mybatis-plus-demo";

    public static void main(String[] args) {


        DataSourceConfig dataSourceConfig =
                new DataSourceConfig.Builder(url, userName, userPassword).build();

        AutoGenerator generator = new AutoGenerator(dataSourceConfig);


        GlobalConfig globalConfig = new GlobalConfig.Builder().openDir(false).
                fileOverride().
                outputDir(projectPath + "/src/main/java").
                author("your name").
                build();

        generator.global(globalConfig);

      // 各个文件生成位置
        PackageConfig packageConfig = new PackageConfig.Builder().
                parent("com.org.example").
                service("manager").
                serviceImpl("manager.impl").
                entity("domain.entity").
                mapper("dao.mapper").
                build();
        generator.packageInfo(packageConfig);

        // 策略，指定生成表等基本信息
        StrategyConfig strategy = new StrategyConfig.Builder().
                addInclude(new String[]{"user_address"}).
                build();
        
        // 设置service 的名字模板
        Service.Builder serviceBuilder = strategy.serviceBuilder();
        serviceBuilder.formatServiceFileName("%sManager").formatServiceImplFileName("%sManagerImpl");
        
        // 设置生成的entity 模板
        Entity.Builder entityBuilder = strategy.entityBuilder();
        entityBuilder.enableLombok().columnNaming(underline_to_camel).
                naming(underline_to_camel);

        
        generator.strategy(strategy);
        
        // 基本模板配置
        TemplateConfig templateConfig = new TemplateConfig.Builder().
                disable(TemplateType.XML, TemplateType.CONTROLLER).
                build();
        generator.template(templateConfig);

        // 执行
        generator.execute();

    }
}

pom 配置文件

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.5.3</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>

    <groupId>org.example</groupId>
    <artifactId>mybatis-plus-demo</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.4.3.1</version>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.5.0</version>
        </dependency>
        <dependency>
            <groupId>org.apache.velocity</groupId>
            <artifactId>velocity-engine-core</artifactId>
            <version>2.3</version>
        </dependency>
        
        <!--驱动-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.26</version>
        </dependency>
        
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.2.6</version>
        </dependency>
        
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.20</version>
            <scope>provided</scope>
        </dependency>
        
        <!--通用工具，json转换等-->
        <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-all</artifactId>
            <version>5.7.7</version>
        </dependency>


    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>
</project>

配置文件

spring:
  datasource:
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://localhost:3306/user?useUnicode=true&characterEncoding=utf-8&useSSL=false&allowMultiQueries=true
    username: root
    password: 123456

mybatis-plus:
# 对枚举类的处理，枚举如果没有实现指定的接口 IEnum<Integer>的话，需要专门的进行配置扫描包
  typeEnumsPackage: com.org.example.domain.type

logging:
  level:
    com.org.example.dao.mapper: debug

生成的的实体类,需要手动的将需要进行转换的属性进行属性的变更以及typeHandler的选取，这个类很重要，之后需要单独维护，整个转换器再这个上边进行使用

package com.org.example.domain.entity;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import java.io.Serializable;
import java.time.Instant;
import java.util.List;

import com.baomidou.mybatisplus.annotation.TableName;
import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler;
import com.org.example.dao.typehandler.InstantMsLongTypeHandler;
import com.org.example.dao.typehandler.LiveAttrListTypeHandler;
import com.org.example.dao.typehandler.LiveChannelListTypeHandler;
import com.org.example.domain.type.PushStreamType;
import com.org.example.domain.type.StatusType;
import lombok.Data;
import lombok.EqualsAndHashCode;

@Data
@TableName(autoResultMap = true)
@EqualsAndHashCode(callSuper = false)
public class PlConfigProject implements Serializable {



    @TableId(value = "id", type = IdType.AUTO)
    private Long id;

    @TableField("project_id")
    private String projectId;

    @TableField(value = "live_channel_info", typeHandler = LiveChannelListTypeHandler.class)
    private List<LiveChannelInfo> liveChannelInfo;

    @TableField(value = "cloud_store_info", typeHandler = JacksonTypeHandler.class)
    private CloudStoreInfo cloudStoreInfo;

    /**
     * 直播初始化属性包括 视频属性，音频属性，录屏属性等
     */
    @TableField(value = "live_attr", typeHandler = LiveAttrListTypeHandler.class)
    private List<LiveAttr> liveAttr;

    /**
     * 状态属性 1 未激活 2激活  3禁用 4删除
     */
    private StatusType status;

    @TableField("push_stream_type")
    private PushStreamType pushStreamType;

    @TableField(value = "create_time", typeHandler = InstantMsLongTypeHandler.class)
    private Instant createTime;

    @TableField(value = "update_time", typeHandler = InstantMsLongTypeHandler.class)
    private Instant updateTime;

    private Integer version;


}

枚举类的创建

实现mybatis-plus的接口，则不需要再配置文件中配置相应的路径，但是需要重写getValue方法
package com.org.example.domain.type;

import com.baomidou.mybatisplus.annotation.EnumValue;
import com.baomidou.mybatisplus.annotation.IEnum;


public enum StatusType implements IEnum<Integer> {
    //状态属性 1 未激活 2激活  3已使用 4删除
    STATUS_INACTIVATED(1, "未激活"),
    STATUS_ACTIVATED(2, "激活"),
    STATUS_USED(3, "已使用"),
    STATUS_DELETED(4, "删除"),
    ;

    @EnumValue
    private int code;

    private String description;

    StatusType(int code, String description) {
        this.code = code;
        this.description = description;
    }

    @Override
    public Integer getValue() {
        return this.code;
    }

    public String getCodeString() {
        return name();
    }

    public String getDescription() {
        return description;
    }


}
如果自定义枚举的话，需要再配置文件中配置文件的路径，进行扫描mybatis-plus.

typeEnumsPackage=com.org.example.domain.type

public enum PushStreamType implements Enumerable {

    RTMP(0, "rtmp推流"),
    RTC(1, "rtc推流"),
    RTMP_RTC(2, "rtc & rtmp 推流"),
    ;

    @EnumValue
    private int code;

    private String description;

    PushStreamType(int code, String description) {
        this.code = code;
        this.description = description;
    }

    @Override
    public int getCode() {
        return code;
    }

    @Override
    public String getCodeString() {
        return name();
    }

    @Override
    public String getDescription() {
        return description;
    }
}


public interface Enumerable {

    int getCode();

    String getCodeString();

    String getDescription();
}

使用的typeHandler定义

// instant 2 bigint的转换
package com.org.example.dao.typehandler;


import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.time.Instant;

public class InstantMsLongTypeHandler extends BaseTypeHandler<Instant> {
    public InstantMsLongTypeHandler() {
    }

    public void setNonNullParameter(PreparedStatement ps, int i, Instant parameter, JdbcType jdbcType) throws SQLException {
        ps.setLong(i, parameter.toEpochMilli());
    }

    public Instant getNullableResult(ResultSet rs, String columnName) throws SQLException {
        long value = rs.getLong(columnName);
        if (rs.wasNull()) {
            return null;
        } else {
            return value > 0L ? Instant.ofEpochMilli(value) : null;
        }
    }

    public Instant getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        long value = rs.getLong(columnIndex);
        if (rs.wasNull()) {
            return null;
        } else {
            return value > 0L ? Instant.ofEpochMilli(value) : null;
        }
    }

    public Instant getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        long value = cs.getLong(columnIndex);
        if (cs.wasNull()) {
            return null;
        } else {
            return value > 0L ? Instant.ofEpochMilli(value) : null;
        }
    }
}


// instant 2 timestamp 的转换器
public class InstantTimestampTypeHandler extends BaseTypeHandler<Instant> {

    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, Instant parameter, JdbcType jdbcType) throws SQLException {
        ps.setTimestamp(i, new Timestamp(parameter.toEpochMilli()));
    }

    @Override
    public Instant getNullableResult(ResultSet rs, String columnName)
            throws SQLException {
        Timestamp value = rs.getTimestamp(columnName);
        if (rs.wasNull()) {
            return null;
        } else {
            return Instant.ofEpochMilli(value.getTime());
        }
    }

    @Override
    public Instant getNullableResult(ResultSet rs, int columnIndex)
            throws SQLException {
        Timestamp value = rs.getTimestamp(columnIndex);
        if (rs.wasNull()) {
            return null;
        } else {
            return Instant.ofEpochMilli(value.getTime());
        }
    }

    @Override
    public Instant getNullableResult(CallableStatement cs, int columnIndex)
            throws SQLException {
        Timestamp value = cs.getTimestamp(columnIndex);
        if (cs.wasNull()) {
            return null;
        } else {
            return Instant.ofEpochMilli(value.getTime());
        }
    }
}

// 其他复杂类型转换器

public class LiveAttrListTypeHandler extends BaseTypeHandler<List<LiveAttr>> {

    @Override
    public void setNonNullParameter(PreparedStatement ps, int i,
                                    List<LiveAttr> parameter, JdbcType jdbcType) throws SQLException {
        ps.setString(i, toValue(parameter));
    }

    @Override
    public List<LiveAttr> getNullableResult(ResultSet rs, String columnName)
            throws SQLException {
        String value = rs.getString(columnName);
        if (rs.wasNull()) {
            return null;
        } else {
            return convert(value);
        }
    }

    @Override
    public List<LiveAttr> getNullableResult(ResultSet rs, int columnIndex)
            throws SQLException {
        String value = rs.getString(columnIndex);
        if (rs.wasNull()) {
            return null;
        } else {
            return convert(value);
        }
    }

    @Override
    public List<LiveAttr> getNullableResult(CallableStatement cs, int columnIndex)
            throws SQLException {
        String value = cs.getString(columnIndex);
        if (cs.wasNull()) {
            return null;
        } else {
            return convert(value);
        }
    }

    private List<LiveAttr> convert(String value) {
        return JSONUtil.toBean(value, new TypeReference<List<LiveAttr>>() {
        }, true);
    }

    private String toValue(List<LiveAttr> parameter) {

        return JSONUtil.toJsonStr(parameter);
    }
}
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 218,386评论 6赞 506
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 93,142评论 3赞 394
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 164,704评论 0赞 353
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 58,702评论 1赞 294
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 67,716评论 6赞 392
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 51,573评论 1赞 305
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 40,314评论 3赞 418
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 39,230评论 0赞 276
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 45,680评论 1赞 314
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,873评论 3赞 336
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,991评论 1赞 348
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 35,706评论 5赞 346
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 41,329评论 3赞 330
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,910评论 0赞 22
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,038评论 1赞 270
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 48,158评论 3赞 370
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 44,941评论 2赞 355