Yaxin
/
fu-hsi-service
31
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

代码生成器

Browse Source
jinan_dev
longbao 3 months ago
parent 51dc2cb9d7
commit 00f85c8dc9
5 changed files with 364 additions and 0 deletions
Show Stats Download Patch File Download Diff File

18
pom.xml
Unescape Escape View File

@ -65,6 +65,24 @@
<version>3.5.5</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.5.9</version>
</dependency>
<!-- 添加模板引擎，thymeleaf 还有freemarker都是模板引擎 -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-freemarker</artifactId>
</dependency>
<!--mybatis-plus 代码生成器的模板引擎-->
<dependency>
<groupId>org.apache.velocity</groupId>
<artifactId>velocity-engine-core</artifactId>
<version>2.3</version>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>

139
src/main/java/com/supervision/police/mybatis/MySqlCodeGenerator.java
Unescape Escape View File

@ -0,0 +1,139 @@
package com.supervision.police.mybatis;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;
import java.sql.Types;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
/**
* @Author：longbao
* @Date：2024/11/19 13:03
* @Description: 代码生成器
*/
public class MySqlCodeGenerator {
public static void main(String[] args) {
String projectPath = System.getProperty("user.dir"); //获取当前项目路径
List<String> tables = new ArrayList<>();
//添加 表 到集合中，可以添加多个表
tables.add("transcription_triple_info");
FastAutoGenerator.create("jdbc:mysql://192.168.1.101:42415/nx_llm?useUnicode=true&characterEncoding=utf-8&useSSL=true&allowMultiQueries=true",
"root",
"llm@984573~!eda")
// 全局配置
.globalConfig(builder -> {
builder
.enableSwagger() // 是否启用swagger注解
.author("longbao") // 作者名称
.dateType(DateType.ONLY_DATE) // 时间策略
.commentDate("yyyy-MM-dd") // 注释日期
.outputDir(projectPath + "/src/main/java") // 输出目录
.disableOpenDir(); // 生成后禁止打开所生成的系统目录
})
//java和数据库字段的类型转换
.dataSourceConfig(builder -> builder.typeConvertHandler((globalConfig, typeRegistry, metaInfo) -> {
int typeCode = metaInfo.getJdbcType().TYPE_CODE;
if (typeCode == Types.SMALLINT || typeCode == Types.TINYINT) {
// 自定义类型转换
return DbColumnType.INTEGER;
}
return typeRegistry.getColumnType(metaInfo);
}))
// 包配置
.packageConfig(builder -> {
builder
.parent("com.supervision") // 父包名
.moduleName("police") // 模块包名
.controller("controller")
.entity("domain") // 实体类包名
.service("service") // service包名
.serviceImpl("service.impl") // serviceImpl包名
.mapper("mapper") // mapper包名
.xml("mapper.xml");
})
// 策略配置
.strategyConfig(builder -> {
builder.enableCapitalMode()//驼峰
.enableSkipView()//跳过视图
.disableSqlFilter()
// .addTablePrefix("t_") // 增加过滤表前缀
// .addTableSuffix("_db") // 增加过滤表后缀
// .addFieldPrefix("t_") // 增加过滤字段前缀
// .addFieldSuffix("_field") // 增加过滤字段后缀
.addInclude(tables) // 表匹配
// Entity 策略配置
.entityBuilder()
.enableFileOverride()
.enableLombok() // 开启lombok
.enableChainModel() // 链式
.enableRemoveIsPrefix() // 开启boolean类型字段移除is前缀
.enableTableFieldAnnotation() //开启生成实体时生成的字段注解
.versionColumnName("version") // 乐观锁数据库字段
.versionPropertyName("version") // 乐观锁实体类名称
.logicDeleteColumnName("delflag") // 逻辑删除数据库中字段名
.logicDeletePropertyName("delFlag") // 逻辑删除实体类中的字段名
.naming(NamingStrategy.underline_to_camel) // 表名 下划线 -》 驼峰命名
.columnNaming(NamingStrategy.underline_to_camel) // 字段名 下划线 -》 驼峰命名
.idType(IdType.ASSIGN_ID) // 主键生成策略 雪花算法生成id
.formatFileName("%s") // Entity 文件名称
.addTableFills(new Column("create_time", FieldFill.INSERT)) // 表字段填充
.addTableFills(new Column("update_time", FieldFill.INSERT_UPDATE)) // 表字段填充
//.enableColumnConstant()
//.enableActiveRecord()//MPlus中启用ActiveRecord模式，生成的实体类会继承activerecord.Model类，直接进行数据库操作
// Controller 策略配置
.controllerBuilder()
.enableFileOverride()
.enableHyphenStyle()
.enableRestStyle() // 开启@RestController
.formatFileName("%sController") // Controller 文件名称
// Service 策略配置
.serviceBuilder()
.enableFileOverride()
.formatServiceFileName("%sService") // Service 文件名称
.formatServiceImplFileName("%sServiceImpl") // ServiceImpl 文件名称
// Mapper 策略配置
.mapperBuilder()
.enableFileOverride()
.enableMapperAnnotation() // 开启@Mapper
.enableBaseColumnList() // 启用 columnList (通用查询结果列)
.enableBaseResultMap() // 启动resultMap
.formatMapperFileName("%sMapper") // Mapper 文件名称
.formatXmlFileName("%sMapper"); // Xml 文件名称
})
.templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
.templateConfig(builder -> {
builder.controller("/templates/controller.java")
.service("/templates/service.java")
.serviceImpl("/templates/serviceImpl.java")
//.mapper()
.build();
})
.execute(); // 执行
}
}

85
src/main/resources/template/controller.java.ftl
Unescape Escape View File

@ -0,0 +1,85 @@
package ${package.Controller};
import ${package.Entity}.${entity};
import ${package.Service}.${table.serviceName};
import io.swagger.annotations.ApiOperation;
<#--import org.apache.shiro.authz.annotation.Logical;-->
<#--import org.apache.shiro.authz.annotation.RequiresPermissions;-->
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import javax.validation.constraints.NotBlank;
import javax.validation.Valid;
import java.util.List;
<#--import com.common.res.DataResult;-->
<#if restControllerStyle>
<#else>
import org.springframework.stereotype.Controller;
</#if>
<#if superControllerClassPackage??>
import ${superControllerClassPackage};
</#if>
/**
* @author ${author}
* @since ${date}
*/
<#if restControllerStyle>
@RestController
<#else>
@Controller
</#if>
@RequestMapping("/<#if controllerMappingHyphenStyle>${controllerMappingHyphen}<#else>${table.entityPath}</#if>")
<#if kotlin>
class ${table.controllerName}<#if superControllerClass??> : ${superControllerClass}()</#if>
<#else>
<#if superControllerClass??>
public class ${table.controllerName} extends ${superControllerClass} {
<#else>
public class ${table.controllerName} {
</#if>
@Autowired
private ${table.serviceName} ${table.serviceName?uncap_first};
@GetMapping("/selectOne")
<#-- @RequiresPermissions("sys:${table.entityName?uncap_first}:list")-->
@ApiOperation("${table.entityName}查询单个")
public ${table.entityName} get${table.entityName}(@RequestParam("id") Integer id){
${table.entityName} ${table.entityName?uncap_first}One = ${table.entityName?uncap_first}Service.get${table.entityName}( id);
return ${table.entityName?uncap_first}One;
}
@GetMapping("/listAll")
<#-- @RequiresPermissions("sys:${table.entityName?uncap_first}:list")-->
@ApiOperation("${table.entityName}查询全部")
public List<${table.entityName}> getAll${table.entityName}(){
List<${table.entityName}> ${table.entityName?uncap_first}List = ${table.entityName?uncap_first}Service.getAll${table.entityName}();
return ${table.entityName?uncap_first}List;
}
@PostMapping("/add")
<#-- @RequiresPermissions("sys:${table.entityName?uncap_first}:add")-->
@ApiOperation("${table.entityName}新增")
public Object add(@Valid @RequestBody ${table.entityName} ${table.entityName?uncap_first}) {
${table.entityName?uncap_first}Service.add( ${table.entityName?uncap_first});
return null;
}
@PutMapping("/update")
<#-- @RequiresPermissions("sys:${table.entityName?uncap_first}:update")-->
@ApiOperation("${table.entityName}修改")
public int update(@Valid @RequestBody ${table.entityName} ${table.entityName?uncap_first}) {
int num = ${table.entityName?uncap_first}Service.modify( ${table.entityName?uncap_first});
return num;
}
@DeleteMapping(value = "/delete/{ids}")
<#-- @RequiresPermissions("sys:${table.entityName?uncap_first}:delete")-->
@ApiOperation("${table.entityName}删除(单个条目)")
public Object remove(@NotBlank(message = "{required}") @PathVariable String ids) {
${table.entityName?uncap_first}Service.remove(ids);
return null;
}
}
</#if>

53
src/main/resources/template/service.java.ftl
Unescape Escape View File

@ -0,0 +1,53 @@
package ${package.Service};
import ${package.Entity}.${entity};
import ${superServiceClassPackage};
import java.util.List;
/**
* @author ${author}
* @since ${date}
*/
<#if kotlin>
interface ${table.serviceName} : ${superServiceClass}<${entity}>
<#else>
public interface ${table.serviceName} extends ${superServiceClass}<${entity}> {
/**
* ${table.entityName!}详情
* @param
* @return
*/
${table.entityName} get${table.entityName}( Integer id);
/**
* ${table.entityName!}详情
* @param
* @return
*/
List<${table.entityName}> getAll${table.entityName}();
/**
* ${table.entityName!}新增
* @param ${table.entityName?uncap_first} 根据需要进行传值
* @return
*/
void add(${entity} ${table.entityName?uncap_first});
/**
* ${table.entityName!}修改
* @param ${table.entityName?uncap_first} 根据需要进行传值
* @return
*/
int modify(${entity} ${table.entityName?uncap_first});
/**
* ${table.entityName!}删除
* @param ids
* @return
*/
void remove(String ids);
}
</#if>

69
src/main/resources/template/serviceImpl.java.ftl
Unescape Escape View File

@ -0,0 +1,69 @@
package ${package.ServiceImpl};
import ${package.Entity}.${entity};
import ${package.Mapper}.${table.mapperName};
import ${package.Service}.${table.serviceName};
import ${superServiceImplClassPackage};
import org.springframework.stereotype.Service;
import org.springframework.beans.factory.annotation.Autowired;
import org.apache.commons.lang3.StringUtils;
import org.springframework.util.CollectionUtils;
import java.util.List;
import java.util.Arrays;
/**
* ${table.comment!} 服务实现类
*
* @author ${author}
* @since ${date}
*/
@Service
<#if kotlin>
open class ${table.serviceImplName} : ${superServiceImplClass}<${table.mapperName}, ${entity}>(), ${table.serviceName} {
}
<#else>
public class ${table.serviceImplName} extends ${superServiceImplClass}<${table.mapperName}, ${entity}> implements ${table.serviceName} {
@Autowired
${table.entityName}Mapper ${table.entityName?uncap_first}Mapper;
@Override
public ${table.entityName} get${table.entityName}(Integer id){
return ${table.entityName?uncap_first}Mapper.selectById(id);
}
@Override
public List<${table.entityName}> getAll${table.entityName}(){
return ${table.entityName?uncap_first}Mapper.selectList(null);
}
@Override
public void add( ${table.entityName} ${table.entityName?uncap_first}) {
${table.entityName?uncap_first}Mapper.insert(${table.entityName?uncap_first});
}
@Override
public int modify( ${table.entityName} ${table.entityName?uncap_first}) {
//乐观锁更新
${table.entityName} current${table.entityName}= ${table.entityName?uncap_first}Mapper.selectById(${table.entityName?uncap_first}.getId());
${table.entityName?uncap_first}.setVersion(current${table.entityName}.getVersion());
return ${table.entityName?uncap_first}Mapper.updateById(${table.entityName?uncap_first});
}
@Override
public void remove( String ids) {
if(StringUtils.isNotEmpty(ids)){
String[] array = ids.split(",");
if (!CollectionUtils.isEmpty(Arrays.asList(array))) {
${table.entityName?uncap_first}Mapper.deleteBatchIds(Arrays.asList(array));
}
}
}
}
</#if>
Write Preview
Loading…
Cancel
Save