EasyCode自定义模板分享

652 阅读1分钟

EasyCode简介:

EasyCode 是基于 IntelliJ IDEA 开发的代码生成插件,支持自定义任意模板(Java,html,js,xml)。只要是与数据库相关的代码都可以通过自定义模板来生成。支持数据库类型与 java 类型映射关系配置。支持同时生成生成多张表的代码。每张表有独立的配置信息。完全的个性化定义,规则由你设置。

配置自定义模板:

entity.java.vm

实体类模板改动如下:

  1. 采用 lombok 的 @Data 注解替换原本的 Getter/Setter方法
  2. 类和字段添加 Swagger 注解
  3. 主键字段添加@TableId注解
  4. 添加去除表前缀配置并添加@TableName注解
##导入宏定义
$!{define.vm}

##保存文件(宏定义)
#save("/domain", ".java")

##包路径(宏定义)
#setPackageSuffix("domain")

##自动导入包(全局变量)
$!autoImport
import com.baomidou.mybatisplus.annotation.TableName;
import java.io.Serializable;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

##表注释(宏定义)
#tableComment("表实体类")
@Data
@ApiModel("$tableInfo.comment")
@TableName(value = "$tableInfo.obj.name")
public class $!{tableInfo.name} implements Serializable{
private static final long serialVersionUID = $!tool.serial();
##以第一个primary key字段没primary key
#foreach($column in $tableInfo.pkColumn)
    #set($pkColumn = $!column.name)
    #break
#end

#foreach($column in $tableInfo.fullColumn)
    #if(${column.comment})
    /**
    * ${column.comment}
    */#end
    
    #if(${column.name} == ${pkColumn})
    @TableId(type = IdType.AUTO)
    #end
    @ApiModelProperty("$column.comment")
    private $!{tool.getClsNameByFullName($column.type)} $!{column.name};
    
#end
}

mapper.java.vm

Mapper模板改动如下:

  1. 添加 @Repository 注解
##导入宏定义
$!{define.vm}

##设置表后缀(宏定义)
#setTableSuffix("Mapper")

##保存文件(宏定义)
#save("/mapper", "Mapper.java")

##包路径(宏定义)
#setPackageSuffix("mapper")

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import org.springframework.stereotype.Repository;
import $!{tableInfo.savePackageName}.domain.$!tableInfo.name;

##表注释(宏定义)
#tableComment("表数据库访问层")
@Repository
public interface $!{tableName} extends BaseMapper<$!tableInfo.name> {

}

mapper.xml.vm

##引入mybatis支持
$!{mybatisSupport.vm}

##设置保存名称与保存位置
$!callback.setFileName($tool.append($!{tableInfo.name}, "Mapper.xml"))
$!callback.setSavePath($tool.append($modulePath, "/src/main/resources/mapper"))

##拿到主键
#if(!$tableInfo.pkColumn.isEmpty())
    #set($pk = $tableInfo.pkColumn.get(0))
#end

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="$!{tableInfo.savePackageName}.mapper.$!{tableInfo.name}Mapper">

    <resultMap type="$!{tableInfo.savePackageName}.domain.$!{tableInfo.name}" id="$!{tableInfo.name}Map">
#foreach($column in $tableInfo.fullColumn)
        <result property="$!column.name" column="$!column.obj.name" jdbcType="$!column.ext.jdbcType"/>
#end
    </resultMap>

</mapper>

service.java.vm

##导入宏定义
$!{define.vm}

##设置表后缀(宏定义)
#setTableSuffix("Service")

##保存文件(宏定义)
#save("/service", "Service.java")

##包路径(宏定义)
#setPackageSuffix("service")

import com.baomidou.mybatisplus.extension.service.IService;
import $!{tableInfo.savePackageName}.domain.$!tableInfo.name;

##表注释(宏定义)
#tableComment("表服务接口")
public interface $!{tableName} extends IService<$!tableInfo.name> {

}

serviceImpl.java.vm

##导入宏定义
$!{define.vm}

##设置表后缀(宏定义)
#setTableSuffix("ServiceImpl")

##保存文件(宏定义)
#save("/service/impl", "ServiceImpl.java")

##包路径(宏定义)
#setPackageSuffix("service.impl")

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import $!{tableInfo.savePackageName}.mapper.$!{tableInfo.name}Mapper;
import $!{tableInfo.savePackageName}.domain.$!{tableInfo.name};
import $!{tableInfo.savePackageName}.service.$!{tableInfo.name}Service;
import org.springframework.stereotype.Service;
import lombok.AllArgsConstructor;

##表注释(宏定义)
#tableComment("表服务实现类")
@Service
@AllArgsConstructor
public class $!{tableName} extends ServiceImpl<$!{tableInfo.name}Mapper, $!{tableInfo.name}> implements $!{tableInfo.name}Service {
     private final $!{tableInfo.name}Mapper $!tool.firstLowerCase($!{tableInfo.name})Mapper;
}

controller.java.vm

Controller模板改动如下:

  1. 类和方法添加 Swagger 注解
  2. 分页查询修改为LambdaQueryWrapper条件
  3. 删除接口修改为批量删除
##导入宏定义
$!{define.vm}

##设置表后缀(宏定义)
#setTableSuffix("Controller")

##保存文件(宏定义)
#save("/controller", "Controller.java")

##包路径(宏定义)
#setPackageSuffix("controller")

##定义服务名
#set($serviceName = $!tool.append($!tool.firstLowerCase($!tableInfo.name), "Service"))

##定义实体对象名
#set($entityName = $!tool.firstLowerCase($!tableInfo.name))

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import $!{tableInfo.savePackageName}.domain.$!tableInfo.name;
import $!{tableInfo.savePackageName}.service.$!{tableInfo.name}Service;
import org.springframework.web.bind.annotation.*;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;

import lombok.AllArgsConstructor;
import java.io.Serializable;
import java.util.Arrays;
import java.util.List;

##表注释(宏定义)
#tableComment("表控制层")
@RestController
@AllArgsConstructor
@Api(tags = "$!{tableInfo.comment}")
@RequestMapping("/$!tool.firstLowerCase($!tableInfo.name)")
public class $!{tableName} {
    /**
     * 服务对象
     */
    private final $!{tableInfo.name}Service $!{serviceName};

    /**
     * 分页查询所有数据
     *
     * @param page 分页对象
     * @param $!entityName 查询实体
     * @return 所有数据
     */
    @GetMapping("/page")
    @ApiOperation(value = "分页查询")
    public R selectAll(Page<$!tableInfo.name> page, $!tableInfo.name $!entityName) {
        return R.data($!{serviceName}.page(page, new LambdaQueryWrapper<>($!entityName)));
    }

    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    @GetMapping("{id}")
    @ApiOperation(value = "主键查询")
    public R selectOne(@PathVariable Serializable id) {
        return R.data($!{serviceName}.getById(id));
    }

    /**
     * 新增数据
     *
     * @param $!entityName 实体对象
     * @return 新增结果
     */
    @PostMapping("/save")
    @ApiOperation(value = "新增")
    public R insert(@RequestBody $!tableInfo.name $!entityName) {
        return R.data($!{serviceName}.save($!entityName));
    }

    /**
     * 修改数据
     *
     * @param $!entityName 实体对象
     * @return 修改结果
     */
    @PutMapping
    @ApiOperation(value = "修改")
    public R<String> update(@RequestBody $!tableInfo.name $!entityName) {
        $!{serviceName}.updateById($!entityName);
        return R.success("修改成功");
    }

    /**
     * 删除数据
     *
     * @param ids 主键结合
     * @return 删除结果
     */
    @DeleteMapping("/{ids}")
    @ApiOperation(value = "删除")
    public R<String> delete(@PathVariable("ids") String ids) {
        String[] ids1 = ids.split(",");
        $!{serviceName}.removeByIds(Arrays.asList(ids1));
        return R.success("删除成功");
    }
}