ru.ilb.common.jaxrs.async.AsyncTaskManager Maven / Gradle / Ivy
/*
* Copyright 2016 Bystrobank
*
* 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.
*/
package ru.ilb.common.jaxrs.async;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.Callable;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriBuilder;
import javax.ws.rs.core.UriInfo;
/**
*
* @author slavb
*/
public class AsyncTaskManager {
private final Map tasks = new ConcurrentHashMap<>();
private final ExecutorService taskExecutor = Executors.newCachedThreadPool();
public void destroy() {
for (Map.Entry entry : tasks.entrySet()) {
if (!entry.getValue().getFuture().isDone()) {
entry.getValue().getFuture().cancel(true);
}
}
}
public AsyncTask submit(Callable clbl) {
String taskId = UUID.randomUUID().toString();
Future future = taskExecutor.submit(clbl);
AsyncTask task = new AsyncTask(taskId, future);
tasks.put(taskId, task);
return task;
}
public AsyncTask get(String taskId) {
return tasks.get(taskId);
}
public Response execute(Callable clbl, UriInfo uriInfo) {
Response response;
MultivaluedMap queryParams = uriInfo.getQueryParameters();
String mode = queryParams.getFirst("mode");
String seeOther = queryParams.getFirst("seeOther");
String taskId = queryParams.getFirst("taskId");
if ("wait".equals(mode)) {
String i = queryParams.getFirst("i");
AsyncTask task = get(taskId);
if (task == null) {
//Returns 410, “Gone” if job doesn’t exist anymore
response = Response.status(Response.Status.GONE).type("text/plain; charset=UTF-8").entity("Задача " + taskId + " не найдена").build();
} else if (!task.getFuture().isDone()) {
UriBuilder uri = uriInfo.getAbsolutePathBuilder().queryParam("mode", "wait").queryParam("taskId", taskId).queryParam("i", Integer.parseInt(i) + 1);
if (seeOther != null) {
uri.queryParam("seeOther", seeOther);
int timeout = 2;
if (seeOther.length() > 0) {
timeout = Integer.parseInt(seeOther);
}
try {
TimeUnit.SECONDS.sleep(timeout);
} catch (InterruptedException ex) {
throw new RuntimeException(ex);
}
response = Response.seeOther(uri.build()).build();
} else {
response = Response.status(Response.Status.ACCEPTED).type("text/plain; charset=UTF-8").entity("Выполняется.. ").header("Refresh", "2;" + uri.build().toString()).build();
}
} else {
try {
if (task.getFuture().get() instanceof Response) {
response = (Response) task.getFuture().get();
} else {
response = Response.status(Response.Status.OK).entity(task.getFuture().get()).type("application/xml").build();
}
remove(taskId);
} catch (InterruptedException ex) {
throw new RuntimeException(ex);
} catch (ExecutionException ex) {
if (ex.getCause() != null) {
throw (RuntimeException) ex.getCause();
} else {
throw new RuntimeException(ex);
}
}
}
} else if ("cancel".equals(mode)) {
AsyncTask task = get(taskId);
if (task == null) {
//Returns 410, “Gone” if job doesn’t exist anymore
response = Response.status(Response.Status.GONE).type("text/plain; charset=UTF-8").entity("Задача " + taskId + " не найдена").build();
} else if (!task.getFuture().isDone()) {
task.getFuture().cancel(true);
response = Response.status(Response.Status.OK).type("text/plain; charset=UTF-8").entity("Задача " + taskId + " отменена").build();
remove(taskId);
} else {
response = Response.status(Response.Status.OK).type("text/plain; charset=UTF-8").entity("Задача " + taskId + " уже завершена").build();
}
} else {
AsyncTask task = submit(clbl);
UriBuilder uri = uriInfo.getAbsolutePathBuilder().queryParam("mode", "wait").queryParam("taskId", task.getTaskId()).queryParam("i", 0);
if (seeOther != null) {
uri.queryParam("seeOther", seeOther);
}
response = Response.seeOther(uri.build()).build();
}
return response;
}
public AsyncTask remove(String taskId) {
return tasks.remove(taskId);
}
}