MyBatis-Plus操作数据库
导入依赖
<!-- 启动器-->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.0.5</version>
</dependency>
<!-- velocity模板引擎 和下面的二选一-->
<dependency>
<groupId>org.apache.velocity</groupId>
<artifactId>velocity-engine-core</artifactId>
<version>2.0</version>
</dependency>
<!-- freemarker 模板引擎 -->
<dependency>
<groupId>org.freemarker</groupId>
<artifactId>freemarker</artifactId>
<version>2.3.21</version>
</dependency>
<!-- MySQL驱动-->
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>5.1.4</version>
</dependency>
<!-- 数据源-->
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>druid</artifactId>
<version>1.1.8</version>
</dependency>
<!-- web场景启动器-->
<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>
创建CodeGenerator
我的结构
package com.llz.gujian;
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.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;
public class CodeGenerator {
public static void main(String[] args) {
/*
* 代码生成器 需要多个配置
* 1.全局配置 GolbalConfig
* 2.数据源配置 DataSourceConfig
* 3.包路径配置 PackageConfig
* 4.自定义配置 InjectionConfig 根据需要自己配置
* 需要一个 FileOutConfig
* 5.模板配置 TemplateConfig 注意自己使用的模板引擎
* 6.策略配置 strategyConfig
*
* 使用需要修改的配置 全局路径 数据源 包路径 策略配置的表配置
*/
AutoGenerator mpg = new AutoGenerator();
/*
* 1.全局配置
*/
GlobalConfig gc = new GlobalConfig();
// 模块路径
String projectPath = "你的路径";
gc.setOutputDir(projectPath + "/src/main/java");
//作者名
gc.setAuthor("XXX");
gc.setOpen(false);
// gc.setSwagger2(true); 实体属性 Swagger2 注解
mpg.setGlobalConfig(gc);
/*
* 2.数据源配置
*/
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl("jdbc:mysql://localhost:3306/你的数据库?allowMultiQueries=true&useUnicode=true&characterEncoding=UTF-8&useSSL=false");
dsc.setDriverName("com.mysql.jdbc.Driver");
dsc.setUsername("用户名");
dsc.setPassword("密码");
mpg.setDataSource(dsc);
/*
* 3.包配置
*/
PackageConfig pc = new PackageConfig();
//自定义模块名称 在下面mapper路径中可以使用 gc.getMoudleName()获取该模块名 根据需求,酌情使用
//pc.setModuleName("customer");
pc.setParent("com.llz.gujian");
// 路径生成结果 com.llz.web.customer
mpg.setPackageInfo(pc);
/*
*4.自定义配置
*/
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/" // + gc.getMoudleName()
+ "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
}
});
/*
cfg.setFileCreate(new IFileCreate() {
@Override
public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
// 判断自定义文件夹是否需要创建
checkDir("调用默认方法创建的目录,自定义目录用");
if (fileType == FileType.MAPPER) {
// 已经生成 mapper 文件判断存在,不想重新生成返回 false
return !new File(filePath).exists();
}
// 允许生成模板文件
return true;
}
});
*/
cfg.setFileOutConfigList(focList);
mpg.setCfg(cfg);
/*
* 5.配置模板
*/
TemplateConfig templateConfig = new TemplateConfig();
// 配置自定义输出模板
//指定自定义模板路径,注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
// templateConfig.setEntity("templates/entity2.java");
// templateConfig.setService();
// templateConfig.setController();
templateConfig.setXml(null);
mpg.setTemplate(templateConfig);
/*
* 6.策略配置
*/
StrategyConfig strategy = new StrategyConfig();
strategy.setNaming(NamingStrategy.underline_to_camel);
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
// strategy.setSuperControllerClass("Controller的父类实体");
// strategy.setSuperEntityClass("Entity的父类实体,没有就不用设置!");
strategy.setEntityLombokModel(true);
strategy.setRestControllerStyle(true);
// 公共父类
// strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
// 写于父类中的公共字段
/*strategy.setSuperEntityColumns("id");*/
// 指定需要生成代码的表 默认是全部(注释掉)
strategy.setInclude("x_customer");
strategy.setControllerMappingHyphenStyle(true);
// 忽略表的前缀
strategy.setTablePrefix("x_");
mpg.setStrategy(strategy);
mpg.setTemplateEngine(new FreemarkerTemplateEngine());
mpg.execute();
}
}
执行代码生成器类生成代码
我的结构
配置数据库连接
我喜欢使用yml文件 所以创建了application.yml
spring:
datasource:
username: 用户名
password: 密码
driver-class-name: com.mysql.jdbc.Driver
url: jdbc:mysql://localhost:3306/你的数据库?serverTimezone=GMT%2B8
type: com.alibaba.druid.pool.DruidDataSource
小测一下
- 在测试类中注入Mapper
- 当然在这之前我们有一个很重要的步骤为mapper添加注解,非常重要
@Mapper // 一定要加!!!
public interface DictMapper extends BaseMapper<Dict> {
}
- 运行我们的代码
@Autowired
DictMapper dictMapper;
@Test
void contextLoads() {
List<Dict> dicts = dictMapper.selectList(null);
System.out.println(dicts.size());
}
- 大功告成 嘻嘻!!!