diff --git a/ruoyi-modules/ruoyi-generator/src/main/java/org/ruoyi/generator/controller/GenController.java b/ruoyi-modules/ruoyi-generator/src/main/java/org/ruoyi/generator/controller/GenController.java index 11a489dd..13382de7 100644 --- a/ruoyi-modules/ruoyi-generator/src/main/java/org/ruoyi/generator/controller/GenController.java +++ b/ruoyi-modules/ruoyi-generator/src/main/java/org/ruoyi/generator/controller/GenController.java @@ -8,6 +8,7 @@ import org.ruoyi.generator.service.IGenTableService; import org.ruoyi.generator.service.SchemaFieldService; import org.springframework.context.annotation.Profile; import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; @@ -40,8 +41,8 @@ public class GenController extends BaseController { * * @param tableNameStr 表名 */ - @GetMapping("/batchGenCode") - public R batchGenCode(String tableNameStr) { + @GetMapping("/batchGenCode/{tableNameStr}") + public R batchGenCode(@PathVariable("tableNameStr") String tableNameStr) { genTableService.generateCodeToClasspathByTableNames(tableNameStr); return R.ok("代码生成成功"); } diff --git a/ruoyi-modules/ruoyi-generator/src/main/java/org/ruoyi/generator/impl/GenTableServiceImpl.java b/ruoyi-modules/ruoyi-generator/src/main/java/org/ruoyi/generator/impl/GenTableServiceImpl.java index f37271b6..55532f13 100644 --- a/ruoyi-modules/ruoyi-generator/src/main/java/org/ruoyi/generator/impl/GenTableServiceImpl.java +++ b/ruoyi-modules/ruoyi-generator/src/main/java/org/ruoyi/generator/impl/GenTableServiceImpl.java @@ -289,24 +289,12 @@ public class GenTableServiceImpl implements IGenTableService { */ private void addImportForJavaType(String javaType, Set importList) { switch (javaType) { - case "BigDecimal": - importList.add("java.math.BigDecimal"); - break; - case "Date": - importList.add("java.util.Date"); - break; - case "LocalDateTime": - importList.add("java.time.LocalDateTime"); - break; - case "LocalDate": - importList.add("java.time.LocalDate"); - break; - case "LocalTime": - importList.add("java.time.LocalTime"); - break; - default: - // 基本类型不需要导入 - break; + case "BigDecimal" -> importList.add("java.math.BigDecimal"); + case "Date" -> importList.add("java.util.Date"); + case "LocalDateTime" -> importList.add("java.time.LocalDateTime"); + case "LocalDate" -> importList.add("java.time.LocalDate"); + case "LocalTime" -> importList.add("java.time.LocalTime"); + default -> {} } }