io.dropwizard.jobs.JobsBundle Maven / Gradle / Ivy

Go to download
package io.dropwizard.jobs;

import com.codahale.metrics.SharedMetricRegistries;

import org.quartz.Scheduler;

import io.dropwizard.ConfiguredBundle;
import io.dropwizard.setup.Bootstrap;
import io.dropwizard.setup.Environment;

public class JobsBundle implements ConfiguredBundle {

    private final Job[] jobs;
    protected JobManager jobManager;

    public JobsBundle(Job... jobs) {
        this.jobs = jobs;
    }

    @Override
    public void run(JobConfiguration configuration, Environment environment) throws Exception {
        jobManager = new JobManager(configuration, jobs);
        environment.lifecycle().manage(jobManager);
    }

    @Override
    public void initialize(Bootstrap bootstrap) {
        // add shared metrics registry to be used by Jobs, since defaultRegistry
        // has been removed
        SharedMetricRegistries.add(Job.DROPWIZARD_JOBS_KEY, bootstrap.getMetricRegistry());
    }

    public Scheduler getScheduler() {
        return jobManager.getScheduler();
    }

}