GenTableServiceImpl.java 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581
  1. package com.boman.gen.service;
  2. import java.io.ByteArrayOutputStream;
  3. import java.io.File;
  4. import java.io.IOException;
  5. import java.io.StringWriter;
  6. import java.util.LinkedHashMap;
  7. import java.util.List;
  8. import java.util.Map;
  9. import java.util.stream.Collectors;
  10. import java.util.zip.ZipEntry;
  11. import java.util.zip.ZipOutputStream;
  12. import com.boman.common.core.constant.UserConstants;
  13. import com.boman.common.core.utils.DateUtils;
  14. import com.boman.common.core.web.domain.AjaxResult;
  15. import com.boman.system.api.RemoteSysMenuService;
  16. import com.boman.system.api.domain.SysMenu;
  17. import org.apache.commons.io.IOUtils;
  18. import org.apache.velocity.Template;
  19. import org.apache.velocity.VelocityContext;
  20. import org.apache.velocity.app.Velocity;
  21. import org.slf4j.Logger;
  22. import org.slf4j.LoggerFactory;
  23. import org.springframework.beans.factory.annotation.Autowired;
  24. import org.springframework.stereotype.Service;
  25. import org.springframework.transaction.annotation.Transactional;
  26. import com.alibaba.fastjson.JSON;
  27. import com.alibaba.fastjson.JSONObject;
  28. import com.boman.common.core.constant.Constants;
  29. import com.boman.common.core.constant.GenConstants;
  30. import com.boman.common.core.exception.CustomException;
  31. import com.boman.common.core.text.CharsetKit;
  32. import com.boman.common.core.utils.SecurityUtils;
  33. import com.boman.common.core.utils.StringUtils;
  34. import com.boman.common.core.utils.file.FileUtils;
  35. import com.boman.gen.domain.GenTable;
  36. import com.boman.gen.domain.GenTableColumn;
  37. import com.boman.gen.mapper.GenTableColumnMapper;
  38. import com.boman.gen.mapper.GenTableMapper;
  39. import com.boman.gen.util.GenUtils;
  40. import com.boman.gen.util.VelocityInitializer;
  41. import com.boman.gen.util.VelocityUtils;
  42. /**
  43. * 业务 服务层实现
  44. *
  45. * @author ruoyi
  46. */
  47. @Service
  48. public class GenTableServiceImpl implements IGenTableService {
  49. private static final Logger log = LoggerFactory.getLogger(GenTableServiceImpl.class);
  50. @Autowired
  51. private GenTableMapper genTableMapper;
  52. @Autowired
  53. private GenTableColumnMapper genTableColumnMapper;
  54. @Autowired
  55. private RemoteSysMenuService remoteSysMenuService;
  56. /**
  57. * 查询业务信息
  58. *
  59. * @param id 业务ID
  60. * @return 业务信息
  61. */
  62. @Override
  63. public GenTable selectGenTableById(Long id) {
  64. GenTable genTable = genTableMapper.selectGenTableById(id);
  65. setTableFromOptions(genTable);
  66. return genTable;
  67. }
  68. /**
  69. * 查询业务列表
  70. *
  71. * @param genTable 业务信息
  72. * @return 业务集合
  73. */
  74. @Override
  75. public List<GenTable> selectGenTableList(GenTable genTable) {
  76. List<GenTable> genTables = genTableMapper.selectGenTableList(genTable);
  77. return genTables;
  78. }
  79. /**
  80. * 查询据库列表
  81. *
  82. * @param genTable 业务信息
  83. * @return 数据库表集合
  84. */
  85. @Override
  86. public List<GenTable> selectDbTableList(GenTable genTable) {
  87. return genTableMapper.selectDbTableList(genTable);
  88. }
  89. /**
  90. * 查询据库列表
  91. *
  92. * @param tableNames 表名称组
  93. * @return 数据库表集合
  94. */
  95. @Override
  96. public List<GenTable> selectDbTableListByNames(String[] tableNames) {
  97. return genTableMapper.selectDbTableListByNames(tableNames);
  98. }
  99. /**
  100. * 查询表名称业务信息
  101. *
  102. * @param tableName 表名称
  103. * @return 业务信息
  104. */
  105. @Override
  106. public GenTable selectGenTableByName(String tableName) {
  107. return genTableMapper.selectGenTableByName(tableName);
  108. }
  109. /**
  110. * 查询所有表信息
  111. *
  112. * @return 表信息集合
  113. */
  114. @Override
  115. public List<GenTable> selectGenTableAll() {
  116. return genTableMapper.selectGenTableAll();
  117. }
  118. /**
  119. * 修改业务
  120. *
  121. * @param genTable 业务信息
  122. * @return 结果
  123. */
  124. @Override
  125. @Transactional
  126. public void updateGenTable(GenTable genTable) {
  127. String options = JSON.toJSONString(genTable.getParams());
  128. genTable.setOptions(options);
  129. int row = genTableMapper.updateGenTable(genTable);
  130. if (row > 0) {
  131. for (GenTableColumn cenTableColumn : genTable.getColumns()) {
  132. genTableColumnMapper.updateGenTableColumn(cenTableColumn);
  133. }
  134. }
  135. }
  136. /**
  137. * 删除业务对象
  138. *
  139. * @param tableIds 需要删除的数据ID
  140. * @return 结果
  141. */
  142. @Override
  143. @Transactional
  144. public void deleteGenTableByIds(Long[] tableIds) {
  145. genTableMapper.deleteGenTableByIds(tableIds);
  146. genTableColumnMapper.deleteGenTableColumnByIds(tableIds);
  147. }
  148. /**
  149. * 导入表结构
  150. *
  151. * @param tableList 导入表列表
  152. */
  153. @Override
  154. @Transactional
  155. public void importGenTable(List<GenTable> tableList) {
  156. String operName = SecurityUtils.getUsername();
  157. try {
  158. for (GenTable table : tableList) {
  159. String tableName = table.getTableName();
  160. GenUtils.initTable(table, operName);
  161. int row = genTableMapper.insertGenTable(table);
  162. if (row > 0) {
  163. // 保存列信息
  164. List<GenTableColumn> genTableColumns = genTableColumnMapper.selectDbTableColumnsByName(tableName);
  165. for (GenTableColumn column : genTableColumns) {
  166. GenUtils.initColumnField(column, table);
  167. genTableColumnMapper.insertGenTableColumn(column);
  168. }
  169. }
  170. }
  171. } catch (Exception e) {
  172. throw new CustomException("导入失败:" + e.getMessage());
  173. }
  174. }
  175. /**
  176. * 预览代码
  177. *
  178. * @param tableId 表编号
  179. * @return 预览数据列表
  180. */
  181. @Override
  182. public Map<String, String> previewCode(Long tableId) {
  183. Map<String, String> dataMap = new LinkedHashMap<>();
  184. // 查询表信息
  185. GenTable table = genTableMapper.selectGenTableById(tableId);
  186. // 设置主子表信息
  187. setSubTable(table);
  188. // 设置主键列信息
  189. setPkColumn(table);
  190. VelocityInitializer.initVelocity();
  191. VelocityContext context = VelocityUtils.prepareContext(table);
  192. // 获取模板列表
  193. List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory());
  194. for (String template : templates) {
  195. // 渲染模板
  196. StringWriter sw = new StringWriter();
  197. Template tpl = Velocity.getTemplate(template, Constants.UTF8);
  198. tpl.merge(context, sw);
  199. dataMap.put(template, sw.toString());
  200. }
  201. return dataMap;
  202. }
  203. /**
  204. * 生成代码(下载方式)
  205. *
  206. * @param tableName 表名称
  207. * @return 数据
  208. */
  209. @Override
  210. public byte[] downloadCode(String tableName) {
  211. ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
  212. ZipOutputStream zip = new ZipOutputStream(outputStream);
  213. generatorCode(tableName, zip);
  214. IOUtils.closeQuietly(zip);
  215. return outputStream.toByteArray();
  216. }
  217. /**
  218. * 生成代码(自定义路径)
  219. *
  220. * @param tableName 表名称
  221. */
  222. @Override
  223. public void generatorCode(String tableName) {
  224. // 查询表信息
  225. GenTable table = genTableMapper.selectGenTableByName(tableName);
  226. // 设置主子表信息
  227. setSubTable(table);
  228. // 设置主键列信息
  229. setPkColumn(table);
  230. VelocityInitializer.initVelocity();
  231. VelocityContext context = VelocityUtils.prepareContext(table);
  232. // 获取模板列表
  233. List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory());
  234. for (String template : templates) {
  235. if (!StringUtils.containsAny(template, "sql.vm", "api.js.vm", "index.vue.vm", "index-tree.vue.vm")) {
  236. // 渲染模板
  237. StringWriter sw = new StringWriter();
  238. Template tpl = Velocity.getTemplate(template, Constants.UTF8);
  239. tpl.merge(context, sw);
  240. try {
  241. String path = getGenPath(table, template);
  242. FileUtils.writeStringToFile(new File(path), sw.toString(), CharsetKit.UTF_8);
  243. } catch (IOException e) {
  244. throw new CustomException("渲染模板失败,表名:" + table.getTableName());
  245. }
  246. }
  247. }
  248. }
  249. /**
  250. * 同步数据库
  251. *
  252. * @param tableName 表名称
  253. */
  254. @Override
  255. @Transactional(rollbackFor = Exception.class)
  256. public void synchDb(String tableName) {
  257. GenTable table = genTableMapper.selectGenTableByName(tableName);
  258. //获取表字段集合
  259. List<GenTableColumn> tableColumns = table.getColumns();
  260. //获取列名称集合
  261. List<String> tableColumnNames = tableColumns.stream().map(GenTableColumn::getColumnName).collect(Collectors.toList());
  262. //根据表名称查询原列信息
  263. List<GenTableColumn> dbTableColumns = genTableColumnMapper.selectDbTableColumnsByName(tableName);
  264. if (StringUtils.isEmpty(dbTableColumns)) {
  265. throw new CustomException("同步数据失败,原表结构不存在");
  266. }
  267. //获取原表列名称集合
  268. List<String> dbTableColumnNames = dbTableColumns.stream().map(GenTableColumn::getColumnName).collect(Collectors.toList());
  269. dbTableColumns.forEach(column -> {
  270. if (!tableColumnNames.contains(column.getColumnName())) {
  271. GenUtils.initColumnField(column, table);
  272. genTableColumnMapper.insertGenTableColumn(column);
  273. }
  274. });
  275. List<GenTableColumn> delColumns = tableColumns.stream().filter(column -> !dbTableColumnNames.contains(column.getColumnName())).collect(Collectors.toList());
  276. if (StringUtils.isNotEmpty(delColumns)) {
  277. genTableColumnMapper.deleteGenTableColumns(delColumns);
  278. }
  279. }
  280. /**
  281. * 批量生成代码(下载方式)
  282. *
  283. * @param tableNames 表数组
  284. * @return 数据
  285. */
  286. @Override
  287. public byte[] downloadCode(String[] tableNames) {
  288. ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
  289. ZipOutputStream zip = new ZipOutputStream(outputStream);
  290. for (String tableName : tableNames) {
  291. generatorCode(tableName, zip);
  292. }
  293. IOUtils.closeQuietly(zip);
  294. return outputStream.toByteArray();
  295. }
  296. /**
  297. * 查询表信息并生成代码
  298. */
  299. private void generatorCode(String tableName, ZipOutputStream zip) {
  300. // 查询表信息
  301. GenTable table = genTableMapper.selectGenTableByName(tableName);
  302. // 设置主子表信息
  303. setSubTable(table);
  304. // 设置主键列信息
  305. setPkColumn(table);
  306. VelocityInitializer.initVelocity();
  307. VelocityContext context = VelocityUtils.prepareContext(table);
  308. // 获取模板列表
  309. List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory());
  310. for (String template : templates) {
  311. // 渲染模板
  312. StringWriter sw = new StringWriter();
  313. Template tpl = Velocity.getTemplate(template, Constants.UTF8);
  314. tpl.merge(context, sw);
  315. try {
  316. // 添加到zip
  317. zip.putNextEntry(new ZipEntry(VelocityUtils.getFileName(template, table)));
  318. IOUtils.write(sw.toString(), zip, Constants.UTF8);
  319. IOUtils.closeQuietly(sw);
  320. zip.flush();
  321. zip.closeEntry();
  322. } catch (IOException e) {
  323. log.error("渲染模板失败,表名:" + table.getTableName(), e);
  324. }
  325. }
  326. }
  327. /**
  328. * 修改保存参数校验
  329. *
  330. * @param genTable 业务信息
  331. */
  332. @Override
  333. public void validateEdit(GenTable genTable) {
  334. if (GenConstants.TPL_TREE.equals(genTable.getTplCategory())) {
  335. String options = JSON.toJSONString(genTable.getParams());
  336. JSONObject paramsObj = JSONObject.parseObject(options);
  337. if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_CODE))) {
  338. throw new CustomException("树编码字段不能为空");
  339. } else if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_PARENT_CODE))) {
  340. throw new CustomException("树父编码字段不能为空");
  341. } else if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_NAME))) {
  342. throw new CustomException("树名称字段不能为空");
  343. } else if (GenConstants.TPL_SUB.equals(genTable.getTplCategory())) {
  344. if (StringUtils.isEmpty(genTable.getSubTableName())) {
  345. throw new CustomException("关联子表的表名不能为空");
  346. } else if (StringUtils.isEmpty(genTable.getSubTableFkName())) {
  347. throw new CustomException("子表关联的外键名不能为空");
  348. }
  349. }
  350. }
  351. }
  352. /**
  353. * 代码生成业务表新增
  354. * 注意解决事务问题
  355. *
  356. * @param genTable
  357. * @return
  358. */
  359. @Override
  360. public AjaxResult insertGenTable(GenTable genTable) {
  361. //新增表成功的时候,新增字段公共字段
  362. int i = genTableMapper.insertGenTable(genTable);
  363. if (i > 0) {
  364. insertCommonColumn(genTable.getTableId());
  365. //判断是否是菜单
  366. isMenu(genTable);
  367. return AjaxResult.success();
  368. }
  369. return AjaxResult.error();
  370. }
  371. /**
  372. * 新建表时新增对应公共字段
  373. *
  374. * @param tableId
  375. */
  376. public void insertCommonColumn(Long tableId) {
  377. GenTableColumn genTableColumnLog = new GenTableColumn();
  378. GenTableColumn genTableColumnBaseInfo = new GenTableColumn();
  379. genTableColumnLog.setTableId(tableId);
  380. genTableColumnLog.setColumnName("log_info");
  381. genTableColumnLog.setColumnComment("日志信息");
  382. genTableColumnLog.setColumnType("varchar(255)");
  383. genTableColumnLog.setIsPk("0");
  384. genTableColumnLog.setIsIncrement("0");
  385. genTableColumnLog.setIsInsert("1");
  386. genTableColumnLog.setQueryType("LIKE");
  387. genTableColumnLog.setHtmlType("input");
  388. genTableColumnLog.setSort(7);
  389. genTableColumnLog.setCreateBy("admin");
  390. genTableColumnLog.setCreateTime(DateUtils.getNowDate());
  391. genTableColumnMapper.insertGenTableColumn(genTableColumnLog);
  392. genTableColumnBaseInfo.setTableId(tableId);
  393. genTableColumnBaseInfo.setColumnName("base_info");
  394. genTableColumnBaseInfo.setColumnComment("基本信息");
  395. genTableColumnBaseInfo.setColumnType("varchar(255)");
  396. genTableColumnBaseInfo.setIsPk("0");
  397. genTableColumnBaseInfo.setIsIncrement("0");
  398. genTableColumnBaseInfo.setIsInsert("1");
  399. genTableColumnBaseInfo.setQueryType("LIKE");
  400. genTableColumnBaseInfo.setHtmlType("input");
  401. genTableColumnBaseInfo.setSort(8);
  402. genTableColumnBaseInfo.setCreateBy("admin");
  403. genTableColumnBaseInfo.setCreateTime(DateUtils.getNowDate());
  404. genTableColumnMapper.insertGenTableColumn(genTableColumnBaseInfo);
  405. String[] columnName = new String[]{"id", "create_by", "create_time", "update_by", "update_time", "is_del"};
  406. String[] columnComment = new String[]{"编号","创建者", "创建时间", "更新者", "更新时间", "是否删除"};
  407. String[] columnType = new String[]{"bigint(20)","varchar(64)", "datetime", "varchar(64)", "datetime", "char(1)"};
  408. String[] htmlType = new String[]{"input", "input", "datetime", "input", "datetime", "select"};
  409. for (int i = 0; i < columnName.length; i++) {
  410. GenTableColumn genTableColumn = new GenTableColumn();
  411. genTableColumn.setTableId(tableId);
  412. genTableColumn.setColumnName(columnName[i]);
  413. genTableColumn.setColumnComment(columnComment[i]);
  414. genTableColumn.setColumnType(columnType[i]);
  415. genTableColumn.setIsPk("0");
  416. genTableColumn.setIsIncrement("0");
  417. genTableColumn.setIsInsert("1");
  418. genTableColumn.setQueryType("EQ");
  419. genTableColumn.setHtmlType(htmlType[i]);
  420. genTableColumn.setSort(i + 1);
  421. genTableColumn.setCreateBy("admin");
  422. genTableColumn.setCreateTime(DateUtils.getNowDate());
  423. genTableColumn.setHrParentId(genTableColumnBaseInfo.getColumnId());
  424. if (i == 0){
  425. genTableColumn.setIsIncrement("1");
  426. genTableColumn.setHrParentId(genTableColumnLog.getColumnId());
  427. }
  428. genTableColumnMapper.insertGenTableColumn(genTableColumn);
  429. }
  430. }
  431. /**
  432. * 判断是否是菜单
  433. * @param genTable
  434. */
  435. public void isMenu(GenTable genTable) {
  436. String isMenu = genTable.getIsMenu();
  437. if (StringUtils.isNotBlank(isMenu) && isMenu.equals(UserConstants.NOT_UNIQUE)) {
  438. //先生成主菜单
  439. SysMenu sysMenu = new SysMenu();
  440. //表描述当菜单名称
  441. sysMenu.setMenuName(genTable.getTableComment());
  442. sysMenu.setParentId(0L);
  443. sysMenu.setOrderNum("1");
  444. //路由地址
  445. sysMenu.setPath(genTable.getBusinessName());
  446. sysMenu.setComponent(genTable.getModuleName() + "/" + genTable.getTableComment() + "/index");
  447. sysMenu.setMenuType("C");
  448. sysMenu.setIcon("system");
  449. sysMenu.setCreateBy(SecurityUtils.getUsername());
  450. sysMenu.setSysTableName(genTable.getTableName());
  451. remoteSysMenuService.add(sysMenu);
  452. //是菜单,生成crud按钮
  453. for (int i = 0; i < Constants.MENU_NAME.length; i++) {
  454. String[] menuName = Constants.MENU_NAME;
  455. SysMenu menu = new SysMenu();
  456. menu.setParentId(sysMenu.getId());
  457. menu.setMenuName(menuName[i]);
  458. menu.setOrderNum(String.valueOf(i+1));
  459. menu.setMenuType("F");
  460. menu.setIcon("#");
  461. sysMenu.setCreateBy(SecurityUtils.getUsername());
  462. remoteSysMenuService.add(menu);
  463. }
  464. }
  465. }
  466. /**
  467. * 设置主键列信息
  468. *
  469. * @param table 业务表信息
  470. */
  471. public void setPkColumn(GenTable table) {
  472. for (GenTableColumn column : table.getColumns()) {
  473. if (column.isPk()) {
  474. table.setPkColumn(column);
  475. break;
  476. }
  477. }
  478. if (StringUtils.isNull(table.getPkColumn())) {
  479. table.setPkColumn(table.getColumns().get(0));
  480. }
  481. if (GenConstants.TPL_SUB.equals(table.getTplCategory())) {
  482. for (GenTableColumn column : table.getSubTable().getColumns()) {
  483. if (column.isPk()) {
  484. table.getSubTable().setPkColumn(column);
  485. break;
  486. }
  487. }
  488. if (StringUtils.isNull(table.getSubTable().getPkColumn())) {
  489. table.getSubTable().setPkColumn(table.getSubTable().getColumns().get(0));
  490. }
  491. }
  492. }
  493. /**
  494. * 设置主子表信息
  495. *
  496. * @param table 业务表信息
  497. */
  498. public void setSubTable(GenTable table) {
  499. String subTableName = table.getSubTableName();
  500. if (StringUtils.isNotEmpty(subTableName)) {
  501. table.setSubTable(genTableMapper.selectGenTableByName(subTableName));
  502. }
  503. }
  504. /**
  505. * 设置代码生成其他选项值
  506. *
  507. * @param genTable 设置后的生成对象
  508. */
  509. public void setTableFromOptions(GenTable genTable) {
  510. JSONObject paramsObj = JSONObject.parseObject(genTable.getOptions());
  511. if (StringUtils.isNotNull(paramsObj)) {
  512. String treeCode = paramsObj.getString(GenConstants.TREE_CODE);
  513. String treeParentCode = paramsObj.getString(GenConstants.TREE_PARENT_CODE);
  514. String treeName = paramsObj.getString(GenConstants.TREE_NAME);
  515. String parentMenuId = paramsObj.getString(GenConstants.PARENT_MENU_ID);
  516. String parentMenuName = paramsObj.getString(GenConstants.PARENT_MENU_NAME);
  517. genTable.setTreeCode(treeCode);
  518. genTable.setTreeParentCode(treeParentCode);
  519. genTable.setTreeName(treeName);
  520. genTable.setParentMenuId(parentMenuId);
  521. genTable.setParentMenuName(parentMenuName);
  522. }
  523. }
  524. /**
  525. * 获取代码生成地址
  526. *
  527. * @param table 业务表信息
  528. * @param template 模板文件路径
  529. * @return 生成地址
  530. */
  531. public static String getGenPath(GenTable table, String template) {
  532. String genPath = table.getGenPath();
  533. if (StringUtils.equals(genPath, "/")) {
  534. return System.getProperty("user.dir") + File.separator + "src" + File.separator + VelocityUtils.getFileName(template, table);
  535. }
  536. return genPath + File.separator + VelocityUtils.getFileName(template, table);
  537. }
  538. }