文件工具类变更

chenhj

2026-04-24 2e71bdbcdf853bb35e68016b84f0254f7366bfeb

 src/main/java/com/ruoyi/project/tool/gen/service/GenTableServiceImpl.java

@@ -1,5 +1,30 @@
package com.ruoyi.project.tool.gen.service;



import com.alibaba.fastjson2.JSON;

import com.alibaba.fastjson2.JSONObject;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.constant.GenConstants;

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

import com.ruoyi.common.exception.ServiceException;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.project.tool.gen.domain.GenTable;

import com.ruoyi.project.tool.gen.domain.GenTableColumn;

import com.ruoyi.project.tool.gen.mapper.GenTableColumnMapper;

import com.ruoyi.project.tool.gen.mapper.GenTableMapper;

import com.ruoyi.project.tool.gen.util.GenUtils;

import com.ruoyi.project.tool.gen.util.VelocityInitializer;

import com.ruoyi.project.tool.gen.util.VelocityUtils;

import lombok.RequiredArgsConstructor;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.IOUtils;

import org.apache.velocity.Template;

import org.apache.velocity.VelocityContext;

import org.apache.velocity.app.Velocity;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.stereotype.Service;

import org.springframework.transaction.annotation.Transactional;



import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.IOException;

@@ -11,31 +36,6 @@
import java.util.stream.Collectors;

import java.util.zip.ZipEntry;

import java.util.zip.ZipOutputStream;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.IOUtils;

import org.apache.velocity.Template;

import org.apache.velocity.VelocityContext;

import org.apache.velocity.app.Velocity;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

import org.springframework.stereotype.Service;

import org.springframework.transaction.annotation.Transactional;

import com.alibaba.fastjson.JSON;

import com.alibaba.fastjson.JSONObject;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.constant.GenConstants;

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

import com.ruoyi.common.exception.ServiceException;

import com.ruoyi.common.utils.SecurityUtils;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.project.tool.gen.domain.GenTable;

import com.ruoyi.project.tool.gen.domain.GenTableColumn;

import com.ruoyi.project.tool.gen.mapper.GenTableColumnMapper;

import com.ruoyi.project.tool.gen.mapper.GenTableMapper;

import com.ruoyi.project.tool.gen.util.GenUtils;

import com.ruoyi.project.tool.gen.util.VelocityInitializer;

import com.ruoyi.project.tool.gen.util.VelocityUtils;



/**

 * 业务 服务层实现

@@ -43,15 +43,13 @@
 * @author ruoyi

 */

@Service

@RequiredArgsConstructor

public class GenTableServiceImpl implements IGenTableService

{

    private static final Logger log = LoggerFactory.getLogger(GenTableServiceImpl.class);



    @Autowired

    private GenTableMapper genTableMapper;



    @Autowired

    private GenTableColumnMapper genTableColumnMapper;

    private final GenTableMapper genTableMapper;

    private final GenTableColumnMapper genTableColumnMapper;



    /**

     * 查询业务信息

@@ -129,9 +127,9 @@
        int row = genTableMapper.updateGenTable(genTable);

        if (row > 0)

        {

            for (GenTableColumn cenTableColumn : genTable.getColumns())

            for (GenTableColumn genTableColumn : genTable.getColumns())

            {

                genTableColumnMapper.updateGenTableColumn(cenTableColumn);

                genTableColumnMapper.updateGenTableColumn(genTableColumn);

            }

        }

    }

@@ -151,15 +149,26 @@
    }



    /**

     * 创建表

     *

     * @param sql 创建表语句

     * @return 结果

     */

    @Override

    public boolean createTable(String sql)

    {

        return genTableMapper.createTable(sql) == 0;

    }



    /**

     * 导入表结构

     * 

     * @param tableList 导入表列表

     */

    @Override

    @Transactional

    public void importGenTable(List<GenTable> tableList)

    public void importGenTable(List<GenTable> tableList, String operName)

    {

        String operName = SecurityUtils.getUsername();

        try

        {

            for (GenTable table : tableList)

@@ -206,7 +215,7 @@
        VelocityContext context = VelocityUtils.prepareContext(table);



        // 获取模板列表

        List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory());

        List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory(), table.getTplWebType());

        for (String template : templates)

        {

            // 渲染模板

@@ -254,7 +263,7 @@
        VelocityContext context = VelocityUtils.prepareContext(table);



        // 获取模板列表

        List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory());

        List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory(), table.getTplWebType());

        for (String template : templates)

        {

            if (!StringUtils.containsAny(template, "sql.vm", "api.js.vm", "index.vue.vm", "index-tree.vue.vm"))

@@ -367,7 +376,7 @@
        VelocityContext context = VelocityUtils.prepareContext(table);



        // 获取模板列表

        List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory());

        List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory(), table.getTplWebType());

        for (String template : templates)

        {

            // 渲染模板

@@ -401,7 +410,7 @@
        if (GenConstants.TPL_TREE.equals(genTable.getTplCategory()))

        {

            String options = JSON.toJSONString(genTable.getParams());

            JSONObject paramsObj = JSONObject.parseObject(options);

            JSONObject paramsObj = JSON.parseObject(options);

            if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_CODE)))

            {

                throw new ServiceException("树编码字段不能为空");

@@ -485,13 +494,13 @@
     */

    public void setTableFromOptions(GenTable genTable)

    {

        JSONObject paramsObj = JSONObject.parseObject(genTable.getOptions());

        JSONObject paramsObj = JSON.parseObject(genTable.getOptions());

        if (StringUtils.isNotNull(paramsObj))

        {

            String treeCode = paramsObj.getString(GenConstants.TREE_CODE);

            String treeParentCode = paramsObj.getString(GenConstants.TREE_PARENT_CODE);

            String treeName = paramsObj.getString(GenConstants.TREE_NAME);

            String parentMenuId = paramsObj.getString(GenConstants.PARENT_MENU_ID);

            Long parentMenuId = paramsObj.getLongValue(GenConstants.PARENT_MENU_ID);

            String parentMenuName = paramsObj.getString(GenConstants.PARENT_MENU_NAME);



            genTable.setTreeCode(treeCode);