/**
 * Copyright 2018-2118 the original author or authors.
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.vci.dbsync;

import org.quartz.Job;
import org.quartz.JobDataMap;
import org.quartz.JobExecutionContext;
import org.quartz.JobExecutionException;

import com.vci.corba.common.VCIError;
import com.vci.dbsync.entity.DBInfo;
import com.vci.dbsync.entity.JobInfo;
import com.vci.dbsync.entity.JobItem;
import com.vci.dbsync.entity.VMInfo;
import com.vci.dbsync.filesync.FileSync;
import com.vci.dbsync.log.SyncLog;
import com.vci.dbsync.DBSyncFactory;
import com.vci.dbsync.DBSync;

//import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Date;

/**
 * @author liuyazhuang
 * @date 2018/9/11 10:30
 * @description 同步数据库任务的具体实现
 * @version 1.0.0
 */
public class JobTask implements Job {
    //private Logger logger = Logger.getLogger(JobTask.class);

    /**
     * 执行同步数据库任务
     *
     */
    @Override
    public void execute(JobExecutionContext context) throws JobExecutionException {
        JobDataMap data = context.getJobDetail().getJobDataMap();
        JobInfo jobInfo = (JobInfo) data.get("jobInfo");

        SyncLog.logger.info("开始任务【" + jobInfo.getName() + "】 调度: " + new Date());

    	Connection inConn = null;
        Connection outConn = null;
        DBInfo srcDb = (DBInfo) data.get("srcDb");
        DBInfo destDb = (DBInfo) data.get("destDb");
        
        VMInfo srcVM = (VMInfo) data.get("srcVM");
        VMInfo destVM = (VMInfo) data.get("destVM");
        
        String logTitle = (String) data.get("logTitle");
        
        try {
            inConn = createConnection(srcDb);
            outConn = createConnection(destDb);
            if (inConn == null) {
            	SyncLog.logger.info("请检查源数据连接!");
                return;
            } else if (outConn == null) {
            	SyncLog.logger.info("请检查目标数据连接!");
                return;
            }
            
            DBSync dbHelper = DBSyncFactory.create(destDb.getDbtype());
            
            if (dbHelper == null) {
            	SyncLog.logger.info("不支持的数据库类型!");
            	return;
            }
            
            
            JobItem[] items = jobInfo.Items();
            
            //ArrayList<String> lstSql = new ArrayList<String>();
			for (JobItem item : items) {
				SyncLog.logger.info("开始同步： " + item.getName());
				
	            long start = new Date().getTime();
	            
	            FileSync fileSync = null;
	            if (item.getName().equalsIgnoreCase("PLATFORMBTM_FILEOBJECT")){
	            	fileSync = new FileSync(srcVM, destVM);
		            if (item.getDirection()) {
		            	fileSync.Init(true);
		            } else {
		            	fileSync.Init(false);
		            }
	            }

	            long count = 0;
	            if (item.getDirection())
	            {
	            	count = dbHelper.assembleSQL(item, inConn, outConn, fileSync);
	            }
	            else
	            {
	            	count = dbHelper.assembleSQL(item, outConn, inConn, fileSync);
	            }
	            
				SyncLog.logger.info("同步【" + item.getName() + "】查询到 " + count + " 条记录，同步耗时： " + (new Date().getTime() - start) + "ms");
				
	            if (fileSync != null){
					SyncLog.logger.info("同步【" + item.getName() + "】文件数： " + fileSync.getCount());
	            	fileSync.close();
	            }

//				try {
//					FileWriter file = new FileWriter("e:\\test.sql");
//
//					String sNewLine = System.getProperty("line.separator");
//					for (int i = 0; i < sqls.length; i++) {
//						file.write(sqls[i]); // 向文件中写入数据
//						file.write(sNewLine); // \r\n表示换行
//					}
//					// file.write(sqls.toString());
//					file.close();
//				} catch (IOException e) {
//					e.printStackTrace();
//				}

//				SyncLog.logger.info("组装SQL耗时：: " + (new Date().getTime() - start) + "ms");
//				if (lstSql.size() > 0) {
//					SyncLog.logger.debug(lstSql);
//					long eStart = new Date().getTime();
//					dbHelper.executeSQL(lstSql, outConn);
//					SyncLog.logger.info("执行SQL语句耗时： " + (new Date().getTime() - eStart) + "ms");
//				}
			}
        } catch (SQLException e) {
        	try {
				outConn.rollback();
			} catch (SQLException e1) {
				// TODO Auto-generated catch block
				e1.printStackTrace();
			}
        	e.printStackTrace();
            SyncLog.logger.error(logTitle + e.getMessage());
            SyncLog.logger.error(logTitle + " SQL执行出错，请检查是否存在语法错误");
        } catch (VCIError ex){
        	ex.printStackTrace();
        } catch (Exception ex){
        	ex.printStackTrace();
        }
        finally {
            SyncLog.logger.info("关闭源数据库连接");
            destoryConnection(inConn);
            SyncLog.logger.info("关闭目标数据库连接");
            destoryConnection(outConn);
        }
    }

    /**
     * 创建数据库连接
     * @param db
     * @return
     */
    private Connection createConnection(DBInfo db) {
        try {
            Class.forName(db.getDriver());
            Connection conn = DriverManager.getConnection(db.getUrl(), db.getUsername(), db.getPassword());
            conn.setAutoCommit(false);
            return conn;
        } catch (Exception e) {
            SyncLog.logger.error(e.getMessage());
        }
        return null;
    }

    /**
     * 关闭并销毁数据库连接
     * @param conn
     */
    private void destoryConnection(Connection conn) {
        try {
            if (conn != null) {
                conn.close();
                conn = null;
                //SyncLog.logger.info("数据库连接关闭");
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}