package org.drools.task.service;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import org.drools.SystemEventListener;
import org.drools.eventmessaging.EventResponseHandler;
import org.drools.eventmessaging.Payload;
import org.drools.task.Content;
import org.drools.task.Task;
import org.drools.task.query.TaskSummary;
public class TaskClientHandler {
private TaskClient client;
private SystemEventListener systemEventListener;
private final Map<Integer, ResponseHandler> responseHandlers;
public TaskClientHandler(Map<Integer, ResponseHandler> responseHandlers, SystemEventListener systemEventListener) {
this.responseHandlers = responseHandlers;
this.systemEventListener = systemEventListener;
}
public TaskClient getClient() {
return client;
}
public void setClient(TaskClient client) {
this.client = client;
}
public void exceptionCaught(SessionWriter session,
Throwable cause) throws Exception {
systemEventListener.exception("Uncaught exception on client", cause);
}
public void messageReceived(SessionWriter session,
Object message) throws Exception {
Command cmd = (Command) message;
systemEventListener.debug("Message receieved on client : " + cmd.getName());
systemEventListener.debug("Arguments : " + Arrays.toString(cmd.getArguments().toArray()));
switch (cmd.getName()) {
case OperationResponse: {
TaskOperationResponseHandler responseHandler = (TaskOperationResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
responseHandler.setIsDone(true);
}
}
break;
}
case GetTaskResponse: {
GetTaskResponseHandler responseHandler = (GetTaskResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
Task task = (Task) cmd.getArguments().get(0);
responseHandler.execute(task);
}
}
break;
}
case AddTaskResponse: {
AddTaskResponseHandler responseHandler = (AddTaskResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
long taskId = (Long) cmd.getArguments().get(0);
responseHandler.execute(taskId);
}
}
break;
}
case AddCommentResponse: {
AddCommentResponseHandler responseHandler = (AddCommentResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
long commentId = (Long) cmd.getArguments().get(0);
responseHandler.execute(commentId);
}
}
break;
}
case DeleteCommentResponse: {
DeleteCommentResponseHandler responseHandler = (DeleteCommentResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
responseHandler.setIsDone(true);
}
}
break;
}
case AddAttachmentResponse: {
AddAttachmentResponseHandler responseHandler = (AddAttachmentResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
long attachmentId = (Long) cmd.getArguments().get(0);
long contentId = (Long) cmd.getArguments().get(1);
responseHandler.execute(attachmentId,
contentId);
}
}
break;
}
case DeleteAttachmentResponse: {
DeleteAttachmentResponseHandler responseHandler = (DeleteAttachmentResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
responseHandler.setIsDone(true);
}
}
break;
}
case GetContentResponse: {
GetContentResponseHandler responseHandler = (GetContentResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
Content content = (Content) cmd.getArguments().get(0);
responseHandler.execute(content);
}
}
break;
}
case SetDocumentContentResponse: {
SetDocumentResponseHandler responseHandler = (SetDocumentResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
long contentId = (Long) cmd.getArguments().get(0);
responseHandler.execute(contentId);
}
}
break;
}
case QueryTaskSummaryResponse: {
TaskSummaryResponseHandler responseHandler = (TaskSummaryResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
List<TaskSummary> results = (List<TaskSummary>) cmd.getArguments().get(0);
responseHandler.execute(results);
}
}
break;
}
case QueryTaskByWorkItemIdResponse: {
GetTaskResponseHandler responseHandler = (GetTaskResponseHandler) responseHandlers.remove(cmd.getId());
if (responseHandler != null) {
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
Task result = (Task) cmd.getArguments().get(0);
responseHandler.execute(result);
}
}
break;
}
case EventTriggerResponse: {
EventResponseHandler responseHandler = (EventResponseHandler) responseHandlers.get(cmd.getId());
if (responseHandler != null) {
if (responseHandler.isRemove()) {
responseHandlers.remove(cmd.getId());
}
if (!cmd.getArguments().isEmpty() && cmd.getArguments().get(0) instanceof RuntimeException) {
responseHandler.setError((RuntimeException) cmd.getArguments().get(0));
} else {
Payload payload = (Payload) cmd.getArguments().get(0);
responseHandler.execute(payload);
}
}
break;
}
}
}
public static interface GetTaskResponseHandler
extends
ResponseHandler {
public void execute(Task task);
}
public static interface AddTaskResponseHandler
extends
ResponseHandler {
public void execute(long taskId);
}
public static interface TaskOperationResponseHandler
extends
ResponseHandler {
public void setIsDone(boolean done);
}
public static interface AddCommentResponseHandler
extends
ResponseHandler {
public void execute(long commentId);
}
public static interface DeleteCommentResponseHandler
extends
ResponseHandler {
public void setIsDone(boolean done);
}
public static interface AddAttachmentResponseHandler
extends
ResponseHandler {
public void execute(long attachmentId,
long contentId);
}
public static interface DeleteAttachmentResponseHandler
extends
ResponseHandler {
public void setIsDone(boolean done);
}
public static interface SetDocumentResponseHandler
extends
ResponseHandler {
public void execute(long contentId);
}
public static interface GetContentResponseHandler
extends
ResponseHandler {
public void execute(Content content);
}
public static interface TaskSummaryResponseHandler
extends
ResponseHandler {
public void execute(List<TaskSummary> results);
}
}