自动导入配置

RuoYi

2026-04-10 31d588a7c9d1b4749740307df2ec0bc0d930ada7

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

@@ -4,7 +4,7 @@
import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.servlet.http.HttpServletResponse;

import jakarta.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.validation.annotation.Validated;

@@ -15,6 +15,7 @@
import org.springframework.web.bind.annotation.PutMapping;

import org.springframework.web.bind.annotation.RequestBody;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RequestParam;

import org.springframework.web.bind.annotation.RestController;

import com.ruoyi.common.core.text.Convert;

import com.ruoyi.common.core.web.controller.BaseController;

@@ -22,7 +23,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 +48,7 @@
    /**

     * 查询代码生成列表

     */

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

    @RequiresPermissions("tool:gen:list")

    @GetMapping("/list")

    public TableDataInfo genList(GenTable genTable)

    {

@@ -56,26 +58,26 @@
    }



    /**

     * 修改代码生成业务

     * 获取代码生成信息

     */

    @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);

        GenTable table = genTableService.selectGenTableById(tableId);

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

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

        List<GenTableColumn> list = genTableColumnService.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 success(map);

    }



    /**

     * 查询数据库列表

     */

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

    @RequiresPermissions("tool:gen:list")

    @GetMapping("/db/list")

    public TableDataInfo dataList(GenTable genTable)

    {

@@ -87,7 +89,7 @@
    /**

     * 查询数据表字段列表

     */

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

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

    public TableDataInfo columnList(Long tableId)

    {

        TableDataInfo dataInfo = new TableDataInfo();

@@ -100,58 +102,58 @@
    /**

     * 导入表结构（保存）

     */

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

    @RequiresPermissions("tool:gen:import")

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

    @PostMapping("/importTable")

    public AjaxResult importTableSave(String tables)

    public AjaxResult importTableSave(@RequestParam("tables") String tables, @RequestParam("tplWebType") String tplWebType)

    {

        String[] tableNames = Convert.toStrArray(tables);

        // 查询表信息

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

        genTableService.importGenTable(tableList);

        return AjaxResult.success();

        genTableService.importGenTable(tableList, tplWebType);

        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

@@ -163,31 +165,35 @@
    /**

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

     */

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

    @RequiresPermissions("tool:gen:code")

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

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

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

    {

        if (!GenConfig.isAllowOverwrite())

        {

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

        }

        genTableService.generatorCode(tableName);

        return AjaxResult.success();

        return success();

    }



    /**

     * 同步数据库

     */

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

    @RequiresPermissions("tool:gen:edit")

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

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

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

    {

        genTableService.synchDb(tableName);

        return AjaxResult.success();

        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