diff --git a/src/main/java/com/muyu/gen/controller/GenController.java b/src/main/java/com/muyu/gen/controller/GenController.java index b79d70c..18d6520 100644 --- a/src/main/java/com/muyu/gen/controller/GenController.java +++ b/src/main/java/com/muyu/gen/controller/GenController.java @@ -215,4 +215,14 @@ public class GenController extends BaseController public Result> selDbNameAll(){ return Result.success(genTableService.selDbNameAll()); } + + + /** + * 根据数据库名称与表名称查询表字段 + */ + @GetMapping("/selectDbTableColumnsByName/{dbName}/{table}") + public Result> selTableAll(@PathVariable("dbName") String dbName,@PathVariable("table") String table){ + List genTableColumns = genTableColumnService.selectDbTableColumnsByName(table, dbName); + return Result.success(genTableColumns); + } } diff --git a/src/main/java/com/muyu/gen/service/GenTableColumnServiceImpl.java b/src/main/java/com/muyu/gen/service/GenTableColumnServiceImpl.java index 210910f..043c83b 100644 --- a/src/main/java/com/muyu/gen/service/GenTableColumnServiceImpl.java +++ b/src/main/java/com/muyu/gen/service/GenTableColumnServiceImpl.java @@ -66,4 +66,9 @@ public class GenTableColumnServiceImpl implements IGenTableColumnService { return genTableColumnMapper.deleteGenTableColumnByIds(Convert.toLongArray(ids)); } + + @Override + public List selectDbTableColumnsByName(String table, String dbName) { + return genTableColumnMapper.selectDbTableColumnsByName(table,dbName); + } } diff --git a/src/main/java/com/muyu/gen/service/GenTableServiceImpl.java b/src/main/java/com/muyu/gen/service/GenTableServiceImpl.java index 61ffdf3..3d5ba35 100644 --- a/src/main/java/com/muyu/gen/service/GenTableServiceImpl.java +++ b/src/main/java/com/muyu/gen/service/GenTableServiceImpl.java @@ -437,6 +437,8 @@ public class GenTableServiceImpl implements IGenTableService return genTableMapper.selDbNameAll(); } + + /** * 设置主键列信息 * diff --git a/src/main/java/com/muyu/gen/service/IGenTableColumnService.java b/src/main/java/com/muyu/gen/service/IGenTableColumnService.java index 9015c70..e6d4d55 100644 --- a/src/main/java/com/muyu/gen/service/IGenTableColumnService.java +++ b/src/main/java/com/muyu/gen/service/IGenTableColumnService.java @@ -42,4 +42,6 @@ public interface IGenTableColumnService * @return 结果 */ public int deleteGenTableColumnByIds(String ids); + + List selectDbTableColumnsByName(String table, String dbName); } diff --git a/src/main/resources/mapper/generator/GenTableMapper.xml b/src/main/resources/mapper/generator/GenTableMapper.xml index a902b57..0e0af77 100644 --- a/src/main/resources/mapper/generator/GenTableMapper.xml +++ b/src/main/resources/mapper/generator/GenTableMapper.xml @@ -61,7 +61,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" - table_schema = #{dbName} + table_schema in (#{dbName}) table_schema = (select database()) @@ -152,6 +152,9 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" WHERE table_schema NOT IN ('information_schema', 'mysql', 'performance_schema', 'sys') GROUP BY table_schema + insert into gen_table (