SpringBoot整合Mybatis-Plus和代码生成器

SpringBoot整合Mybatis-Plus和代码生成器

使用Mybatis-Plus代码生成类自动生成Entity,Mapper,Service,ServiceImpl,Controller,MapperXml。

Maven依赖

 <!--Mybatis Plus-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.4.1</version>
        </dependency>
        <!--Mybatis Plus 代码生成器-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.4.1</version>
        </dependency>
        <!--代码生成器所依赖模板-->
        <dependency>
            <groupId>org.freemarker</groupId>
            <artifactId>freemarker</artifactId>
            <version>2.3.30</version>
        </dependency>

SpringBoot的maven依赖

 <!--SpringBoot-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <!--Lombok插件-->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
        <!--SpringBoot-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
           <!--thymeleaf-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-thymeleaf</artifactId>
        </dependency>
           <!--mysql-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.13</version>
        </dependency>

Mybatis-Plus的YML文件

spring:
  # 配置数据源
  datasource:
    type: com.mysql.cj.jdbc.MysqlConnectionPoolDataSource
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://localhost:3306/数据库名?useUnicode=true&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=UTC
    username: root
    password: 1234
#配置Mybatis Plus
mybatis-plus:
  mapper-locations: classpath:mapper/*.xml
  configuration:
    # 是否开启自动驼峰命名规则映射:从数据库列名到Java属性驼峰命名的类似映射
    map-underscore-to-camel-case: true
    # 如果查询结果中包含空值的列，则 MyBatis 在映射的时候，不会映射这个字段
    call-setters-on-nulls: false
    # 这个配置会将执行的sql打印出来，在开发或测试的时候可以用
    log-impl: org.apache.ibatis.logging.stdout.StdOutImpl

Mybatis-Plus代码生成类

package com.studies;


import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

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.isNotBlank(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("linlin");
				  //取消弹框
				  gc.setOpen(false);
				  // gc.setSwagger2(true); 实体属性 Swagger2 注解
				  mpg.setGlobalConfig(gc);
				  // 数据源配置
				  DataSourceConfig dsc = new DataSourceConfig();
				  dsc.setUrl("jdbc:mysql://localhost:3306/数据库名?useUnicode=true&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=UTC");
				  dsc.setDriverName("com.mysql.cj.jdbc.Driver");
				  dsc.setUsername("root");
				  dsc.setPassword("1234");
				  mpg.setDataSource(dsc);

				  // 包配置
				  PackageConfig pc = new PackageConfig();
//				  设置模块名 会生成一个新文件夹 在新文件夹里面创建 domain mapper service  controller
				  pc.setModuleName(null);
				  pc.setParent("com.studies");
				  mpg.setPackageInfo(pc);

				  // 自定义配置
				  InjectionConfig cfg = new InjectionConfig() {
						@Override
						public void initMap() {
							  // to do nothing
						}
				  };

				  // 如果模板引擎是 freemarker
				  String templatePath = "/templates/mapper.xml.ftl";
				  // 如果模板引擎是 velocity
				  // String templatePath = "/templates/mapper.xml.vm";

				  // 自定义输出配置
				  List<FileOutConfig> 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.setNaming(NamingStrategy.underline_to_camel);
				  strategy.setColumnNaming(NamingStrategy.underline_to_camel);
//				  strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
				  strategy.setEntityLombokModel(true);
				  strategy.setRestControllerStyle(true);
				  // 公共父类
//				  strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
				  // 写于父类中的公共字段  如果有这个表示不生成主键id
//				  strategy.setSuperEntityColumns("id");
				  strategy.setInclude(scanner("表名，多个英文逗号分割").split(","));
				  strategy.setControllerMappingHyphenStyle(true);
				  strategy.setTablePrefix(pc.getModuleName() + "_");
				  mpg.setStrategy(strategy);
				  mpg.setTemplateEngine(new FreemarkerTemplateEngine());
				  mpg.execute();
			}
}

Mybatis-Plus的分页配置类Config

package com.studies.config;

import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
import com.baomidou.mybatisplus.extension.plugins.pagination.optimize.JsqlParserCountOptimize;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;


@Configuration
public class MybatisPlusConfig {

	  /**
	   * 分页插件
	   * @return
	   */
	  @Bean
	  public PaginationInterceptor paginationInterceptor() {
			PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
			// 设置请求的页面大于最大页后操作， true调回到首页，false 继续请求  默认false
			// paginationInterceptor.setOverflow(false);
			// 设置最大单页限制数量，默认 500 条，-1 不受限制
			// paginationInterceptor.setLimit(500);
			// 开启 count 的 join 优化,只针对部分 left join
			paginationInterceptor.setCountSqlParser(new JsqlParserCountOptimize(true));
			return paginationInterceptor;
	  }
}

Controller测试

package com.studies.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.studies.entity.ShiroUser;
import com.studies.service.impl.ShiroUserServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.Map;


@RestController
@RequestMapping("/User")
public class UserController {

	  @Autowired
	  private ShiroUserServiceImpl shiroUserService;

	  /**
	   * 根据账号查询用户信息
	   * @param user
	   * @return
	   */
	  @RequestMapping("/login")
	  public Map<String,Object> login(@RequestBody ShiroUser user){
			Map<String, Object> map = new HashMap<String, Object>();
			ShiroUser shiroUser = shiroUserService.selectByName(user);
			if(shiroUser!=null){
				  map.put("code",200);
				  map.put("data",shiroUser);
				  return  map;
			}
			map.put("code",200);
			map.put("data","");
			return  map;
	  }

	  @RequestMapping("/getInfoListPage")
	  public Map<String,Object> loginOut(@RequestBody ShiroUser user){
			Map<String, Object> map = new HashMap<String, Object>();
			IPage<ShiroUser> page = new Page<>();
			page.setCurrent(5); //当前页
			page.setSize(1);    //每页条数
			page = shiroUserService.page(page);
			map.put("code",200);
			map.put("data",page);
			return  map;
	  }


//	  @RequestMapping("/loginOut")
//	  public Map<String,Object> loginOut(@RequestBody Test test){
//			Map<String, Object> map = new HashMap<String, Object>();
//			System.out.println("name============"+test.getName());
//			System.out.println("password========="+test.getPassword());
//			map.put("code",200);
//			map.put("data",test);
//			return  map;
//	  }

}

如果启动报错记得在启动类加上
@MapperScan(“com.studies.mapper”)
或者Mapper里面加上@Mapper
最后使用PostMan测试工具测试就完成啦。