| | |
| | | package com.ruoyi.generator.controller; |
| | | |
| | | import cn.dev33.satoken.annotation.SaCheckPermission; |
| | | import cn.hutool.core.convert.Convert; |
| | | import cn.hutool.core.io.IoUtil; |
| | | import com.ruoyi.common.annotation.Log; |
| | | import com.ruoyi.common.core.controller.BaseController; |
| | | import com.ruoyi.common.core.domain.AjaxResult; |
| | | import com.ruoyi.common.core.domain.PageQuery; |
| | | import com.ruoyi.common.core.domain.R; |
| | | import com.ruoyi.common.core.page.TableDataInfo; |
| | | import com.ruoyi.common.enums.BusinessType; |
| | | import com.ruoyi.generator.domain.GenTable; |
| | | import com.ruoyi.generator.domain.GenTableColumn; |
| | | import com.ruoyi.generator.service.IGenTableColumnService; |
| | | import com.ruoyi.generator.service.IGenTableService; |
| | | import io.swagger.annotations.Api; |
| | | import io.swagger.annotations.ApiOperation; |
| | | import io.swagger.v3.oas.annotations.tags.Tag; |
| | | import lombok.RequiredArgsConstructor; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.security.access.prepost.PreAuthorize; |
| | | import org.springframework.validation.annotation.Validated; |
| | | import org.springframework.web.bind.annotation.*; |
| | | |
| | |
| | | * @author Lion Li |
| | | */ |
| | | @Validated |
| | | @Api(value = "代码生成", tags = {"代码生成管理"}) |
| | | @RequiredArgsConstructor(onConstructor_ = @Autowired) |
| | | @Tag(name = "代码生成", description = "代码生成管理") |
| | | @RequiredArgsConstructor |
| | | @RestController |
| | | @RequestMapping("/tool/gen") |
| | | public class GenController extends BaseController { |
| | | |
| | | private final IGenTableService genTableService; |
| | | private final IGenTableColumnService genTableColumnService; |
| | | |
| | | /** |
| | | * 查询代码生成列表 |
| | | */ |
| | | @ApiOperation("查询代码生成列表") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:list')") |
| | | @SaCheckPermission("tool:gen:list") |
| | | @GetMapping("/list") |
| | | public TableDataInfo<GenTable> genList(GenTable genTable) { |
| | | return genTableService.selectPageGenTableList(genTable); |
| | | public TableDataInfo<GenTable> genList(GenTable genTable, PageQuery pageQuery) { |
| | | return genTableService.selectPageGenTableList(genTable, pageQuery); |
| | | } |
| | | |
| | | /** |
| | | * 修改代码生成业务 |
| | | */ |
| | | @ApiOperation("修改代码生成业务") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:query')") |
| | | @GetMapping(value = "/{talbleId}") |
| | | public AjaxResult<Map<String, Object>> getInfo(@PathVariable Long talbleId) { |
| | | GenTable table = genTableService.selectGenTableById(talbleId); |
| | | @SaCheckPermission("tool:gen:query") |
| | | @GetMapping(value = "/{tableId}") |
| | | public R<Map<String, Object>> getInfo(@PathVariable Long tableId) { |
| | | GenTable table = genTableService.selectGenTableById(tableId); |
| | | List<GenTable> tables = genTableService.selectGenTableAll(); |
| | | List<GenTableColumn> list = genTableColumnService.selectGenTableColumnListByTableId(talbleId); |
| | | List<GenTableColumn> list = genTableService.selectGenTableColumnListByTableId(tableId); |
| | | Map<String, Object> map = new HashMap<String, Object>(); |
| | | map.put("info", table); |
| | | map.put("rows", list); |
| | | map.put("tables", tables); |
| | | return AjaxResult.success(map); |
| | | return R.ok(map); |
| | | } |
| | | |
| | | /** |
| | | * 查询数据库列表 |
| | | */ |
| | | @ApiOperation("查询数据库列表") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:list')") |
| | | @SaCheckPermission("tool:gen:list") |
| | | @GetMapping("/db/list") |
| | | public TableDataInfo<GenTable> dataList(GenTable genTable) { |
| | | return genTableService.selectPageDbTableList(genTable); |
| | | public TableDataInfo<GenTable> dataList(GenTable genTable, PageQuery pageQuery) { |
| | | return genTableService.selectPageDbTableList(genTable, pageQuery); |
| | | } |
| | | |
| | | /** |
| | | * 查询数据表字段列表 |
| | | */ |
| | | @ApiOperation("查询数据表字段列表") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:list')") |
| | | @GetMapping(value = "/column/{talbleId}") |
| | | @SaCheckPermission("tool:gen:list") |
| | | @GetMapping(value = "/column/{tableId}") |
| | | public TableDataInfo<GenTableColumn> columnList(Long tableId) { |
| | | TableDataInfo<GenTableColumn> dataInfo = new TableDataInfo<>(); |
| | | List<GenTableColumn> list = genTableColumnService.selectGenTableColumnListByTableId(tableId); |
| | | List<GenTableColumn> list = genTableService.selectGenTableColumnListByTableId(tableId); |
| | | dataInfo.setRows(list); |
| | | dataInfo.setTotal(list.size()); |
| | | return dataInfo; |
| | |
| | | /** |
| | | * 导入表结构(保存) |
| | | */ |
| | | @ApiOperation("导入表结构(保存)") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:import')") |
| | | @SaCheckPermission("tool:gen:import") |
| | | @Log(title = "代码生成", businessType = BusinessType.IMPORT) |
| | | @PostMapping("/importTable") |
| | | public AjaxResult<Void> importTableSave(String tables) { |
| | | public R<Void> importTableSave(String tables) { |
| | | String[] tableNames = Convert.toStrArray(tables); |
| | | // 查询表信息 |
| | | List<GenTable> tableList = genTableService.selectDbTableListByNames(tableNames); |
| | | genTableService.importGenTable(tableList); |
| | | return AjaxResult.success(); |
| | | return R.ok(); |
| | | } |
| | | |
| | | /** |
| | | * 修改保存代码生成业务 |
| | | */ |
| | | @ApiOperation("修改保存代码生成业务") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:edit')") |
| | | @SaCheckPermission("tool:gen:edit") |
| | | @Log(title = "代码生成", businessType = BusinessType.UPDATE) |
| | | @PutMapping |
| | | public AjaxResult<Void> editSave(@Validated @RequestBody GenTable genTable) { |
| | | public R<Void> editSave(@Validated @RequestBody GenTable genTable) { |
| | | genTableService.validateEdit(genTable); |
| | | genTableService.updateGenTable(genTable); |
| | | return AjaxResult.success(); |
| | | return R.ok(); |
| | | } |
| | | |
| | | /** |
| | | * 删除代码生成 |
| | | */ |
| | | @ApiOperation("删除代码生成") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:remove')") |
| | | @SaCheckPermission("tool:gen:remove") |
| | | @Log(title = "代码生成", businessType = BusinessType.DELETE) |
| | | @DeleteMapping("/{tableIds}") |
| | | public AjaxResult<Void> remove(@PathVariable Long[] tableIds) { |
| | | public R<Void> remove(@PathVariable Long[] tableIds) { |
| | | genTableService.deleteGenTableByIds(tableIds); |
| | | return AjaxResult.success(); |
| | | return R.ok(); |
| | | } |
| | | |
| | | /** |
| | | * 预览代码 |
| | | */ |
| | | @ApiOperation("预览代码") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:preview')") |
| | | @SaCheckPermission("tool:gen:preview") |
| | | @GetMapping("/preview/{tableId}") |
| | | public AjaxResult<Map<String, String>> preview(@PathVariable("tableId") Long tableId) throws IOException { |
| | | public R<Map<String, String>> preview(@PathVariable("tableId") Long tableId) throws IOException { |
| | | Map<String, String> dataMap = genTableService.previewCode(tableId); |
| | | return AjaxResult.success(dataMap); |
| | | return R.ok(dataMap); |
| | | } |
| | | |
| | | /** |
| | | * 生成代码(下载方式) |
| | | */ |
| | | @ApiOperation("生成代码(下载方式)") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:code')") |
| | | @SaCheckPermission("tool:gen:code") |
| | | @Log(title = "代码生成", businessType = BusinessType.GENCODE) |
| | | @GetMapping("/download/{tableName}") |
| | | public void download(HttpServletResponse response, @PathVariable("tableName") String tableName) throws IOException { |
| | |
| | | /** |
| | | * 生成代码(自定义路径) |
| | | */ |
| | | @ApiOperation("生成代码(自定义路径)") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:code')") |
| | | @SaCheckPermission("tool:gen:code") |
| | | @Log(title = "代码生成", businessType = BusinessType.GENCODE) |
| | | @GetMapping("/genCode/{tableName}") |
| | | public AjaxResult<Void> genCode(@PathVariable("tableName") String tableName) { |
| | | public R<Void> genCode(@PathVariable("tableName") String tableName) { |
| | | genTableService.generatorCode(tableName); |
| | | return AjaxResult.success(); |
| | | return R.ok(); |
| | | } |
| | | |
| | | /** |
| | | * 同步数据库 |
| | | */ |
| | | @ApiOperation("同步数据库") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:edit')") |
| | | @SaCheckPermission("tool:gen:edit") |
| | | @Log(title = "代码生成", businessType = BusinessType.UPDATE) |
| | | @GetMapping("/synchDb/{tableName}") |
| | | public AjaxResult<Void> synchDb(@PathVariable("tableName") String tableName) { |
| | | public R<Void> synchDb(@PathVariable("tableName") String tableName) { |
| | | genTableService.synchDb(tableName); |
| | | return AjaxResult.success(); |
| | | return R.ok(); |
| | | } |
| | | |
| | | /** |
| | | * 批量生成代码 |
| | | */ |
| | | @ApiOperation("批量生成代码") |
| | | @PreAuthorize("@ss.hasPermi('tool:gen:code')") |
| | | @SaCheckPermission("tool:gen:code") |
| | | @Log(title = "代码生成", businessType = BusinessType.GENCODE) |
| | | @GetMapping("/batchGenCode") |
| | | public void batchGenCode(HttpServletResponse response, String tables) throws IOException { |