package com.dy.pipIrrModel.timingTask;

import com.dy.common.multiDataSource.DataSourceContext;
import com.dy.common.schedulerTask.TaskJob;
import com.dy.common.springUtil.SpringContextUtil;
import com.dy.pipIrrGlobal.util.Org;
import com.dy.pipIrrModel.modelCalculate.ModelCalculator;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.quartz.JobDataMap;
import org.quartz.JobExecutionContext;
import org.quartz.JobExecutionException;

import java.util.List;

/**
 * @Author: liurunyu
 * @Date: 2025/8/6 9:32
 * @Description
 */
public class CalculateJob extends TaskJob {

    private static Logger log = LogManager.getLogger(CalculateJob.class.getName()) ;

    private List<Org.OrgVo> orgList ;

    @Override
    public void execute(JobExecutionContext ctx) throws JobExecutionException {
        JobDataMap jobDataMap = ctx.getJobDetail().getJobDataMap() ;
        if(jobDataMap != null){
            orgList = (List<Org.OrgVo>)jobDataMap.get(ModelListener.orgListKey) ;
        }
        if(orgList == null){
            return ;
        }

        for(Org.OrgVo vo : orgList){
            DataSourceContext.set(vo.tag);//设置数据源
            ModelCalculator bean = (ModelCalculator)SpringContextUtil.getBean(ModelCalculator.selfBeanName);
            if(bean != null){
                bean.execute();
            }
        }
    }
}