SpringBoot | 第九章：Mybatis-plus的集成和使用

MysqlGenerator，此类较长,相关配置可根据实际情况信息修改替换。

public class MysqlGenerator {

    private static final String PACKAGE_NAME = "cn.lqdev.learning.springboot.chapter9";
    private static final String MODULE_NAME = "biz";
    private static final String OUT_PATH = "D:\develop\code";
    private static final String AUTHOR = "oKong";

    private static final String DRIVER = "com.mysql.jdbc.Driver";
    private static final String URL = "jdbc:mysql://127.0.0.1:3306/learning?useUnicode=true&characterEncoding=UTF-8";
    private static final String USER_NAME = "root";
    private static final String PASSWORD = "123456";

    
    public static void main(String[] args) {
 // 自定义需要填充的字段
 List tableFillList = new ArrayList();

 // 代码生成器
 AutoGenerator mpg = new AutoGenerator().setGlobalConfig(
  // 全局配置
  new GlobalConfig().setOutputDir(OUT_PATH)// 输出目录
   .setFileOverride(true)// 是否覆盖文件
   .setActiveRecord(true)// 开启 activeRecord 模式
   .setEnableCache(false)// XML 二级缓存
   .setbaseResultMap(false)// XML ResultMap
   .setbaseColumnList(true)// XML columList
   .setAuthor(AUTHOR)
   // 自定义文件命名，注意 %s 会自动填充表实体属性！
   .setXmlName("%sMapper").setMapperName("%sDao")
 // .setServiceName("MP%sService")
 // .setServiceImplName("%sServiceDiy")
 // .setControllerName("%sAction")
 ).setDataSource(
  // 数据源配置
  new DataSourceConfig().setDbType(DbType.MYSQL)// 数据库类型
   .setTypeConvert(new MySqlTypeConvert() {
// 自定义数据库表字段类型转换【可选】
@Override
public DbColumnType processTypeConvert(String fieldType) {
    System.out.println("转换类型：" + fieldType);
    // if ( fieldType.toLowerCase().contains( "tinyint" ) ) {
    // return DbColumnType.BOOLEAN;
    // }
    return super.processTypeConvert(fieldType);
}
   }).setDriverName(DRIVER).setUsername(USER_NAME).setPassword(PASSWORD).setUrl(URL))
  .setStrategy(
   // 策略配置
   new StrategyConfig()
    // .setCapitalMode(true)// 全局大写命名
    .setDbColumnUnderline(true)// 全局下划线命名
    // .setTablePrefix(new String[]{"demo_"})// 此处可以修改为您的表前缀
    .setNaming(NamingStrategy.underline_to_camel)// 表名生成策略
    // .setInclude(new String[] {"user_org"}) // 需要生成的表
    // .setExclude(new String[]{"test"}) // 排除生成的表
    // 自定义实体，公共字段
    // .setSuperEntityColumns(new String[]{"test_id"})
    .setTableFillList(tableFillList)
    // 自定义实体父类
    // .setSuperEntityClass("com.baomidou.demo.common.base.BsbaseEntity")
    // // 自定义 mapper 父类
    // .setSuperMapperClass("com.baomidou.demo.common.base.BsbaseMapper")
    // // 自定义 service 父类
    // .setSuperServiceClass("com.baomidou.demo.common.base.BsbaseService")
    // // 自定义 service 实现类父类
    // .setSuperServiceImplClass("com.baomidou.demo.common.base.BsbaseServiceImpl")
    // 自定义 controller 父类
    // .setSuperControllerClass("com.baomidou.demo.TestController")
    // 【实体】是否生成字段常量（默认 false）
    // public static final String ID = "test_id";
    .setEntityColumnConstant(true)
    // 【实体】是否为构建者模型（默认 false）
    // public User setName(String name) {this.name = name; return this;}
    .setEntityBuilderModel(true)
    // 【实体】是否为lombok模型（默认 false）document
    .setEntityLombokModel(true)
  // Boolean类型字段是否移除is前缀处理
  // .setEntityBooleanColumnRemoveIsPrefix(true)
  // .setRestControllerStyle(true)
  // .setControllerMappingHyphenStyle(true)
  ).setPackageInfo(
   // 包配置
   new PackageConfig().setModuleName(MODULE_NAME).setParent(PACKAGE_NAME)// 自定义包路径
    .setController("controller")// 这里是控制器包名，默认 web
    .setXml("mapper").setMapper("dao")

  ).setCfg(
   // 注入自定义配置，可以在 VM 中使用 cfg.abc 设置的值
   new InjectionConfig() {
@Override
public void initMap() {
    Map map = new HashMap();
    map.put("abc", this.getConfig().getGlobalConfig().getAuthor() + "-mp");
    this.setMap(map);
}
   }.setFileOutConfigList(
    Collections.singletonList(new FileOutConfig("/templates/mapper.xml.vm") {
 // 自定义输出文件目录
 @Override
 public String outputFile(TableInfo tableInfo) {
     return OUT_PATH + "/xml/" + tableInfo.getEntityName() + "Mapper.xml";
 }
    })))
  .setTemplate(
   // 关闭默认 xml 生成，调整生成 至 根目录
   new TemplateConfig().setXml(null)
 // 自定义模板配置，模板可以参考源码 /mybatis-plus/src/main/resources/template 使用 copy
 // 至您项目 src/main/resources/template 目录下，模板名称也可自定义如下配置：
 // .setController("...");
 // .setEntity("...");
 // .setMapper("...");
 // .setXml("...");
 // .setService("...");
 // .setServiceImpl("...");
 );

 // 执行生成
 mpg.execute();
    }

}

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

    @Autowired
    IUserService userService;

    @PostMapping("add")
    //正常业务时， 需要在user类里面进行事务控制，控制层一般不进行业务控制的。
    //@Transactional(rollbackFor = Exception.class)
    public Map addUser(@Valid @RequestBody UserReq userReq){

 User user = new User();
 user.setCode(userReq.getCode());
 user.setName(userReq.getName());
 //由于设置了主键策略 id可不用赋值 会自动生成
 //user.setId(0L);
 userService.insert(user);
 Map result = new HashMap();
 result.put("respCode", "01");
 result.put("respMsg", "新增成功");
 //事务测试
 //System.out.println(1/0);
 return result;
    }

    @PostMapping("update")
    public Map updateUser(@Valid @RequestBody UserReq userReq){

 if(userReq.getId() == null || "".equals(userReq.getId())) {
     throw new CommonException("0000", "更新时ID不能为空");
 }
 User user = new User();
 user.setCode(userReq.getCode());
 user.setName(userReq.getName());
 user.setId(Long.parseLong(userReq.getId())); 
 userService.updateById(user);
 Map result = new HashMap();
 result.put("respCode", "01");
 result.put("respMsg", "更新成功");
 return result;
    }

    @GetMapping("/get/{id}")
    public Map getUser(@PathVariable("id") String id){
 //查询
 User user = userService.selectById(id);
 if(user == null) {
     throw new CommonException("0001", "用户ID：" + id + "，未找到");
 }
 UserResp resp = UserResp.builder()
  .id(user.getId().toString())
  .code(user.getCode())
  .name(user.getName())
  .status(user.getStatus())
  .build();
 Map result = new HashMap();
 result.put("respCode", "01");
 result.put("respMsg", "成功");
 result.put("data", resp);
 return result;
    }

    @GetMapping("/page")
    public Map pageUser(int current, int size){
 //分页
 Page page = new Page<>(current, size);
 Map result = new HashMap();
 result.put("respCode", "01");
 result.put("respMsg", "成功");
 result.put("data", userService.selectPage(page));
 return result;
    }
}

public enum StatusEnum {

    DISABLE,
    ENABLE;

}

    
    private StatusEnum status;