QingLong/src/main/java/Tools/DataEase/ExcelExport.java

package Tools.DataEase;

import Tools.DataEase.Util.DataEaseUtil;
import cn.hutool.core.io.FileUtil;
import com.dsideal.QingLong.Util.CommonUtil;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.Db;
import com.jfinal.plugin.activerecord.Record;

import java.io.File;
import java.io.IOException;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;

public class ExcelExport {

    public static void main(String[] args) throws SQLException, ClassNotFoundException, IOException {
        DataEaseUtil.Init();

        //JDBC读取sourceSql,可以按列序号获取源数据
        Class.forName(PropKit.get("driverClassName"));
        Connection conn = DriverManager.getConnection(PropKit.get("read_jdbcUrl"), PropKit.get("read_user"), PropKit.get("read_password"));
        Statement statement = conn.createStatement();

        //文本文件
        if (!FileUtil.exist(DataEaseUtil.DataEaseDirectory)) {
            System.out.println("没有找到要导入的目录！");
            System.exit(0);
        }
        File[] files = FileUtil.ls(DataEaseUtil.DataEaseDirectory);
        List<String> sqlList = new ArrayList<>();
        // 遍历文件数组
        for (File file : files) {
            String name = file.getName().substring(file.getName().lastIndexOf(".") + 1);
            if (!name.toLowerCase().equals("txt")) continue;
            // 获取文件名
            String fileName = file.getAbsolutePath();
            // 读取文件内容
            String content = FileUtil.readUtf8String(fileName);
            // 获取表名
            String tableName = content.split("\r\n")[2];
            CommonUtil.log("正在处理：" + file.getName());
            // 使用FileUtil按行读取文件
            List<String> lines = FileUtil.readLines(new File(fileName), "UTF-8");
            // 遍历每一行内容
            boolean flag = false;
            StringBuilder sb = new StringBuilder();
            for (String line : lines) {
                // 处理每一行的内容
                if (line.startsWith("*/")) {
                    flag = true;
                    continue;
                }
                if (line.startsWith("--")) continue;
                if (flag) sb.append(line + " ");
            }
            String sourceSql = sb.toString().trim();
            if (!sourceSql.endsWith(";")) {
                CommonUtil.log("文件" + file.getName() + "没有以;号结尾，请修改后再次运行本程序！");
                System.exit(0);
            }

            String[] array = sourceSql.split(";");
            int lastSelectLine = -1;
            for (int i = array.length - 1; i >= 0; i--) {
                if (array[i].toLowerCase().trim().startsWith("select")) {
                    lastSelectLine = i;
                    break;
                }
            }
            if (lastSelectLine == -1) {
                CommonUtil.log("没有找到select查询语句，程序无法继续运行！");
                System.exit(0);
            }
            sourceSql = array[lastSelectLine];
            for (int i = 0; i < array.length; i++) {
                if (i == lastSelectLine) continue;
                String execSql = array[i];
                CommonUtil.log("正在执行更新类Sql:" + execSql);
                try {
                    PreparedStatement p = conn.prepareStatement(execSql);
                    p.executeUpdate();
                } catch (Exception err) {
                    CommonUtil.log("执行语句失败,失败细节：" + err);
                }
            }
            //有哪些列
            List<String> cList = DataEaseUtil.getColumns(tableName);
            ResultSet rs = statement.executeQuery(sourceSql);
            ResultSetMetaData rsmd = rs.getMetaData();
            //ResultSet的总列数
            int columnCount = rsmd.getColumnCount();
            //获取表的主键ID
            String sql = "select dataease_uuid from " + tableName;
            List<Record> fList = Db.find(sql);

            int idx = 0;
            while (rs.next()) {
                for (int i = 1; i <= columnCount; i++) {
                    String colName = cList.get(i).toLowerCase();
                    sql = "update " + tableName + " set " + colName + "='"
                            + rs.getString(i) + "' where dataease_uuid='"
                            + fList.get(idx).getStr("dataease_uuid") + "';";
                    sqlList.add(sql);
                }
                idx++;
            }
        }
        //关闭数据库
        statement.close();
        conn.close();
        Db.batch(sqlList, 500);
        CommonUtil.log("恭喜，数据导出并导入成功！");
    }
}