import java.io.BufferedReader;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStreamReader;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * @Author: Huang JX
 * @Date: 2021/11/11
 * @Description: datax 批量執(zhí)行腳本，日志存放在當前目錄的 DataxSyncLogs 中；
 * 注意：需要輸入 datax 腳本的文件夾路徑作為 main 函數(shù)的參數(shù)，如 home/xxx/datax/job/
 * <p>
 * 使用說明：
 * 1、去掉文件的包名package；
 * 2、將此Java文件放到 /datax/bin/目錄下；
 * 3、javac DataxSyncUtil.java 
 * 4、java DataxSyncUtil [參數(shù)]datax腳本文件夾路徑
 */
public class DataxSyncUtil {
    public static void main(String[] args) throws Exception {
        FileOutputStream out = null;
        StringBuffer sb = new StringBuffer();
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmss");
        try {
            String jobPath = args[0];
            if (jobPath == null && "".equals(jobPath.trim())) {
                throw new Exception("請輸入datax腳本文件夾路徑");
            }
            String logDir = jobPath + "DataxSyncLogs";
            File logDirFile = new File(logDir);
            if (!logDirFile.exists()) {
                logDirFile.mkdir(); // 創(chuàng)建日志文件夾
            }
            String logFileName = logDir + "/datax_sync_" + sdf.format(new Date()) + ".log";
            File file = new File(logFileName);
            if (!file.exists())
                file.createNewFile();
            out = new FileOutputStream(file, true);
            sb.append("==========================================\n");
            sb.append("==========Datax Sync Job Start!==========\n");
            sb.append("==========================================\n");
            File f = new File(jobPath);
            if (!f.exists()) {
                sb.append(jobPath + " not exists\n");
                return;
            }
            System.out.print(sb.toString());
            out.write(sb.toString().getBytes("utf-8"));
            String cmdStr;
            File fa[] = f.listFiles();
            for (int i = 0; i < fa.length; i++) {
                sb = new StringBuffer();
                File fs = fa[i];
                if (!fs.isDirectory()) {
                    Process pr = null;
                    cmdStr = "python3 datax.py " + jobPath + fs.getName();
                    sb.append("start cmd: " + cmdStr + "\n");
                    System.out.print("start cmd: " + cmdStr + "\n");
                    pr = Runtime.getRuntime().exec(cmdStr);
                    BufferedReader in = new BufferedReader(new InputStreamReader(pr.getInputStream()));
                    String line;
                    while (true) {
                        if (!((line = in.readLine()) != null)) break;
                        sb.append(line + "\n");
                        System.out.print(line + "\n");
                    }
                    in.close();
                    pr.waitFor();
                    out.write(sb.toString().getBytes("utf-8"));
                }
            }
            sb.append("==========================================\n");
            sb.append("===========Datax Sync Job End!===========\n");
            sb.append("==========================================\n");
            System.out.print(sb.toString());
            out.write(sb.toString().getBytes("utf-8"));
        } finally {
            out.flush();
            out.close();
        }
    }
}