com.jaeksoft.searchlib.scheduler.TaskManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of opensearchserver Show documentation
OpenSearchServer is a powerful, enterprise-class, search engine program. Using the web user interface, the crawlers (web, file, database, ...) and the REST/RESTFul API you will be able to integrate quickly and easily advanced full-text search capabilities in your application. OpenSearchServer runs on Windows and Linux/Unix/BSD.
The newest version!
/**
 * License Agreement for OpenSearchServer
 * 
 * Copyright (C) 2010-2013 Emmanuel Keller / Jaeksoft
 * 

 * http://www.open-search-server.com
 * 

 * This file is part of OpenSearchServer.
 * 

 * OpenSearchServer is free software: you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 

 * OpenSearchServer is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even 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 OpenSearchServer.
 * If not, see .
 **/

package com.jaeksoft.searchlib.scheduler;

import com.jaeksoft.searchlib.*;
import com.jaeksoft.searchlib.util.ReadWriteLock;
import com.jaeksoft.searchlib.util.Variables;
import com.jaeksoft.searchlib.web.StartStopListener;
import org.quartz.*;
import org.quartz.impl.DirectSchedulerFactory;

import javax.naming.NamingException;
import java.util.List;
import java.util.Set;

import static org.quartz.CronScheduleBuilder.cronSchedule;
import static org.quartz.JobBuilder.newJob;
import static org.quartz.JobKey.jobKey;
import static org.quartz.TriggerBuilder.newTrigger;
import static org.quartz.impl.matchers.GroupMatcher.jobGroupEquals;

public class TaskManager {

	public static class TaskManagerJob implements Job {

		public TaskManagerJob() {
		}

		@Override
		public void execute(JobExecutionContext context) throws JobExecutionException {
			if (StartStopListener.isShutdown())
				throw new JobExecutionException("Aborted (application stopped)");
			JobDetail jobDetail = context.getJobDetail();
			JobKey jobKey = jobDetail.getKey();
			try {
				INSTANCE.executeJob(jobKey.getGroup(), jobKey.getName());
			} catch (SearchLibException e) {
				Logging.error(e);
			} catch (NamingException e) {
				Logging.error(e);
			}
		}

	}

	private final ReadWriteLock rwl = new ReadWriteLock();

	private Scheduler scheduler;

	private DirectSchedulerFactory schedulerFactory;

	private final static TaskManager INSTANCE = new TaskManager();

	private int threadPoolSize;

	public static TaskManager getInstance() {
		return INSTANCE;
	}

	private TaskManager() {
		scheduler = null;
		schedulerFactory = null;
	}

	public void start() throws SearchLibException {
		if (ClientFactory.INSTANCE.properties.isDisableScheduler())
			Logging.warn("The scheduler is disabled ");
		rwl.w.lock();
		try {
			if (scheduler == null) {
				if (schedulerFactory == null)
					schedulerFactory = DirectSchedulerFactory.getInstance();
				threadPoolSize = ClientFactory.INSTANCE.getSchedulerThreadPoolSize().getValue();
				schedulerFactory.createVolatileScheduler(threadPoolSize);
				scheduler = schedulerFactory.getScheduler();
			}
			scheduler.start();
		} catch (SchedulerException e) {
			throw new SearchLibException(e);
		} finally {
			rwl.w.unlock();
		}
	}

	public void stop() throws SearchLibException {
		rwl.w.lock();
		try {
			if (scheduler != null) {
				scheduler.shutdown();
				scheduler = null;
			}
		} catch (SchedulerException e) {
			throw new SearchLibException(e);
		} finally {
			rwl.w.unlock();
		}
	}

	private void checkSchedulerAvailable() throws SearchLibException {
		if (scheduler == null)
			throw new SearchLibException("The scheduler is not available.");
	}

	public void cronJob(String indexName, String jobName, TaskCronExpression cron) throws SearchLibException {
		rwl.w.lock();
		try {
			checkSchedulerAvailable();
			Trigger trigger =
					newTrigger().withIdentity(jobName, indexName).withSchedule(cronSchedule(cron.getStringExpression()))
							.build();

			// CHECK IF IT ALREADY EXIST
			JobKey jobKey = jobKey(jobName, indexName);
			if (scheduler.checkExists(jobKey)) {
				List triggers = scheduler.getTriggersOfJob(jobKey);
				if (triggers != null) {
					for (Trigger tr : triggers) {
						if (tr instanceof CronTrigger) {
							CronTrigger ctr = (CronTrigger) tr;
							if (ctr.getCronExpression().equals(cron.getStringExpression()))
								return;
						}
					}
				}
				scheduler.deleteJob(jobKey);
			}

			JobDetail job = newJob(TaskManagerJob.class).withIdentity(jobName, indexName).build();
			scheduler.scheduleJob(job, trigger);
		} catch (SchedulerException e) {
			throw new SearchLibException(e);
		} finally {
			rwl.w.unlock();
		}
	}

	/**
	 * Synchronous execution of a full Job
	 *
	 * @param indexName
	 * @param jobName
	 * @throws SearchLibException
	 * @throws NamingException
	 */
	public void executeJob(String indexName, String jobName) throws SearchLibException, NamingException {
		if (ClientFactory.INSTANCE.properties.isDisableScheduler())
			throw new SearchLibException("The scheduler is disabled.");
		Client client = ClientCatalog.getClient(indexName);
		if (client == null)
			throw new SearchLibException("Client not found: " + indexName);
		JobList jobList = client.getJobList();
		JobItem jobItem = jobList.get(jobName);
		if (jobItem == null)
			throw new SearchLibException("Job not found: " + jobName);
		jobItem.run(client, null);
	}

	/**
	 * Asynchronous execution of a full Job
	 *
	 * @param client
	 * @param jobItem
	 * @return
	 * @throws InterruptedException
	 */
	public ImmediateExecution executeJob(Client client, JobItem jobItem, Variables variables)
			throws InterruptedException {
		if (ClientFactory.INSTANCE.properties.isDisableScheduler())
			throw new InterruptedException("The scheduler is disabled.");
		ImmediateExecution execution = new ImmediateExecution(client, jobItem, variables);
		execution.execute(300);
		jobItem.waitForStart(300);
		return execution;
	}

	/**
	 * Asynchronous execution of a Task
	 *
	 * @param client
	 * @param taskItem
	 * @param taskLog
	 * @return
	 * @throws InterruptedException
	 */
	public static ImmediateExecution executeTask(Client client, TaskItem taskItem, TaskLog taskLog)
			throws InterruptedException {
		if (ClientFactory.INSTANCE.properties.isDisableScheduler())
			throw new InterruptedException("The scheduler is disabled.");
		if (taskLog == null)
			taskLog = new TaskLog(taskItem, false, false);
		ImmediateExecution execution = new ImmediateExecution(client, taskItem, taskLog);
		execution.execute(300);
		taskItem.waitForStart(300);
		return execution;
	}

	public void removeJob(String indexName, String jobName) throws SearchLibException {
		rwl.w.lock();
		try {
			scheduler.deleteJob(jobKey(jobName, indexName));
		} catch (SchedulerException e) {
			throw new SearchLibException(e);
		} finally {
			rwl.w.unlock();
		}
	}

	public void removeJobs(String indexName) throws SearchLibException {
		rwl.w.lock();
		try {
			Set jobKeySet = scheduler.getJobKeys(jobGroupEquals(indexName));
			for (JobKey jobKey : jobKeySet)
				scheduler.deleteJob(jobKey);
		} catch (SchedulerException e) {
			throw new SearchLibException(e);
		} finally {
			rwl.w.unlock();
		}
	}

	public String[] getActiveJobs(String indexName) throws SchedulerException {
		rwl.r.lock();
		try {
			Set jobKeySet = scheduler.getJobKeys(jobGroupEquals(indexName));
			String[] jobs = new String[jobKeySet.size()];
			int i = 0;
			for (JobKey jobKey : jobKeySet)
				jobs[i++] = jobKey.getName();
			return jobs;
		} finally {
			rwl.r.unlock();
		}
	}

	public String getStatus() throws SchedulerException {
		rwl.r.lock();
		try {
			StringBuilder sb = new StringBuilder();
			if (scheduler == null)
				sb.append("No scheduler. ");
			else {
				if (scheduler.isInStandbyMode())
					sb.append("Standby. ");
				if (scheduler.isStarted())
					sb.append("Started. ");
				if (scheduler.isShutdown())
					sb.append("Shutdown. ");
			}
			if (schedulerFactory == null)
				sb.append("No factory. ");
			else
				sb.append(schedulerFactory.getAllSchedulers().size());
			return sb.toString();
		} finally {
			rwl.r.unlock();
		}
	}

	public int getThreadPoolSize() {
		rwl.r.lock();
		try {
			return threadPoolSize;
		} finally {
			rwl.r.unlock();
		}
	}
}