idea自动生成代码插件EasyCode 篇二

朱雀 2023-10-05 10:29 150阅读 0赞

上一篇说了EasyCode的使用，这一篇记录一下常用的代码模板。

--------------------

**`entity`**  
使用了`lombok`插件，需要添加`lombok.Data`引用

##引入宏定义
    $!define
    
    ##使用宏定义设置回调（保存位置与文件后缀）
    #save("/entity", ".java")
    
    ##使用宏定义设置包后缀
    #setPackageSuffix("entity")
    
    ##使用全局变量实现默认包导入
    $!autoImport
    import java.io.Serializable;
    import lombok.Data;
    
    ##使用宏定义实现类注释信息
    #tableComment("实体类")
    @Data
    public class $!{
        tableInfo.name} implements Serializable {
        
        private static final long serialVersionUID = $!tool.serial();
    #foreach($column in $tableInfo.fullColumn)
        #if(${
        column.comment})/**
        * ${column.comment}
        */#end
    
        private $!{
        tool.getClsNameByFullName($column.type)} $!{
        column.name};
    #end
    
    }

**`controller`**  
使用了`lombok`插件，需要添加`lombok.Data`引用

##定义初始变量
    #set($tableName = $tool.append($tableInfo.name, "Controller"))
    ##设置回调
    $!callback.setFileName($tool.append($tableName, ".java"))
    $!callback.setSavePath($tool.append($tableInfo.savePath, "/controller"))
    ##拿到主键
    #if(!$tableInfo.pkColumn.isEmpty())
        #set($pk = $tableInfo.pkColumn.get(0))
    #end
    
    #if($tableInfo.savePackageName)package $!{
        tableInfo.savePackageName}.#{
        end}controller;
    
    import com.test.common.core.annotation.Log;
    import com.test.common.core.controller.BaseController;
    import com.test.common.core.domain.AjaxResult;
    import com.test.common.core.enums.BusinessType;
    import com.test.common.core.page.TableDataInfo;
    import $!{
        tableInfo.savePackageName}.entity.$!{
        tableInfo.name};
    import $!{
        tableInfo.savePackageName}.service.$!{
        tableInfo.name}Service;
    import org.springframework.validation.annotation.Validated;
    import org.springframework.web.bind.annotation.*;
    
    import javax.annotation.Resource;
    import java.util.List;
    
    
    
    /**
     * $!{tableInfo.comment}($!{tableInfo.name})表控制层
     *
     * @author $!author
     * @since $!time.currTime()
     */
    @RestController
    @RequestMapping("$!tool.firstLowerCase($tableInfo.name)")
    public class $!{
        tableName} extends BaseController{
        
        /**
         * 服务对象
         */
        @Resource
        private $!{
        tableInfo.name}Service $!tool.firstLowerCase($tableInfo.name)Service;
    
        /**
         * 通过主键查询单条数据
         *
         * @param id 主键
         * @return AjaxResult
         */
        @GetMapping("selectOne/{id}")
        public AjaxResult selectOne(@PathVariable("id") $!pk.shortType id) {
        
    		$!{
        tableInfo.name} $!tool.firstLowerCase($tableInfo.name) = $!{
        tool.firstLowerCase($tableInfo.name)}Service.queryById(id);
    	    return AjaxResult.success($!tool.firstLowerCase($tableInfo.name));
        }
    		
    	 /**
         * 通过条件查询数据
    	 *
    	 * @return TableDataInfo
         */
        @GetMapping("/list")
        public TableDataInfo list($!{
        tableInfo.name} param){
        
            
            startPage();
            List<$!{
        tableInfo.name}> list = $!{
        tool.firstLowerCase($tableInfo.name)}Service.queryAllByParam(param);
            return getDataTable(list);
        }
    	
        /**
         * 新增数据
         *
         * @return AjaxResult
         */
    	@Log(title = "$!{tableInfo.comment}", businessType = BusinessType.INSERT)
        @PostMapping("add")
        public AjaxResult add(@Validated @RequestBody $!{
        tableInfo.name} param) {
        
    		
            $!{
        tableInfo.name} $!tool.firstLowerCase($tableInfo.name) = $!{
        tool.firstLowerCase($tableInfo.name)}Service.insert(param);
    
            return AjaxResult.success($!tool.firstLowerCase($tableInfo.name));
        }
    
        /**
         * 修改数据
         *
         * @return AjaxResult
         */
    	@Log(title = "$!{tableInfo.comment}", businessType = BusinessType.UPDATE)
        @PutMapping("update")
        public AjaxResult update(@Validated @RequestBody $!{
        tableInfo.name} param) {
        
            $!pk.shortType id = param.getId();
            if(id == null){
        
                return AjaxResult.error("id不允许为空");
            }
    
            $!{
        tableInfo.name} $!tool.firstLowerCase($tableInfo.name) = $!{
        tool.firstLowerCase($tableInfo.name)}Service.update(param);
    
            return AjaxResult.success($!tool.firstLowerCase($tableInfo.name));
        }
    
    
        /**
         * 根据id删除数据
         *
         * @return AjaxResult
         */
    	@Log(title = "$!{tableInfo.comment}", businessType = BusinessType.DELETE)
    	@DeleteMapping("/{id}")
        public AjaxResult remove(@PathVariable $!pk.shortType id) {
        
            boolean result = $!{
        tool.firstLowerCase($tableInfo.name)}Service.deleteById(id);
            
            if(result) return AjaxResult.success();
            return AjaxResult.error();
        }
    
    }

**`mapper.xml`**  
insert 需要修改

##引入mybatis支持
    $!mybatisSupport
    
    ##设置保存名称与保存位置
    $!callback.setFileName($tool.append($!{tableInfo.name}, "Dao.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}.dao.$!{tableInfo.name}Dao">
    
        <resultMap type="$!{tableInfo.savePackageName}.entity.$!{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>
    
        
        <select id="queryById" resultMap="$!{tableInfo.name}Map">
            select
              #allSqlColumn()
    
            from $!tableInfo.obj.name
            where $!pk.obj.name = #{$!pk.name}
        </select>
    
        
        <select id="queryAllByLimit" resultMap="$!{tableInfo.name}Map">
            select
              #allSqlColumn()
    
            from $!tableInfo.obj.name
            limit #{offset}, #{limit}
        </select>
    
        
        <select id="queryAll" resultMap="$!{tableInfo.name}Map">
            select
              #allSqlColumn()
    
            from $!tableInfo.obj.name
            <where>
    #foreach($column in $tableInfo.fullColumn)
                <if test="$!column.name != null#if($column.type.equals("java.lang.String")) and $!column.name != ''#end">
                    and $!column.obj.name = #{$!column.name}
                </if>
    #end
            </where>
        </select>
        
         
        <select id="queryAllByParam" parameterType="$!{tableInfo.savePackageName}.entity.$!{tableInfo.name}" resultMap="$!{tableInfo.name}Map">
            select
              #allSqlColumn()
    
            from $!tableInfo.obj.name
            <where>
    #foreach($column in $tableInfo.fullColumn)
                <if test="$!column.name != null#if($column.type.equals("java.lang.String")) and $!column.name != ''#end">
                    and $!column.obj.name = #{$!column.name}
                </if>
    #end
            </where>
        </select>
        
        
    
        
        <insert id="insert" keyProperty="$!pk.name" useGeneratedKeys="true">
            insert into $!{tableInfo.obj.name}
            <trim prefix="(" suffix=")" suffixOverrides=",">
    #foreach($column in $tableInfo.otherColumn)
                <if test="$!column.name != null#if($column.type.equals("java.lang.String")) and $!column.name != ''#end">
                    $!column.obj.name,
                </if>
    #end
            </trim>
            <trim prefix="values (" suffix=")" suffixOverrides=",">
            #foreach($column in $tableInfo.otherColumn)
                <if test="$!column.name != null#if($column.type.equals("java.lang.String")) and $!column.name != ''#end">
                    #{$!column.name},
                </if>
                #end
            </trim>
     </insert>
    
        
        <update id="update">
            update $!{tableInfo.obj.name}
            <set>
    #foreach($column in $tableInfo.otherColumn)
                <if test="$!column.name != null#if($column.type.equals("java.lang.String")) and $!column.name != ''#end">
                    $!column.obj.name = #{$!column.name},
                </if>
    #end
            </set>
            where $!pk.obj.name = #{$!pk.name}
        </update>
    
        
        <delete id="deleteById">
            delete from $!{tableInfo.obj.name} where $!pk.obj.name = #{$!pk.name}
        </delete>
    
    </mapper>

其它使用的是默认模板