com.manydesigns.portofino.quartz.PortofinoJobFactory Maven / Gradle / Ivy

Go to download
/*
* Copyright (C) 2005-2020 ManyDesigns srl.  All rights reserved.
* http://www.manydesigns.com/
*
* Unless you have purchased a commercial license agreement from ManyDesigns srl,
* the following license terms apply:
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as published by
* the Free Software Foundation.
*
* There are special exceptions to the terms and conditions of the GPL
* as it is applied to this software. View the full text of the
* exception in file OPEN-SOURCE-LICENSE.txt in the directory of this
* software distribution.
*
* This program is distributed WITHOUT ANY WARRANTY; and without the
* implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, see http://www.gnu.org/licenses/gpl.txt
* or write to:
* Free Software Foundation, Inc.,
* 59 Temple Place - Suite 330,
* Boston, MA  02111-1307  USA
*
*/

package com.manydesigns.portofino.quartz;

import com.manydesigns.portofino.code.CodeBase;
import com.manydesigns.portofino.persistence.Persistence;
import org.apache.shiro.util.ThreadContext;
import org.apache.shiro.web.env.WebEnvironment;
import org.apache.shiro.web.util.WebUtils;
import org.quartz.*;
import org.quartz.simpl.SimpleJobFactory;
import org.quartz.spi.TriggerFiredBundle;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationContext;

import javax.servlet.ServletContext;

/**
 * @author Paolo Predonzani     - [email protected]
 * @author Angelo Lupo          - [email protected]
 * @author Giampiero Granatella - [email protected]
 * @author Alessio Stalla       - [email protected]
 */
public class PortofinoJobFactory extends SimpleJobFactory {
    public static final String copyright =
            "Copyright (C) 2005-2020 ManyDesigns srl";

    private static final Logger logger = LoggerFactory.getLogger(PortofinoJobFactory.class);
    protected final ServletContext servletContext;
    protected final ApplicationContext applicationContext;
    protected final CodeBase codeBase;

    public PortofinoJobFactory(ServletContext servletContext, ApplicationContext applicationContext) {
        this.servletContext = servletContext;
        this.applicationContext = applicationContext;
        this.codeBase = applicationContext.getBean(CodeBase.class);
    }

    @Override
    public Job newJob(TriggerFiredBundle bundle, Scheduler scheduler) throws SchedulerException {
        Job job = instantiateJob(bundle);
        applicationContext.getAutowireCapableBeanFactory().autowireBean(job);
        return jobExecutionContext -> {
            try {
                WebEnvironment env = WebUtils.getWebEnvironment(servletContext);
                if(env != null) {
                    ThreadContext.bind(env.getWebSecurityManager());
                }
                job.execute(jobExecutionContext);
            } finally {
                cleanup();
            }
        };
    }

    protected void cleanup() {
        try {
            //In a different class to make the database module optional at runtime
            SessionCleaner.closeSessions(applicationContext);
        } catch (NoClassDefFoundError e) {
            logger.debug("Database module not available, not closing sessions", e);
        }
        ThreadContext.unbindSecurityManager();
    }

    protected Job instantiateJob(TriggerFiredBundle bundle) throws SchedulerException {
        JobDetail jobDetail = bundle.getJobDetail();
        Class jobClass = jobDetail.getJobClass();
        //Attempt to reload
        try {
            Class newClass = codeBase.loadClass(jobClass.getName());
            if(Job.class.isAssignableFrom(newClass)) {
                jobClass = newClass;
            } else {
                logger.error(newClass + " does not extend " + Job.class + ", not reloading the job.");
            }
        } catch (Exception e) {
            logger.error("Failed to reload job class " + jobClass.getName(), e);
        }
        Job job;
        try {
            job = (Job) jobClass.getDeclaredConstructor().newInstance();
        } catch (Exception e) {
            throw new SchedulerException("Problem instantiating class '" + jobDetail.getJobClass().getName() + "'", e);
        }
        return job;
    }

}

class SessionCleaner {
    static void closeSessions(ApplicationContext applicationContext) {
        Persistence persistence = applicationContext.getBean(Persistence.class);
        persistence.closeSessions();
    }
}