package com.dsideal.base.Tools;

import com.dsideal.base.DataEase.Model.DataEaseModel;
import com.dsideal.base.Tools.Util.SshMysqlConnectUtil;
import com.jcraft.jsch.JSchException;
import com.jfinal.plugin.activerecord.Record;

import java.io.IOException;
import java.util.List;

import com.jcraft.jsch.Session;

public class DataSetAddXzqh {

    public static DataEaseModel dm = new DataEaseModel();

    public static void main(String[] args) throws IOException, JSchException {
        Session session = SshMysqlConnectUtil.Init();
        //遍历所有以excel_开头的表，找出这些表中是不是存在行政区划,上级行政区划的列名
        for (int identity_id : new int[]{1, 2, 3}) {
            List<Record> list = dm.getDataSetByIdentityId(identity_id);
            //开始检查
            for (Record record : list) {
                String tableName = record.getStr("table_name");
                String dataSetName = record.getStr("dataset_name");

                //这张表中存在行政区划的列名，就跳过，不处理
                if (dm.hasNoColumnName(tableName, "行政区划")) {
                    //System.out.println((identity_id == 2 ? "【市州】" : "【县区】") + dataSetName + "表：" + tableName + "没有行政区划的列名，请手动添加");
                    dm.addColumn(tableName, "行政区划");
                }
                //这张表中存在上级行政区划的列名，就跳过，不处理
                if (dm.hasNoColumnName(tableName, "上级行政区划")) {
                    //System.out.println((identity_id == 2 ? "【市州】" : "【县区】") + dataSetName + "表：" + tableName + "没有行政区划的列名，请手动添加");
                    dm.addColumn(tableName, "上级行政区划");
                }
            }
        }
        System.out.println("恭喜，所有数据处理完毕。");
        session.disconnect();
    }
}