Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.rapidoid.job.Jobs Maven / Gradle / Ivy
/*-
* #%L
* rapidoid-commons
* %%
* Copyright (C) 2014 - 2017 Nikolche Mihajlovski and contributors
* %%
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* #L%
*/
package org.rapidoid.job;
import org.rapidoid.activity.RapidoidThreadFactory;
import org.rapidoid.annotation.Authors;
import org.rapidoid.annotation.Since;
import org.rapidoid.concurrent.Callback;
import org.rapidoid.config.Conf;
import org.rapidoid.config.Config;
import org.rapidoid.config.RapidoidInitializer;
import org.rapidoid.ctx.Ctx;
import org.rapidoid.ctx.Ctxs;
import org.rapidoid.ctx.WithContext;
import org.rapidoid.log.Log;
import org.rapidoid.u.U;
import org.rapidoid.util.Once;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicLong;
@Authors("Nikolche Mihajlovski")
@Since("4.1.0")
public class Jobs extends RapidoidInitializer {
public static final Config JOBS = Conf.JOBS;
private static final AtomicLong errorCounter = new AtomicLong();
private static ScheduledThreadPoolExecutor SCHEDULER;
private static ThreadPoolExecutor EXECUTOR;
private static final Once init = new Once();
private Jobs() {
}
public static synchronized void reset() {
errorCounter.set(0);
}
public static synchronized ScheduledExecutorService scheduler() {
if (SCHEDULER == null) {
int threads = JOBS.sub("scheduler").entry("threads").or(64);
SCHEDULER = new ScheduledThreadPoolExecutor(threads, new RapidoidThreadFactory("scheduler", true));
new ManageableExecutor("scheduler", SCHEDULER);
if (init.go()) init();
}
return SCHEDULER;
}
public static synchronized Executor executor() {
if (EXECUTOR == null) {
int threads = JOBS.sub("executor").entry("threads").or(64);
int maxThreads = JOBS.sub("executor").entry("maxThreads").or(1024);
int maxQueueSize = JOBS.sub("executor").entry("maxQueueSize").or(1000000);
BlockingQueue queue = new ArrayBlockingQueue<>(maxQueueSize);
EXECUTOR = new ThreadPoolExecutor(threads, maxThreads, 300, TimeUnit.SECONDS, queue, new RapidoidThreadFactory("executor", true));
new ManageableExecutor("executor", EXECUTOR);
if (init.go()) init();
}
return EXECUTOR;
}
private static void init() {
Runtime.getRuntime().addShutdownHook(new Thread() {
public void run() {
shutdownNow();
}
});
}
public static ScheduledFuture schedule(Runnable job, long delay, TimeUnit unit) {
return scheduler().schedule(wrap(job), delay, unit);
}
public static ScheduledFuture schedule(Callable job, long delay, TimeUnit unit, Callback callback) {
return schedule(callbackJob(job, callback), delay, unit);
}
public static ScheduledFuture scheduleAtFixedRate(Runnable job, long initialDelay, long period, TimeUnit unit) {
return scheduler().scheduleAtFixedRate(wrap(job), initialDelay, period, unit);
}
public static ScheduledFuture scheduleAtFixedRate(Callable job, long initialDelay, long period,
TimeUnit unit, Callback callback) {
return scheduleAtFixedRate(callbackJob(job, callback), initialDelay, period, unit);
}
public static ScheduledFuture scheduleWithFixedDelay(Runnable job, long initialDelay, long delay, TimeUnit unit) {
return scheduler().scheduleWithFixedDelay(wrap(job), initialDelay, delay, unit);
}
public static ScheduledFuture scheduleWithFixedDelay(Callable job, long initialDelay, long delay,
TimeUnit unit, Callback callback) {
return scheduleWithFixedDelay(callbackJob(job, callback), initialDelay, delay, unit);
}
public static void execute(Runnable job) {
ContextPreservingJobWrapper jobWrapper = wrap(job);
try {
executor().execute(jobWrapper);
} catch (RejectedExecutionException e) {
Log.warn("Job execution was rejected!", "job", job);
}
}
public static void executeAndWait(Runnable job) {
ContextPreservingJobWrapper jobWrapper = wrap(job);
try {
executor().execute(jobWrapper);
} catch (RejectedExecutionException e) {
Log.warn("Job execution was rejected!", "job", job);
}
while (!jobWrapper.isDone()) {
U.sleep(10);
}
}
public static void execute(Callable job, Callback callback) {
execute(callbackJob(job, callback));
}
public static ContextPreservingJobWrapper wrap(Runnable job) {
Ctx ctx = Ctxs.get();
if (ctx != null) {
// increment reference counter
ctx = ctx.span(); // currently the same ctx is returned
}
return new ContextPreservingJobWrapper(job, ctx);
}
public static void call(Callback callback, T result, Throwable error) {
Jobs.execute(new CallbackExecutorJob(callback, result, error));
}
private static Runnable callbackJob(final Callable job, final Callback callback) {
return new Runnable() {
@Override
public void run() {
T result;
try {
result = job.call();
} catch (Throwable e) {
call(callback, null, e);
return;
}
call(callback, result, null);
}
};
}
public static void executeInContext(WithContext context, Runnable action) {
try {
executor().execute(new PredefinedContextJobWrapper(context, action));
} catch (RejectedExecutionException e) {
Log.warn("The job was rejected by the executor/scheduler!", "context", context.tag());
}
}
public static JobsDelayDSL after(long delay) {
return new JobsDelayDSL(delay);
}
public static JobsDSL after(long delay, TimeUnit unit) {
return new JobsDSL(delay, -1, unit);
}
public static JobsDSL every(long period, TimeUnit unit) {
return new JobsDSL(-1, period, unit);
}
public static AtomicLong errorCounter() {
return errorCounter;
}
public static synchronized void shutdown() {
if (EXECUTOR != null) {
EXECUTOR.shutdown();
EXECUTOR = null;
}
if (SCHEDULER != null) {
SCHEDULER.shutdown();
SCHEDULER = null;
}
}
public static synchronized void shutdownNow() {
if (EXECUTOR != null) {
EXECUTOR.shutdownNow();
EXECUTOR = null;
}
if (SCHEDULER != null) {
SCHEDULER.shutdownNow();
SCHEDULER = null;
}
}
}
