package com.sitech.jmx.mbean;

import com.sitech.base.AgentProperties;
import com.sitech.database.dao.TbAtoDynamicThresholdDao;
import com.sitech.database.dao.TbCfgEventDao;
import com.sitech.database.dao.TbCfgOidMibDao;
import com.sitech.database.domain.TbCfgEvent;
import com.sitech.database.domain.TbCfgOidMib;
import com.sitech.ismp.check.downtime.DowntimeDao;
import com.sitech.ismp.coll.cron.CronConstants;
import com.sitech.ismp.messageObject.*;
import com.sitech.jmx.manage.CollScheduleManager;
import com.sitech.jmx.manage.MBeanManager;
import com.sitech.schedule.ScheduleCache;
import com.sitech.schedule.Scheduler;
import com.sitech.snmptrap.TbTopoPhysicalDeviceDao;
import com.sitech.util.Formater;
import com.sitech.util.mq.MQConstants;
import com.sitech.util.mq.TunnelFactory;
import org.apache.log4j.Logger;

import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 采集/告警配置同步
 * 
 * @author linxc
 * @version
 * @since Ver 7.0.0
 * @Date 2012 Jun 5, 2012 2:04:01 PM
 */
public class ConfigSync implements ConfigSyncMBean {
	private Logger logger = Logger.getLogger("LOGER");
	private Logger error = Logger.getLogger("ERROR");

	public static final int TYPE_SYNC_ALL = 0;
	public static final int TYPE_CREATE = 1;
	public static final int TYPE_UPDATE = 2;
	public static final int TYPE_DELETE = 3;

	public void syncCollCfg(AgentSyncObject obj) throws Exception {
		int type = obj.getType();
        String endTime = Formater.datetimeToString(new Date(),"yyyy-MM-dd HH:mm:ss");
        String logInfo = "";
        Map<String,Object> schedDatas = null;
        String oprType = obj.getOperateType();
        String nextFireTime = "";
        String scheStatus = "0";
        Map<String,Object> extInfo = new HashMap<String, Object>();
        try{
            switch (type) {
                case TYPE_CREATE:
                    schedDatas = addCollSchedule(obj);
                    nextFireTime = (String) schedDatas.get("NEXT_FIRE_TIME");
                    scheStatus = "1";
                    logInfo = "agent添加调度任务成功";
                    break;
                case TYPE_UPDATE:
                    schedDatas = updateCollSchedule(obj);
                    nextFireTime = (String) schedDatas.get("NEXT_FIRE_TIME");
                    scheStatus = "1";
                    logInfo = "agent修改调度任务成功";
                    break;
                case TYPE_DELETE:
                    String scheduleOperate = obj.getParams().get("SCHEDULE_OPERATE");
                    extInfo.put("SCHEDULE_OPERATE",scheduleOperate);
                    removeCollSchedule(obj);
//                    removeMemDbCfgData(obj);
					removeDowntimeMBean(obj);
                    // 删除内存数据库中相关的配置数据
                    scheStatus = "0";
                    if("1".equals(scheduleOperate)){
                        logInfo = "agent删除调度任务成功";
                    } else if("2".equals(scheduleOperate)){
                        logInfo = "agent停止调度任务成功";
                    }
                    break;
            }
        }catch (Exception e){
            switch (type){
                case TYPE_CREATE:
                    if(e.getMessage().startsWith("ScheduleExist")){
                        logInfo = "调度任务[ScheduleID="+obj.getScheduleId()+"]已经存在！";
                        scheStatus = "1";
                    }else{
                        logInfo = "agent增加调度任务失败:"+e.getMessage();
                        scheStatus = "0";
                    }
                    break;
                case TYPE_UPDATE:
                    logInfo = "agent修改调度任务失败:"+e.getMessage();
                    scheStatus = "1";
                    break;
                case TYPE_DELETE:
                    String scheduleOperate = obj.getParams().get("SCHEDULE_OPERATE");
                    if("1".equals(scheduleOperate)){
                        logInfo = "agent删除调度任务失败";
                    } else if("2".equals(scheduleOperate)){
                        logInfo = "agent停止调度任务失败";
                    }
                    logInfo = logInfo+":"+e.getMessage();
                    scheStatus = "1";
                    break;
            }
            logger.error("Exception while syncCollCfg",e);
        }
        if(null!=oprType && !"".equals(oprType.trim()) && "2".equals(oprType.trim())){
            logger.info("CronSchedule Next_fire_time is : "+nextFireTime);
            extInfo.put("SCHEDULE_STATUS",scheStatus);
            sendScheduleLog(obj, logInfo, CronConstants.SEQ_1, endTime, nextFireTime, extInfo);
        }

	}

	public void syncEventCfg(AlarmEventConfigSyncObject obj) throws Exception {
		int type = obj.getSyncType();
		List<AlarmEventConfig> configList = obj.getConfigList();
		
		logger.info("Begin refresh table[TB_CFG_EVENT]...");
		
		if (configList == null || configList.size() == 0) {
			logger.info("End refresh table[TB_CFG_EVENT], update [0] config");
			return;
		}
		
		switch (type) {
		case TYPE_SYNC_ALL:
			refreshAllEventCfg(configList);
			break;
		case TYPE_CREATE:
			addEventCfg(configList);
			break;
		case TYPE_UPDATE:
			updateEventCfg(configList);
			break;
		case TYPE_DELETE:
			deleteEventCfg(configList);
			break;
		}
	}

	private void deleteEventCfg(List<AlarmEventConfig> configList) {
		String[] guid = new String[configList.size()];
		for (int i = 0; i < configList.size(); i++) {
			guid[i] = configList.get(i).getGuid();
		}
		
		TbCfgEventDao dao = new TbCfgEventDao();
		dao.deleteTbCfgEvent(guid);
		
		logger.info("End refresh table[TB_CFG_EVENT], delete ["+configList.size()+"] config");		
	}

	/**
	 * 删除添加主机的MBean。
	 * @param obj
     */
	private void removeMemDbCfgData(AgentSyncObject obj){
		if(obj.getClassName().equals("com.sitech.snmptrap.TbTopoPhysicalDevice")){
			TbTopoPhysicalDeviceDao physicalDeviceDao = new TbTopoPhysicalDeviceDao();
			physicalDeviceDao.deletePhysicalDeviceByIp(obj.getParams().get("DEVICE_IP"));
		}
	}

	/**
	 * 删除需要监控宕机的主机。
	 * @param obj
     */
	private void removeDowntimeMBean(AgentSyncObject obj){
		if(obj.getClassName().equals("com.sitech.ismp.check.downtime.Downtime")){
			DowntimeDao dao = new DowntimeDao();
			dao.deleteDowntimeHostByIp(obj.getParams().get("DEVICE_IP"));
		}
	}


	/**
	 * 添加监控宕机的MBean。
	 * 监控主机连接状态，有异常的时候，产生告警。
	 * frank 2017-04-11
	 */
	/*private void regeditDowntimeMBean(AgentSyncObject obj,String scheduleId){
		boolean isExist=CollScheduleManager.getInstance().scheduleExist(scheduleId);
		if(!isExist){
			DowntimeServer server=new DowntimeServer();
			if(obj.getClassName().equals("com.sitech.snmptrap.TbTopoPhysicalDevice")){
				AgentSyncObject downtimeAgentSyncObject= server.getAgentSyncObject(obj,scheduleId);
				log.info("get DowntimeMBean,className ====== "+downtimeAgentSyncObject.getClassName());
				try {
					addCollSchedule(downtimeAgentSyncObject);
					log.info("******************************************************************");
					log.info("*************  注册DowntimeMBean完成，开始监听宕机情况  **************");
					log.info("******************************************************************");
				} catch (Exception e) {
					log.error("regedit DowntimeMBean has wrong",e);
					e.printStackTrace();
				}
			}
		}
	}*/

	private void updateEventCfg(List<AlarmEventConfig> configList) {
		String[] guid = new String[configList.size()];
		for (int i = 0; i < configList.size(); i++) {
			guid[i] = configList.get(i).getGuid();
		}
		
		TbCfgEventDao dao = new TbCfgEventDao();
		dao.deleteTbCfgEvent(guid);
		
		for (AlarmEventConfig config : configList) {
			TbCfgEvent tbCfgEvent = new TbCfgEvent(config);				
			dao.insertTbCfgEvent(tbCfgEvent);
		}
		
		logger.info("End refresh table[TB_CFG_EVENT], update ["+configList.size()+"] config");
		
	}

	private void addEventCfg(List<AlarmEventConfig> configList) {
		TbCfgEventDao dao = new TbCfgEventDao();
		for (AlarmEventConfig config : configList) {
			TbCfgEvent tbCfgEvent = new TbCfgEvent(config);
			dao.insertTbCfgEvent(tbCfgEvent);
		}		
		logger.info("End refresh table[TB_CFG_EVENT], add ["+configList.size()+"] config");		
	}

	private void refreshAllEventCfg(List<AlarmEventConfig> configList) {
		TbCfgEventDao dao = new TbCfgEventDao();
		int result = dao.deleteTbCfgEventAll();
		if (result != -1) {
			for (AlarmEventConfig config : configList) {
				TbCfgEvent tbCfgEvent = new TbCfgEvent(config);
				dao.insertTbCfgEvent(tbCfgEvent);
			}
		}
		
		logger.info("End refresh table[TB_CFG_EVENT], update ["+configList.size()+"] config");
		
	}

	public void syncMibCfg(MibConfigSyncObject obj) throws Exception {
		int type = obj.getSyncType();
		List<MibConfig> configList = obj.getConfigList();
		
		logger.info("Begin refresh table[TB_CFG_OID_MIB]...");
		
		if (configList == null || configList.size() == 0) {
			logger.info("End refresh table[TB_CFG_OID_MIB], update [0] config");
			return;
		}
		
		switch (type) {
		case TYPE_SYNC_ALL:
			refreshAllMibCfg(configList);
			break;
		case TYPE_CREATE:
			addMibCfg(configList);
			break;
		case TYPE_UPDATE:
			updateMibCfg(configList);
			break;
		case TYPE_DELETE:
			deleteMibCfg(configList);
			break;
		}

	}

	private void deleteMibCfg(List<MibConfig> configList) {
		String[] guid = new String[configList.size()];
		for (int i = 0; i < configList.size(); i++) {
			guid[i] = configList.get(i).getGuid();
		}
		
		TbCfgOidMibDao dao = new TbCfgOidMibDao();
		dao.deleteTbCfgOidMib(guid);
		
		logger.info("End refresh table[TB_CFG_OID_MIB], delete ["+configList.size()+"] config");		
	}

	private void updateMibCfg(List<MibConfig> configList) {
		String[] guid = new String[configList.size()];
		for (int i = 0; i < configList.size(); i++) {
			guid[i] = configList.get(i).getGuid();
		}
		
		TbCfgOidMibDao dao = new TbCfgOidMibDao();
		dao.deleteTbCfgOidMib(guid);
		
		for (MibConfig config : configList) {
			TbCfgOidMib TbCfgOidMib = new TbCfgOidMib(config);				
			dao.insertTbCfgOidMib(TbCfgOidMib);
		}

		logger.info("End refresh table[TB_CFG_OID_MIB], update ["+configList.size()+"] config");
		
	}

	private void addMibCfg(List<MibConfig> configList) {
		TbCfgOidMibDao dao = new TbCfgOidMibDao();
		for (MibConfig config : configList) {
			TbCfgOidMib TbCfgOidMib = new TbCfgOidMib(config);
			dao.insertTbCfgOidMib(TbCfgOidMib);
		}		
		logger.info("End refresh table[TB_CFG_OID_MIB], add ["+configList.size()+"] config");		
	}

	private void refreshAllMibCfg(List<MibConfig> configList) {
		TbCfgOidMibDao dao = new TbCfgOidMibDao();
		int result = dao.deleteTbCfgOidMibAll();
		if (result != -1) {
			for (MibConfig config : configList) {
				TbCfgOidMib TbCfgOidMib = new TbCfgOidMib(config);
				dao.insertTbCfgOidMib(TbCfgOidMib);
			}
		}
		
		logger.info("End refresh table[TB_CFG_OID_MIB], update ["+configList.size()+"] config");
		
	}
	/**
	 * 添加采集调度任务
	 */
	public Map<String,Object> addCollSchedule(AgentSyncObject obj) throws Exception {
        Map<String,Object> schedDatas = null;
		try {

			Scheduler scheduler = new Scheduler(obj);
			logger.info("Add new scheduler: " + scheduler.toString());
			
			MBeanManager.getInstance().registCollMBean(scheduler);
            schedDatas = CollScheduleManager.getInstance().addCollSchedule(scheduler, obj);

			ScheduleCache.addScheduler(scheduler);
			ScheduleCache.display();
		} catch (Exception e) {
			error.error("Exception while addCollSchedule", e);
			throw e;
		}
        return schedDatas;
	}
	
	/**
	 * 删除采集调度任务
	 */
	public void removeCollSchedule(AgentSyncObject obj) throws Exception {
		try {
			Scheduler scheduler = new Scheduler(obj);
			logger.info("Delete scheduler: " + scheduler.toString());

			CollScheduleManager.getInstance().deleteCollSchedule(scheduler.getScheduleId());
			
			ScheduleCache.removeScheduler(scheduler);
			ScheduleCache.display();
		} catch (Exception e) {
			error.error("Exception while removeCollSchedule", e);
			throw e;
		}
	}
	
	/**
	 * 更新采集调度任务
	 */
	public Map<String,Object> updateCollSchedule(AgentSyncObject obj) throws Exception {
        Map<String,Object> scheDatas = null;
		try {
			Scheduler scheduler = new Scheduler(obj);
			logger.info("Update scheduler: " + scheduler.toString());

			MBeanManager.getInstance().registCollMBean(scheduler);
            scheDatas = CollScheduleManager.getInstance().updateCollSchedule(scheduler, obj);

			ScheduleCache.removeScheduler(scheduler);
			ScheduleCache.addScheduler(scheduler);
			ScheduleCache.display();
		} catch (Exception e) {
			error.error("Exception while updateCollSchedule", e);
			throw e;
		}
        return scheDatas;
	}


    private void sendScheduleLog(AgentSyncObject agentSyncObj, String logContent, String seq, String endTime, String nextFireTime, Map<String,Object> extInfo){
        ScheduleLog log = new ScheduleLog();
        log.setLogType(CronConstants.SCHEDULE_LOG_MANUAL);
        extInfo.put("SCHEDULE_OPERATE",(String)agentSyncObj.getParams().get("SCHEDULE_OPERATE"));
        String beginTime = Formater.datetimeToString(new Date(), "yyyy-MM-dd HH:mm");
        log.setSchId(agentSyncObj.getScheduleId());
        log.setRequestId(agentSyncObj.getRequestId());
        if(null!=endTime && !"".equals(endTime)){
            log.setEndTime(endTime);
        }
        log.setAgentId(AgentProperties.AGENT_ID);
        log.setSeq(seq);
        log.setOperateType(CronConstants.OPERATE_TYPE_SYNC_CRONTASK);
        log.setType("std_out");
        log.setSyncType("" + agentSyncObj.getType());
        log.setLogInfo(logContent);
        log.setBeginTime(beginTime);
        log.setNextFireTime(nextFireTime);
        log.setTrigerType(agentSyncObj.getTriggerType());
        log.setExtInfo(extInfo);
        TunnelFactory.getTunnel(MQConstants.Q_ROPORT_FROM_AGENT).writeData(log);
    }

    public void syncDynamicCfg(AlarmDynamicConfigSyncObject obj) throws Exception {
        int type = obj.getSyncType();
        List<TbAtoDynamicThreshold> configList = obj.getConfigList();

        logger.info("Begin refresh table[TB_ATO_DYNAMIC_THRESHOLD]...");

        if (configList == null || configList.size() == 0) {
            logger.info("End refresh table[TB_ATO_DYNAMIC_THRESHOLD], update [0] config");
            return;
        }

        switch (type) {
            case TYPE_SYNC_ALL:
                refreshAllDynamicCfg(configList);
                break;
            case TYPE_CREATE:
                addDynamicCfg(configList);
                break;
            case TYPE_UPDATE:
                updateDynamicCfg(configList);
                break;
            case TYPE_DELETE:
                deleteDynamicCfg(configList);
                break;
        }
    }

    /**
     * 刷新所有动态阀值配置
     * @param configList
     */
    private void refreshAllDynamicCfg(List<TbAtoDynamicThreshold> configList) {
        TbAtoDynamicThresholdDao dao = new TbAtoDynamicThresholdDao();
        int result = dao.deleteTbAtoDynamicThresholdAll();
        if (result != -1) {
            for (TbAtoDynamicThreshold config : configList) {
                //TbCfgEvent tbCfgEvent = new TbCfgEvent(config);
                dao.insertTbAtoDynamicThreshold(config);
            }
        }
        logger.info("End refresh table[TB_ATO_DYNAMIC_THRESHOLD], update ["+configList.size()+"] config");
    }

    /**
     * 新增动态阀值配置
     * @param configList
     */
    private void addDynamicCfg(List<TbAtoDynamicThreshold> configList) {
        TbAtoDynamicThresholdDao dao = new TbAtoDynamicThresholdDao();
        for (TbAtoDynamicThreshold config : configList) {
            //TbCfgEvent tbCfgEvent = new TbCfgEvent(config);
            dao.insertTbAtoDynamicThreshold(config);
        }
        logger.info("End refresh table[TB_ATO_DYNAMIC_THRESHOLD], add ["+configList.size()+"] config");
    }

    /**
     * 更新动态阀值配置
     * @param configList
     */
    private void updateDynamicCfg(List<TbAtoDynamicThreshold> configList) {
        String[] id = new String[configList.size()];
        for (int i = 0; i < configList.size(); i++) {
            id[i] = configList.get(i).getID();
        }
        TbAtoDynamicThresholdDao dao = new TbAtoDynamicThresholdDao();
        dao.deleteTbAtoDynamicThreshold(id);
        for (TbAtoDynamicThreshold config : configList) {
            dao.insertTbAtoDynamicThreshold(config);
        }
        logger.info("End refresh table[TB_ATO_DYNAMIC_THRESHOLD], update ["+configList.size()+"] config");
    }

    /**
     * 删除动态阀值配置
     * @param configList
     */
    private void deleteDynamicCfg(List<TbAtoDynamicThreshold> configList) {
        String[] guid = new String[configList.size()];
        for (int i = 0; i < configList.size(); i++) {
            guid[i] = configList.get(i).getGUID();
        }
        TbAtoDynamicThresholdDao dao = new TbAtoDynamicThresholdDao();
        dao.deleteTbAtoDynamicThreshold(guid);
        logger.info("End refresh table[TB_ATO_DYNAMIC_THRESHOLD], delete ["+configList.size()+"] config");
    }

}