博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
SpringBoot 之整合MybatisPlus+代码生成器
阅读量:2456 次
发布时间:2019-05-10

本文共 11392 字,大约阅读时间需要 37 分钟。

文章目录

导入依赖

org.projectlombok
lombok
true
com.baomidou
mybatis-plus-boot-starter
3.2.0
com.baomidou
mybatis-plus-generator
3.2.0
org.apache.velocity
velocity-engine-core
2.1
org.freemarker
freemarker
2.3.28
mysql
mysql-connector-java
5.1.42
runtime

Application

import org.mybatis.spring.annotation.MapperScan;import org.springframework.boot.SpringApplication;import org.springframework.boot.autoconfigure.SpringBootApplication;@SpringBootApplication@MapperScan("com.demo.user.mapper")public class DemoApplication {
public static void main(String[] args) {
SpringApplication.run(DemoApplication.class, args); }}

application.yml

spring:  datasource:    driver-class-name: com.mysql.jdbc.Driver    url: jdbc:mysql://localhost:3306/test?useUnicode=true&useSSL=false&characterEncoding=utf8    username: root    password: 123456mybatis-plus:  type-aliases-package: com.demo.user.entity  configuration:    map-underscore-to-camel-case: true    use-generated-keys: true  mapper-locations: classpath:/mapper/*Mapper.xmllogging:  level:    com.demo.mapper: debug

MyBatisPlusConfig

import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;@Configurationpublic class MyBatisPlusConfig {
/** * mybatis-plus 分页插件 */ @Bean public PaginationInterceptor paginationInterceptor() {
PaginationInterceptor paginationInterceptor = new PaginationInterceptor(); return paginationInterceptor; }}

SQL脚本

CREATE TABLE `tb_user`  (  `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '主键ID',  `user_name` varchar(30) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '姓名',  `age` int(11) NULL DEFAULT NULL COMMENT '年龄',  `userEmail` varchar(50) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '邮箱',  PRIMARY KEY (`id`) USING BTREE) ENGINE = InnoDB AUTO_INCREMENT = 12 CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic;INSERT INTO `tb_user` VALUES (1, 'Jone', 18, 'test1@baomidou.com');INSERT INTO `tb_user` VALUES (2, 'Jack', 20, 'test2@baomidou.com');INSERT INTO `tb_user` VALUES (3, 'Tom', 28, 'test3@baomidou.com');INSERT INTO `tb_user` VALUES (4, 'Sandy', 21, 'test4@baomidou.com');INSERT INTO `tb_user` VALUES (5, 'Billie', 24, 'test5@baomidou.com');

代码生成器CodeGenerator.java

package com.demo;import com.baomidou.mybatisplus.annotation.DbType;import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;import com.baomidou.mybatisplus.core.toolkit.StringPool;import com.baomidou.mybatisplus.generator.AutoGenerator;import com.baomidou.mybatisplus.generator.InjectionConfig;import com.baomidou.mybatisplus.generator.config.*;import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;import com.baomidou.mybatisplus.generator.config.po.TableInfo;import com.baomidou.mybatisplus.generator.config.rules.IColumnType;import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;import org.apache.commons.lang3.StringUtils;import java.util.ArrayList;import java.util.List;import java.util.Scanner;// 执行 main 方法控制台输入模块表名回车自动生成对应项目目录中public class CodeGenerator {
/** *

* 读取控制台内容 *

*/ public static String scanner(String tip) {
Scanner scanner = new Scanner(System.in); StringBuilder help = new StringBuilder(); help.append("请输入" + tip + ":"); System.out.println(help.toString()); if (scanner.hasNext()) {
String ipt = scanner.next(); if (StringUtils.isNotEmpty(ipt)) {
return ipt; } } throw new MybatisPlusException("请输入正确的" + tip + "!"); } public static void main(String[] args) {
// 代码生成器 AutoGenerator mpg = new AutoGenerator(); // 全局配置 GlobalConfig gc = new GlobalConfig(); String projectPath = System.getProperty("user.dir"); gc.setOutputDir(projectPath + "/src/main/java"); gc.setAuthor("yq"); // 是否打开输出目录,默认true gc.setOpen(false); // 开启 ActiveRecord 模式,默认false gc.setActiveRecord(true); // 是否覆盖已有文件 gc.setFileOverride(true); // XML 开启 BaseResultMap gc.setBaseResultMap(true); // XML 开启 baseColumnList gc.setBaseColumnList(true); // 自定义文件命名,注意 %s 会自动填充表实体属性! // gc.setMapperName("%sDao"); // gc.setXmlName("%sMapper"); gc.setServiceName("%sService"); // gc.setServiceImplName("%sServiceDiy"); // gc.setControllerName("%sAction"); mpg.setGlobalConfig(gc); // 数据源配置 DataSourceConfig dsc = new DataSourceConfig(); dsc.setDbType(DbType.MYSQL); dsc.setUrl("jdbc:mysql://localhost:3306/test?useUnicode=true&useSSL=false&characterEncoding=utf8"); dsc.setDriverName("com.mysql.jdbc.Driver"); dsc.setUsername("root"); dsc.setPassword("123456"); dsc.setTypeConvert(new MySqlTypeConvert(){
// 自定义数据库表字段类型转换【可选】 @Override public IColumnType processTypeConvert(GlobalConfig globalConfig, String fieldType) {
System.out.println("转换类型:" + fieldType); // 注意!!processTypeConvert 存在默认类型转换,如果不是你要的效果请自定义返回、非如下直接返回。 return super.processTypeConvert(globalConfig, fieldType); } }); mpg.setDataSource(dsc); // 包配置 PackageConfig pc = new PackageConfig(); // com.demo.user pc.setParent("com.demo");// pc.setModuleName("user"); pc.setModuleName(scanner("模块名")); mpg.setPackageInfo(pc); // 自定义配置 InjectionConfig cfg = new InjectionConfig() {
@Override public void initMap() {
// TODO } }; mpg.setCfg(cfg); // 如果模板引擎是 freemarker String templatePath = "/templates/mapper.xml.ftl"; // 自定义输出配置 List
focList = new ArrayList<>(); // 自定义配置会被优先输出 focList.add(new FileOutConfig(templatePath) {
@Override public String outputFile(TableInfo tableInfo) {
// 自定义输出文件名 , 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化!! return projectPath + "/src/main/resources/mapper/" + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML; } }); cfg.setFileOutConfigList(focList); mpg.setCfg(cfg); // 配置模板 TemplateConfig templateConfig = new TemplateConfig(); templateConfig.setXml(null); mpg.setTemplate(templateConfig); // 策略配置 StrategyConfig strategy = new StrategyConfig(); strategy.setEntityLombokModel(true); strategy.setRestControllerStyle(true); // 此处可以修改为您的表前缀 strategy.setTablePrefix(new String[] {
"tb_", "tsys_" }); // 表名生成策略 strategy.setNaming(NamingStrategy.underline_to_camel); // 需要生成的表 strategy.setInclude(scanner("表名,多个英文逗号分割").split(","));// strategy.setInclude(new String[] { "tb_user" }); strategy.setEntityBuilderModel(true); mpg.setStrategy(strategy); mpg.setTemplateEngine(new FreemarkerTemplateEngine()); mpg.execute(); }}

执行代码生成器类

在这里插入图片描述

生成的代码

为了排版,已将多余的注释去掉。

User.java

import com.baomidou.mybatisplus.annotation.TableName;import com.baomidou.mybatisplus.annotation.IdType;import com.baomidou.mybatisplus.extension.activerecord.Model;import com.baomidou.mybatisplus.annotation.TableId;import com.baomidou.mybatisplus.annotation.TableField;import java.io.Serializable;import lombok.Data;import lombok.EqualsAndHashCode;import lombok.experimental.Accessors;@Data@EqualsAndHashCode(callSuper = false)@Accessors(chain = true)@TableName("tb_user")public class User extends Model
{
private static final long serialVersionUID = 1L; @TableId(value = "id", type = IdType.AUTO) private Long id; private String userName; private Integer age; @TableField("userEmail") private String userEmail; @Override protected Serializable pkVal() {
return this.id; }}

UserMapper.java

import com.demo.user.entity.User;import com.baomidou.mybatisplus.core.mapper.BaseMapper;public interface UserMapper extends BaseMapper
{
}

UserMapper.xml

id, user_name, age, userEmail

UserService.java

import com.demo.user.entity.User;import com.baomidou.mybatisplus.extension.service.IService;public interface UserService extends IService
{
}

UserServiceImpl.java

import com.demo.user.entity.User;import com.demo.user.mapper.UserMapper;import com.demo.user.service.UserService;import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;import org.springframework.stereotype.Service;@Servicepublic class UserServiceImpl extends ServiceImpl
implements UserService {
}

UserController.java

import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.RestController;@RestController@RequestMapping("/user/user")public class UserController {
}

单元测试

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;import com.baomidou.mybatisplus.core.metadata.IPage;import com.baomidou.mybatisplus.extension.plugins.pagination.Page;import com.demo.user.entity.User;import com.demo.user.service.UserService;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.util.List;@RunWith(SpringRunner.class)@SpringBootTestpublic class DemoTests {
@Autowired UserService userService; @Test public void selectAll() {
List
users = userService.list(); users.forEach(System.out::println); } @Test public void selectById() {
User user = userService.getById(2); System.out.println(user); } @Test public void selectByPage() {
Page
page = new Page<>(2,3); IPage
pageResult = userService.page(page); List
records = pageResult.getRecords(); records.forEach(System.out::println); System.out.println("---------------黄金分割线---------------"); IPage
pageResultAge = userService.page(page, new QueryWrapper
().eq("age",18)); List
recordsAge = pageResultAge.getRecords(); recordsAge.forEach(System.out::println); } @Test public void insertPlus() { User user = new User().setUserName("James").setAge(36).setUserEmail("456465@sql.com"); userService.save(user); System.out.println(user); } @Test public void update() { User user = new User().setUserName("James").setAge(50).setId(12L); userService.updateById(user); }}

参考:

转载地址:http://ygdhb.baihongyu.com/

你可能感兴趣的文章
angular 命令行项目_Angular命令行界面介绍
查看>>
react 监听组合键_投资组合中需要的5个React项目
查看>>
css 选择器 伪元素_CSS伪元素-解释选择器之前和之后
查看>>
机器学习数据拆分_解释了关键的机器学习概念-数据集拆分和随机森林
查看>>
snapd_snapd使管理Nextcloud变得轻而易举
查看>>
devops和docker_通过免费的2小时Docker课程学习DevOps基础知识
查看>>
python构建json_如何使用Python构建JSON API
查看>>
桌面应用程序 azure_如何开始使用Microsoft Azure-功能应用程序,HTTP触发器和事件队列...
查看>>
矩阵奇异值分解特征值分解_推荐系统中的奇异值分解与矩阵分解
查看>>
异步JavaScript的演变:从回调到承诺,再到异步/等待
查看>>
自己写一个微型数据库_“最国际化的微型机构:”两名伦敦训练营的毕业生如何建造了一个远程…...
查看>>
构建静态服务器_为静态网站构建无服务器联系表
查看>>
塞尔达传说顺序_编码《塞尔达传说》克隆图例
查看>>
spring vertx_如何在Spring设置Vertx
查看>>
在Unity中创建3D直升机游戏
查看>>
编码和编码格式一样吗?_学习如何像专业人士一样编码
查看>>
CSS-in-JS的权衡
查看>>
ssh框架实现数据库_自顶向下介绍SSH及其如何实现安全的数据共享
查看>>
测试驱动开发 测试前移_测试驱动开发简介
查看>>
css 网格布局_我从CSS网格布局中学到的东西
查看>>