diff --git a/src/main/java/com/zhangmeng/tools/controller/MybatisPlusGenerator.java b/src/main/java/com/zhangmeng/tools/controller/MybatisPlusGenerator.java
index 7a15d2c..5128ac8 100644
--- a/src/main/java/com/zhangmeng/tools/controller/MybatisPlusGenerator.java
+++ b/src/main/java/com/zhangmeng/tools/controller/MybatisPlusGenerator.java
@@ -226,15 +226,16 @@ public class MybatisPlusGenerator {
                 }).strategyConfig(builder -> {
                     builder.addInclude(dataSourceInfo.getTables()) // 设置需要生成的表名
                             .addTablePrefix(dataSourceInfo.getTable_prefixes())
+                            .enableCapitalMode()
                             .entityBuilder()
+                            .idType(IdType.AUTO)
                             .enableChainModel()
                             .enableLombok()
                             .enableRemoveIsPrefix()
                             .enableTableFieldAnnotation()
                             .enableActiveRecord()
-                            .naming(NamingStrategy.no_change)
-                            .columnNaming(NamingStrategy.underline_to_camel)
-                            .idType(IdType.AUTO)
+                            .controllerBuilder()
+                            .enableRestStyle()
                             .build();
                 }).templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                 .execute();