feat:代码生成脚本

2026-01-30 21:41:20 +08:00
parent 217dc6879d
commit 59b3dd6320
123 changed files with 8832 additions and 2 deletions
--- a/script/src/main/resources/templates/code-generator/mybatis-plus/pluscontroller.ftl
+++ b/script/src/main/resources/templates/code-generator/mybatis-plus/pluscontroller.ftl
@@ -0,0 +1,171 @@
+<#if isWithPackage?exists && isWithPackage==true>package ${packageName}.controller;</#if>
+<#if isAutoImport?exists && isAutoImport==true>
+import com.alibaba.fastjson.JSON;
+import ${packageName}.entity.${classInfo.className};
+import ${packageName}.mapper.${classInfo.className}Mapper;
+import ${packageName}.util.ReturnT;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import org.springframework.web.servlet.ModelAndView;
+<#if isSwagger?exists && isSwagger==true>import io.swagger.v3.oas.annotations.tags.Tag;
+import io.swagger.v3.oas.annotations.responses.ApiResponse;</#if>
+import java.util.Date;
+import java.util.List;
+import java.util.Map;
+</#if>
+/**
+* @description ${classInfo.classComment}控制器
+* @author ${authorName}
+* @date ${.now?string('yyyy-MM-dd')}
+*/
+@Slf4j
+@RestController
+@RequestMapping("/${classInfo.className?uncap_first}")
+<#if isSwagger?exists && isSwagger==true>@Tag(name = "${classInfo.className}",description = "${classInfo.classComment}控制器")</#if>
+public class ${classInfo.className}Controller {
+
+    @Autowired
+    private ${classInfo.className}Mapper ${classInfo.className?uncap_first}Mapper;
+
+    /**
+    * 新增或编辑
+    */
+    @PostMapping("/save")
+    public Object save(@RequestBody ${classInfo.className} ${classInfo.className?uncap_first}){
+        log.info("${classInfo.className?uncap_first}:"+JSON.toJSONString(${classInfo.className?uncap_first}));
+        ${classInfo.className} old${classInfo.className} = ${classInfo.className?uncap_first}Mapper.selectOne(new QueryWrapper<${classInfo.className}>().eq("${classInfo.className?uncap_first}_id",${classInfo.className?uncap_first}.get${classInfo.className}Id()));
+        ${classInfo.className?uncap_first}.setUpdateTime(new Date());
+        if(old${classInfo.className}!=null){
+            ${classInfo.className?uncap_first}Mapper.updateById(${classInfo.className?uncap_first});
+        }else{
+            if(${classInfo.className?uncap_first}Mapper.selectOne(new QueryWrapper<${classInfo.className}>().eq("${classInfo.className?uncap_first}_name",${classInfo.className?uncap_first}.get${classInfo.className}Name()))!=null){
+                return ${returnUtilFailure}("保存失败，名字重复");
+            }
+            ${classInfo.className?uncap_first}.setCreateTime(new Date());
+            ${classInfo.className?uncap_first}Mapper.insert(${classInfo.className?uncap_first});
+        }
+        return ${returnUtilSuccess}("保存成功");
+    }
+
+    /**
+    * 删除
+    */
+    @PostMapping("/delete")
+    public Object delete(int id){
+    ${classInfo.className} ${classInfo.className?uncap_first} = ${classInfo.className?uncap_first}Mapper.selectOne(new QueryWrapper<${classInfo.className}>().eq("${classInfo.className?uncap_first}_id",id));
+        if(${classInfo.className?uncap_first}!=null){
+            ${classInfo.className?uncap_first}Mapper.deleteById(id);
+            return ${returnUtilSuccess}("删除成功");
+        }else{
+            return ${returnUtilFailure}("没有找到该对象");
+        }
+    }
+
+    /**
+    * 查询
+    */
+    @PostMapping("/find")
+    public Object find(int id){
+    ${classInfo.className} ${classInfo.className?uncap_first} = ${classInfo.className?uncap_first}Mapper.selectOne(new QueryWrapper<${classInfo.className}>().eq("${classInfo.className?uncap_first}_id",id));
+        if(${classInfo.className?uncap_first}!=null){
+            return ${returnUtilSuccess}(${classInfo.className?uncap_first});
+        }else{
+            return ${returnUtilFailure}("没有找到该对象");
+        }
+    }
+
+    /**
+    * 自动分页查询
+    */
+    @PostMapping("/list")
+    public Object list(@RequestBody ${classInfo.className} ${classInfo.className?uncap_first},@RequestParam(required = false, defaultValue = "0") int page,@RequestParam(required = false, defaultValue = "10") int limit) {
+        log.info("page:"+page+"-limit:"+limit+"-json:"+ JSON.toJSONString(searchParams));
+        //分页构造器
+        Page<${classInfo.className}> buildPage = new Page<${classInfo.className}>(page,limit);
+        //条件构造器
+        QueryWrapper<${classInfo.className}> queryWrapper = new QueryWrapper<${classInfo.className}>();
+        if(JSON.stringify(${classInfo.className?uncap_first}).length()>2) {
+            //自行删除不需要动态查询字段
+            queryWrapper.
+            <#list classInfo.fieldList as fieldItem>
+                eq(StringUtils.isNoneEmpty(${classInfo.className?uncap_first}.get${fieldItem.fieldName}()), "${fieldItem.fieldName}", ${classInfo.className?uncap_first}.get${fieldItem.fieldName}())
+            </#list>
+            ;
+        }
+        //执行分页
+        IPage<${classInfo.className}> pageList = ${classInfo.className?uncap_first}Mapper.selectPage(buildPage, queryWrapper);
+        //返回结果
+        return ${returnUtil}.PAGE(pageList.getRecords(),pageList.getTotal());
+    }
+
+    /**
+    * 动态条件手工分页查询
+    * 根据对象属性自动构建条件，如果字段有值则进行分页+指定条件查询，否则仅进行分页查询
+    */
+    @PostMapping("/list")
+    public Object list(String searchParams, @RequestParam(required = false, defaultValue = "1") int page, @RequestParam(required = false, defaultValue = "10") int limit) {
+        log.info("page:"+page+"-limit:"+limit+"-json:"+ JSON.toJSONString(searchParams));
+        
+        // 分页参数处理
+        int offset = (page - 1) * limit;
+        
+        // 查询参数处理
+        ${classInfo.className} queryParamDTO = new ${classInfo.className}();
+        if(StringUtils.isNotEmpty(searchParams) && JSON.isValid(searchParams)) {
+            queryParamDTO = JSON.parseObject(searchParams, ${classInfo.className}.class);
+        }
+        
+        // 使用动态条件查询
+        List<${classInfo.className}> itemList = ${classInfo.className?uncap_first}Mapper.selectPageByCondition(queryParamDTO, offset, limit);
+        int itemCount = ${classInfo.className?uncap_first}Mapper.selectPageByConditionCount(queryParamDTO);
+        
+        //返回结果
+        return ${returnUtil}.PAGE(itemList, itemCount);
+    }
+
+    @GetMapping("/list")
+    public ModelAndView listPage(){
+        return new ModelAndView("${classInfo.className?uncap_first}-list");
+    }
+
+    @GetMapping("/edit")
+    public ModelAndView editPage(int id){
+        ${classInfo.className} ${classInfo.className?uncap_first} = ${classInfo.className?uncap_first}Mapper.selectOne(new QueryWrapper<${classInfo.className}>().eq("${classInfo.className?uncap_first}_id",id));
+        return new ModelAndView("${classInfo.className?uncap_first}-edit","${classInfo.className?uncap_first}",${classInfo.className?uncap_first});
+    }
+
+    /**
+    * 激活/停用(如不需要请屏蔽)
+    */
+    @PostMapping("/active")
+    public Object active(int id,Integer status){
+        ${classInfo.className} ${classInfo.className?uncap_first} = ${classInfo.className?uncap_first}Mapper.selectOne(new QueryWrapper<${classInfo.className}>().eq("${classInfo.className?uncap_first}_id",id));
+        if(${classInfo.className?uncap_first}!=null){
+            ${classInfo.className?uncap_first}.setUpdateTime(new Date());
+            ${classInfo.className?uncap_first}.setStatus(status);
+            ${classInfo.className?uncap_first}Mapper.updateById(${classInfo.className?uncap_first});
+            return ${returnUtilSuccess}((status==1)?"已激活":"已停用");
+        }else if(status.equals(${classInfo.className?uncap_first}.getStatus())){
+            return ${returnUtilFailure}("状态不正确");
+        }else{
+            return ${returnUtilFailure}();
+        }
+    }
+
+    /**
+    * 测试(如不需要请屏蔽)
+    */
+    @GetMapping("/test")
+    public Object test(){
+        return ${returnUtilSuccess};
+    }
+
+}
+
+
+
--- a/script/src/main/resources/templates/code-generator/mybatis-plus/plusentity.ftl
+++ b/script/src/main/resources/templates/code-generator/mybatis-plus/plusentity.ftl
@@ -0,0 +1,45 @@
+<#if isWithPackage?exists && isWithPackage==true>package ${packageName}.entity;</#if>
+
+<#if isAutoImport?exists && isAutoImport==true>
+<#if isLombok?exists && isLombok==true>import lombok.Data;</#if>
+import java.util.Date;
+import java.util.List;
+import java.io.Serializable;
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableId;
+<#if isSwagger?exists && isSwagger==true>
+import io.swagger.v3.oas.annotations.media.Schema;</#if>
+</#if>
+/**
+ * @description ${classInfo.classComment}
+ * @author ${authorName}
+ * @date ${.now?string('yyyy-MM-dd')}
+ */
+<#if isLombok?exists && isLombok==true>@Data</#if><#if isSwagger?exists && isSwagger==true>
+@Schema(description = "${classInfo.classComment}")</#if>
+public class ${classInfo.className} implements Serializable {
+
+    private static final long serialVersionUID = 1L;
+
+    @TableId(type = IdType.AUTO)
+<#if classInfo.fieldList?exists && classInfo.fieldList?size gt 0>
+<#list classInfo.fieldList as fieldItem >
+    <#if isComment?exists && isComment==true>/**
+    * ${fieldItem.fieldComment}
+    */</#if><#if isSwagger?exists && isSwagger==true>
+    @Schema(description = "${fieldItem.fieldComment}")</#if>
+    private ${fieldItem.fieldClass} ${fieldItem.fieldName};
+
+<#if isLombok?exists && isLombok==false>
+    public ${fieldItem.fieldClass} get${fieldItem.fieldName?cap_first}() {
+        return ${fieldItem.fieldName};
+    }
+
+    public void set${fieldItem.fieldName?cap_first}(${fieldItem.fieldClass} ${fieldItem.fieldName}) {
+        this.${fieldItem.fieldName} = ${fieldItem.fieldName};
+    }
+</#if>
+</#list>
+</#if>
+    public ${classInfo.className}() {}
+}
--- a/script/src/main/resources/templates/code-generator/mybatis-plus/plusmapper.ftl
+++ b/script/src/main/resources/templates/code-generator/mybatis-plus/plusmapper.ftl
@@ -0,0 +1,69 @@
+<#if isWithPackage?exists && isWithPackage==true>package ${packageName}.mapper;</#if>
+<#if isAutoImport?exists && isAutoImport==true>
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
+import org.apache.ibatis.annotations.Select;
+import ${packageName}.entity.${classInfo.className};
+import java.util.List;
+</#if>
+/**
+ * @description ${classInfo.classComment}Mapper
+ * @author ${authorName}
+ * @date ${.now?string('yyyy-MM-dd')}
+ */
+@Mapper
+public interface ${classInfo.className}Mapper extends BaseMapper<${classInfo.className}> {
+
+    /**
+     * 动态条件分页查询 - 根据对象属性自动构建条件
+     * 如果字段有值则进行分页+指定条件查询，否则仅进行分页查询
+     */
+    @Select("""
+        <script>
+        SELECT * FROM ${classInfo.tableName}
+        <where>
+        <#list classInfo.fieldList as fieldItem>
+            <#if fieldItem.fieldClass?contains("String")>
+            <if test='queryParamDTO.${fieldItem.fieldName} != null and queryParamDTO.${fieldItem.fieldName} != ""'>
+                AND ${fieldItem.columnName} = 井{queryParamDTO.${fieldItem.fieldName}}
+            </if>
+            <#else>
+            <if test='queryParamDTO.${fieldItem.fieldName} != null'>
+                AND ${fieldItem.columnName} = 井{queryParamDTO.${fieldItem.fieldName}}
+            </if>
+            </#if>
+        </#list>
+        </where>
+        ORDER BY id DESC
+        LIMIT 井{offset}, 井{limit}
+        </script>
+    """)
+    List<${classInfo.className}> selectPageByCondition(@Param("queryParamDTO") ${classInfo.className} queryParamDTO, 
+                                                     @Param("offset") int offset, 
+                                                     @Param("limit") int limit);
+
+    /**
+     * 动态条件分页查询总数
+     */
+    @Select("""
+        <script>
+        SELECT COUNT(*) FROM ${classInfo.tableName}
+        <where>
+        <#list classInfo.fieldList as fieldItem>
+            <#if fieldItem.fieldClass?contains("String")>
+            <if test='queryParamDTO.${fieldItem.fieldName} != null and queryParamDTO.${fieldItem.fieldName} != ""'>
+                AND ${fieldItem.columnName} = 井{queryParamDTO.${fieldItem.fieldName}}
+            </if>
+            <#else>
+            <if test='queryParamDTO.${fieldItem.fieldName} != null'>
+                AND ${fieldItem.columnName} = 井{queryParamDTO.${fieldItem.fieldName}}
+            </if>
+            </#if>
+        </#list>
+        </where>
+        </script>
+    """)
+    int selectPageByConditionCount(@Param("queryParamDTO") ${classInfo.className} queryParamDTO);
+
+}
--- a/script/src/main/resources/templates/code-generator/mybatis-plus/plusservice.ftl
+++ b/script/src/main/resources/templates/code-generator/mybatis-plus/plusservice.ftl
@@ -0,0 +1,16 @@
+<#if isWithPackage?exists && isWithPackage==true>package ${packageName}.service;</#if>
+<#if isAutoImport?exists && isAutoImport==true>
+import org.springframework.stereotype.Service;
+import com.baomidou.mybatisplus.extension.service.IService;
+</#if>
+/**
+ * @description ${classInfo.classComment}服务层
+ * @author ${authorName}
+ * @date ${.now?string('yyyy-MM-dd')}
+ */
+@Service
+public interface ${classInfo.className}Service extends IService<${classInfo.className}> {
+
+
+
+}