优化代码

RuoYi

2025-03-05 3dcee7057dee0a67204dbf528cd4211a925b6142

 ruoyi-modules/ruoyi-gen/src/main/java/com/ruoyi/gen/controller/GenController.java

@@ -22,7 +22,8 @@
import com.ruoyi.common.core.web.page.TableDataInfo;

import com.ruoyi.common.log.annotation.Log;

import com.ruoyi.common.log.enums.BusinessType;

import com.ruoyi.common.security.annotation.PreAuthorize;

import com.ruoyi.common.security.annotation.RequiresPermissions;

import com.ruoyi.gen.config.GenConfig;

import com.ruoyi.gen.domain.GenTable;

import com.ruoyi.gen.domain.GenTableColumn;

import com.ruoyi.gen.service.IGenTableColumnService;

@@ -46,7 +47,7 @@
    /**

     * 查询代码生成列表

     */

    @PreAuthorize(hasPermi = "tool:gen:list")

    @RequiresPermissions("tool:gen:list")

    @GetMapping("/list")

    public TableDataInfo genList(GenTable genTable)

    {

@@ -58,22 +59,24 @@
    /**

     * 修改代码生成业务

     */

    @PreAuthorize(hasPermi = "tool:gen:query")

    @GetMapping(value = "/{talbleId}")

    public AjaxResult getInfo(@PathVariable Long talbleId)

    @RequiresPermissions("tool:gen:query")

    @GetMapping(value = "/{tableId}")

    public AjaxResult getInfo(@PathVariable Long tableId)

    {

        GenTable table = genTableService.selectGenTableById(talbleId);

        List<GenTableColumn> list = genTableColumnService.selectGenTableColumnListByTableId(talbleId);

        GenTable table = genTableService.selectGenTableById(tableId);

        List<GenTable> tables = genTableService.selectGenTableAll();

        List<GenTableColumn> list = genTableColumnService.selectGenTableColumnListByTableId(tableId);

        Map<String, Object> map = new HashMap<String, Object>();

        map.put("info", table);

        map.put("rows", list);

        return AjaxResult.success(map);

        map.put("tables", tables);

        return success(map);

    }



    /**

     * 查询数据库列表

     */

    @PreAuthorize(hasPermi = "tool:gen:list")

    @RequiresPermissions("tool:gen:list")

    @GetMapping("/db/list")

    public TableDataInfo dataList(GenTable genTable)

    {

@@ -85,7 +88,7 @@
    /**

     * 查询数据表字段列表

     */

    @GetMapping(value = "/column/{talbleId}")

    @GetMapping(value = "/column/{tableId}")

    public TableDataInfo columnList(Long tableId)

    {

        TableDataInfo dataInfo = new TableDataInfo();

@@ -98,7 +101,7 @@
    /**

     * 导入表结构（保存）

     */

    @PreAuthorize(hasPermi = "tool:gen:list")

    @RequiresPermissions("tool:gen:import")

    @Log(title = "代码生成", businessType = BusinessType.IMPORT)

    @PostMapping("/importTable")

    public AjaxResult importTableSave(String tables)

@@ -107,49 +110,49 @@
        // 查询表信息

        List<GenTable> tableList = genTableService.selectDbTableListByNames(tableNames);

        genTableService.importGenTable(tableList);

        return AjaxResult.success();

        return success();

    }



    /**

     * 修改保存代码生成业务

     */

    @PreAuthorize(hasPermi = "tool:gen:edit")

    @RequiresPermissions("tool:gen:edit")

    @Log(title = "代码生成", businessType = BusinessType.UPDATE)

    @PutMapping

    public AjaxResult editSave(@Validated @RequestBody GenTable genTable)

    {

        genTableService.validateEdit(genTable);

        genTableService.updateGenTable(genTable);

        return AjaxResult.success();

        return success();

    }



    /**

     * 删除代码生成

     */

    @PreAuthorize(hasPermi = "tool:gen:remove")

    @RequiresPermissions("tool:gen:remove")

    @Log(title = "代码生成", businessType = BusinessType.DELETE)

    @DeleteMapping("/{tableIds}")

    public AjaxResult remove(@PathVariable Long[] tableIds)

    {

        genTableService.deleteGenTableByIds(tableIds);

        return AjaxResult.success();

        return success();

    }



    /**

     * 预览代码

     */

    @PreAuthorize(hasPermi = "tool:gen:preview")

    @RequiresPermissions("tool:gen:preview")

    @GetMapping("/preview/{tableId}")

    public AjaxResult preview(@PathVariable("tableId") Long tableId) throws IOException

    {

        Map<String, String> dataMap = genTableService.previewCode(tableId);

        return AjaxResult.success(dataMap);

        return success(dataMap);

    }



    /**

     * 生成代码（下载方式）

     */

    @PreAuthorize(hasPermi = "tool:gen:code")

    @RequiresPermissions("tool:gen:code")

    @Log(title = "代码生成", businessType = BusinessType.GENCODE)

    @GetMapping("/download/{tableName}")

    public void download(HttpServletResponse response, @PathVariable("tableName") String tableName) throws IOException

@@ -161,19 +164,35 @@
    /**

     * 生成代码（自定义路径）

     */

    @PreAuthorize(hasPermi = "tool:gen:code")

    @RequiresPermissions("tool:gen:code")

    @Log(title = "代码生成", businessType = BusinessType.GENCODE)

    @GetMapping("/genCode/{tableName}")

    public AjaxResult genCode(HttpServletResponse response, @PathVariable("tableName") String tableName)

    public AjaxResult genCode(@PathVariable("tableName") String tableName)

    {

        if (!GenConfig.isAllowOverwrite())

        {

            return AjaxResult.error("【系统预设】不允许生成文件覆盖到本地");

        }

        genTableService.generatorCode(tableName);

        return AjaxResult.success();

        return success();

    }



    /**

     * 同步数据库

     */

    @RequiresPermissions("tool:gen:edit")

    @Log(title = "代码生成", businessType = BusinessType.UPDATE)

    @GetMapping("/synchDb/{tableName}")

    public AjaxResult synchDb(@PathVariable("tableName") String tableName)

    {

        genTableService.synchDb(tableName);

        return success();

    }



    /**

     * 批量生成代码

     */

    @PreAuthorize(hasPermi = "tool:gen:code")

    @RequiresPermissions("tool:gen:code")

    @Log(title = "代码生成", businessType = BusinessType.GENCODE)

    @GetMapping("/batchGenCode")

    public void batchGenCode(HttpServletResponse response, String tables) throws IOException