WorkBuddy支持MyBatis-Plus代码一键生成:一、基于数据库表结构生成Entity、Mapper、Service、Controller;二、智能适配项目上下文(版本、配置、Lombok等);三、扩展支持RBAC与多租户;四、允许自定义FreeMarker模板;五、集成IDE实现生成后静态校验与自动修复。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

如果您在Java项目中使用MyBatis-Plus进行持久层开发,但需反复编写实体类、Mapper接口、Service及Controller等重复结构,WorkBuddy可直接介入生成流程,显著压缩手动编码耗时。以下是WorkBuddy对MyBatis-Plus代码生成的具体支持方式:
一、基于数据库表结构一键生成完整MP代码骨架
WorkBuddy能解析当前项目已连接的MySQL/PostgreSQL数据库元信息,自动识别表名、字段类型、主键、外键、索引及注释,并据此生成符合MyBatis-Plus规范的全套代码。该过程严格遵循MP命名约定(如驼峰转下划线)、逻辑删除字段识别(如deleted、del_flag)、自动填充字段标记(如create_time、update_time)等语义规则。
1、确保项目已配置正确数据源(application.yml中含spring.datasource.url等项),且数据库服务可连通。
2、在IDE中右键点击项目根目录,选择“WorkBuddy → Generate MyBatis-Plus Code”。
立即学习“Java免费学习笔记(深入)”;
3、在弹出面板中勾选目标数据表(如t_user、t_order),并指定包路径前缀(如com.example.project)。
4、点击“Generate”,WorkBuddy自动生成Entity类(含@TableId、@TableField注解)、Mapper接口(继承BaseMapper)、Service接口与Impl、以及RESTful风格Controller类。
二、融合项目上下文智能适配生成策略
WorkBuddy不依赖孤立模板,而是读取当前工程的Spring Boot版本、MyBatis-Plus依赖坐标、Lombok启用状态、以及已存在的全局配置(如id-type、table-prefix、logic-delete-field),动态调整生成逻辑。例如:当检测到pom.xml中声明了mybatis-plus-generator 3.5.15与freemarker 2.3.34,即自动选用Freemarker引擎渲染;若发现entity包下已有BaseEntity抽象类,则新生成实体将统一继承该类而非Object。
1、打开项目pom.xml,确认存在com.baomidou:mybatis-plus-generator依赖且版本不低于3.5.0。
2、检查src/main/resources目录下是否存在mybatis-plus配置片段(如mybatis-plus.configuration.map-underscore-to-camel-case=true)。
3、在WorkBuddy生成界面点击“Detect Project Context”,系统自动提取包结构、注解偏好、字段映射规则等上下文参数。
4、生成结果中,String类型字段自动添加@TableField(fill = FieldFill.INSERT)标注(若配置含fill策略),逻辑删除字段被注入@TableLogic注解。
三、支持RBAC权限模型与多租户场景的扩展生成
针对含角色权限控制或SaaS多租户架构的项目,WorkBuddy可在基础MP代码之上叠加领域特定逻辑。它识别表结构中的tenant_id字段或role_code关联关系,自动注入租户隔离拦截器配置提示、权限校验注解占位符(如@PreAuthorize("hasPermission('user:read')")),并在Service方法中预留checkPermission()调用点,避免后期补丁式开发。
1、在数据库表中确认存在tenant_id(BIGINT)或role_id(BIGINT)等典型多租户/RBAC字段。
你接手一个需求时,是不是经常这样:"改一个小字段 → 结果把不相关的代码也改了"、"需求说得很模糊 → 做着做着发现跑偏了"、"数据库加个字段 → 上线才发现漏了数据兼容"、"做了个新功能 → 一个月后自己都忘了当时怎么设计的"、"写到一半被中断 → 再回来不知道从哪继续";这个 Skill 就是专门解决这些问题的。 它会自动探测你的项目环境(JDK、Spring Boot、ORM、数据库、前端框架),记在
2、于WorkBuddy生成配置页勾选“Enable RBAC Extension”或“Enable Tenant Isolation”复选框。
3、生成的Controller方法顶部自动添加@RequiresPermissions("user:list")(Shiro)或@PreAuthorize表达式(Spring Security)。
4、生成的ServiceImpl类中,在save()和updateById()方法内插入this.checkTenantAccess(entity.getTenantId())调用桩。
四、提供模板级自定义能力,支持企业编码规范落地
WorkBuddy允许开发者上传自定义FreeMarker模板文件(如entity.java.ftl、mapper.xml.ftl),并将其绑定至特定项目。生成时,系统优先加载项目本地模板,替代默认模板,确保生成代码完全匹配团队命名规范(如DTO后缀强制为VO/DTO/QO)、注释格式(Javadoc含@author @since)、以及字段校验注解(如@NotBlank、@Min(1))的自动注入。
1、在项目根目录新建templates/mp-generator/文件夹,放入已编辑好的.ftl模板文件。
2、在WorkBuddy设置中进入“Template Management”,点击“Import Custom Templates”,指向该路径。
3、验证模板语法:WorkBuddy内置模板校验器会提示缺失变量(如${table.comment}未定义)或语法错误。
4、执行生成时,所有输出文件均按自定义模板渲染,例如原生生成的private Long id;将变为private Long id; // 主键ID,自增。
五、与IDE深度集成,实现生成后即时校验与修复
WorkBuddy通过本地语言服务器协议(LSP)客户端,在代码生成完成后立即触发静态分析:检查Mapper XML中resultMap与Entity字段是否一一对应、@SelectProvider方法签名是否匹配、ServiceImpl是否遗漏@Transactional注解。发现问题后,直接在IDE编辑器中标红高亮,并提供“Quick Fix”按钮一键修正。
1、生成完成,WorkBuddy自动扫描新生成的Mapper XML文件。
2、若发现
3、点击“Fix Now”,自动将resultType替换为com.example.project.entity.User。
4、同时检查Controller中@RequestMapping路径是否与Swagger@Api注解描述冲突,冲突处标黄提示并给出同步建议。

















