feat: 调整生成代码结构

This commit is contained in:
l90215
2025-07-29 23:34:39 +08:00
parent 6b2cf1a3c3
commit 897222c41c
2 changed files with 9 additions and 20 deletions

View File

@@ -8,6 +8,7 @@ import org.ruoyi.generator.service.IGenTableService;
import org.ruoyi.generator.service.SchemaFieldService; import org.ruoyi.generator.service.SchemaFieldService;
import org.springframework.context.annotation.Profile; import org.springframework.context.annotation.Profile;
import org.springframework.web.bind.annotation.GetMapping; 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.RequestMapping;
import org.springframework.web.bind.annotation.RestController; import org.springframework.web.bind.annotation.RestController;
@@ -40,8 +41,8 @@ public class GenController extends BaseController {
* *
* @param tableNameStr 表名 * @param tableNameStr 表名
*/ */
@GetMapping("/batchGenCode") @GetMapping("/batchGenCode/{tableNameStr}")
public R<String> batchGenCode(String tableNameStr) { public R<String> batchGenCode(@PathVariable("tableNameStr") String tableNameStr) {
genTableService.generateCodeToClasspathByTableNames(tableNameStr); genTableService.generateCodeToClasspathByTableNames(tableNameStr);
return R.ok("代码生成成功"); return R.ok("代码生成成功");
} }

View File

@@ -289,24 +289,12 @@ public class GenTableServiceImpl implements IGenTableService {
*/ */
private void addImportForJavaType(String javaType, Set<String> importList) { private void addImportForJavaType(String javaType, Set<String> importList) {
switch (javaType) { switch (javaType) {
case "BigDecimal": case "BigDecimal" -> importList.add("java.math.BigDecimal");
importList.add("java.math.BigDecimal"); case "Date" -> importList.add("java.util.Date");
break; case "LocalDateTime" -> importList.add("java.time.LocalDateTime");
case "Date": case "LocalDate" -> importList.add("java.time.LocalDate");
importList.add("java.util.Date"); case "LocalTime" -> importList.add("java.time.LocalTime");
break; default -> {}
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;
} }
} }