|
|
@@ -1704,4 +1704,40 @@ public class FreeMarkerGeneratorTest {
|
|
|
}
|
|
|
|
|
|
|
|
|
+ @Test
|
|
|
+ public void gcRoomRule() throws IOException {
|
|
|
+ List<TableConfig> tableConfigs = new ArrayList<>();
|
|
|
+ TableConfig mainTable = new TableConfig();
|
|
|
+ mainTable.setTableName("room_rule");//init_sql中的表名
|
|
|
+ mainTable.setIsMain(true);//是否是主表,一般默认为true
|
|
|
+ mainTable.setPkField(GlobalConstant.DEFAULT_PK);//设置主键
|
|
|
+ mainTable.setPkType(GlobalConstant.DEFAULT_PK_TYPE);//设置主键类型
|
|
|
+ tableConfigs.add(mainTable);
|
|
|
+
|
|
|
+ TableConfig childTable1 = new TableConfig();
|
|
|
+ childTable1.setTableName("room_rule_item");
|
|
|
+ childTable1.setIsMain(false);
|
|
|
+ childTable1.setPkField(GlobalConstant.DEFAULT_PK);
|
|
|
+ childTable1.setPkType(GlobalConstant.DEFAULT_PK_TYPE);
|
|
|
+ childTable1.setRelationField("room_rule_id");
|
|
|
+ childTable1.setRelationTableField(GlobalConstant.DEFAULT_PK);
|
|
|
+ tableConfigs.add(childTable1);
|
|
|
+
|
|
|
+
|
|
|
+ ApiGenerateCodesDto params = new ApiGenerateCodesDto();
|
|
|
+ params.setAuthor("dzx");//作者名称
|
|
|
+ params.setPackageName("room");//包名
|
|
|
+ params.setTableConfigs(tableConfigs);
|
|
|
+ params.setPage(true);//是否生成分页接口
|
|
|
+ params.setImport(false);//是否生成导入接口
|
|
|
+ params.setExport(false);//是否生成导出接口
|
|
|
+ params.setOutMainDir(true);//是否生成在主目录,前期测试可设置成false
|
|
|
+ params.setDs(ds);
|
|
|
+
|
|
|
+ IApiGeneratorService apiGeneratorService = new ApiGeneratorServiceImpl();
|
|
|
+
|
|
|
+ apiGeneratorService.generateCodes(params);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
}
|