|
|
|
|
|
package com.dsideal.base.Tools;
|
|
|
|
|
|
|
|
|
|
|
|
import com.dsideal.base.DataEase.Model.DataEaseModel;
|
|
|
|
|
|
import com.dsideal.base.Tools.Util.SshConnectUtil;
|
|
|
|
|
|
import com.jcraft.jsch.JSchException;
|
|
|
|
|
|
import com.jcraft.jsch.Session;
|
|
|
|
|
|
|
|
|
|
|
|
import java.io.IOException;
|
|
|
|
|
|
|
|
|
|
|
|
public class ExportExcel {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public static void main(String[] args) throws IOException, JSchException {
|
|
|
|
|
|
|
|
|
|
|
|
//通过ssh链接到Mysql数据库
|
|
|
|
|
|
Session session = SshConnectUtil.Init();
|
|
|
|
|
|
|
|
|
|
|
|
DataEaseModel dm = new DataEaseModel();
|
|
|
|
|
|
|
|
|
|
|
|
//1、添加到数据集表中
|
|
|
|
|
|
dm.collectDataSet();
|
|
|
|
|
|
//2、加上主键
|
|
|
|
|
|
dm.addPrimaryKey();
|
|
|
|
|
|
//3、将所有非空列去掉不允许为空的限制
|
|
|
|
|
|
dm.updateNotNullColumns();
|
|
|
|
|
|
|
|
|
|
|
|
// //4、遍历所有数据集的表
|
|
|
|
|
|
// int total = 0;
|
|
|
|
|
|
// for (int i = 1; i <= 3; i++) {//1：省管理员，2：市州管理员，3：县区管理员
|
|
|
|
|
|
// switch (i) {
|
|
|
|
|
|
// case 1:
|
|
|
|
|
|
// System.out.println("正在处理省管理员需要看到的数据集~");
|
|
|
|
|
|
// break;
|
|
|
|
|
|
// case 2:
|
|
|
|
|
|
// System.out.println("正在处理市州管理员需要看到的数据集~");
|
|
|
|
|
|
// break;
|
|
|
|
|
|
// case 3:
|
|
|
|
|
|
// System.out.println("正在处理县区管理员需要看到的数据集~");
|
|
|
|
|
|
// break;
|
|
|
|
|
|
// }
|
|
|
|
|
|
//
|
|
|
|
|
|
// //Excel的保存目录
|
|
|
|
|
|
// String exportPath = "D:/360Downloads/Excel";
|
|
|
|
|
|
// File directory = new File(exportPath);
|
|
|
|
|
|
// // 检查目录是否存在
|
|
|
|
|
|
// if (!directory.exists()) {
|
|
|
|
|
|
// directory.mkdirs();
|
|
|
|
|
|
// }
|
|
|
|
|
|
//
|
|
|
|
|
|
// List<Record> list = dm.getDataSetByIdentityId(i);
|
|
|
|
|
|
// System.out.println("数据集个数=" + list.size());
|
|
|
|
|
|
// total += list.size();
|
|
|
|
|
|
// for (Record record : list) {
|
|
|
|
|
|
// String tableName = record.getStr("table_name");
|
|
|
|
|
|
// String dataSetName = record.getStr("dataset_name");
|
|
|
|
|
|
// System.out.println("正在处理数据集：" + dataSetName);
|
|
|
|
|
|
//
|
|
|
|
|
|
// //省的导出全部，市州管理员只导出市州数据，县区管理员只导出县区数据
|
|
|
|
|
|
// List<Record> tableData = new ArrayList<>();
|
|
|
|
|
|
// String excelFileName = "";
|
|
|
|
|
|
// String privinceName = "云南省";
|
|
|
|
|
|
// Record privinceRecord = dm.getAreaByName(privinceName);
|
|
|
|
|
|
// String privinceId = privinceRecord.getStr("id");
|
|
|
|
|
|
// if (i == 1) {
|
|
|
|
|
|
// //对此表中的数据进行直接导出EXCEL
|
|
|
|
|
|
// String sql = "select * from `" + tableName + "`";
|
|
|
|
|
|
// tableData = Db.use(DataEaseModel.DB_NAME).find(sql);
|
|
|
|
|
|
// excelFileName = "【" + privinceName + "】" + dataSetName + ".xlsx";
|
|
|
|
|
|
// ExcelCommonUtil.writeExcel(tableData, exportPath + "/" + excelFileName, true);
|
|
|
|
|
|
// } else if (i == 2) {
|
|
|
|
|
|
// //枚举所有的市州数据，然后进行导出EXCEL
|
|
|
|
|
|
// List<Record> cityList = dm.getAreaList(privinceId);
|
|
|
|
|
|
// String cityName = "";
|
|
|
|
|
|
// String cityId = "";
|
|
|
|
|
|
// for (Record rCity : cityList) {
|
|
|
|
|
|
// cityName = rCity.getStr("area_name");
|
|
|
|
|
|
// cityId = rCity.getStr("id");
|
|
|
|
|
|
// if (i == 3) {
|
|
|
|
|
|
//
|
|
|
|
|
|
// //枚举所有的县区数据，然后进行导出EXCEL
|
|
|
|
|
|
// ExcelCommonUtil.writeExcel(tableData, exportPath + "/" + excelFileName, true);
|
|
|
|
|
|
// }
|
|
|
|
|
|
// }
|
|
|
|
|
|
// String sql = "select * from `" + tableName + "` where `行政区划`=?";
|
|
|
|
|
|
// System.out.println(tableName);
|
|
|
|
|
|
// tableData = Db.use(DataEaseModel.DB_NAME).find(sql, cityName);
|
|
|
|
|
|
// excelFileName = "【" + cityName + "】" + dataSetName + ".xlsx";
|
|
|
|
|
|
// ExcelCommonUtil.writeExcel(tableData, exportPath + "/" + excelFileName, true);
|
|
|
|
|
|
// }
|
|
|
|
|
|
// }
|
|
|
|
|
|
// }
|
|
|
|
|
|
// System.out.println("恭喜，所有数据处理完毕，共生成数据集数量=" + total + "个。");
|
|
|
|
|
|
|
|
|
|
|
|
session.disconnect();
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
