package org.drools.task.service;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.drools.SystemEventListener;
import org.drools.eventmessaging.EventKey;
import org.drools.task.Attachment;
import org.drools.task.Comment;
import org.drools.task.Content;
import org.drools.task.Task;
import org.drools.task.query.TaskSummary;
public class TaskServerHandler {
private final TaskService service;
private final Map<String, SessionWriter> clients;
/**
* Listener used for logging
*/
private final SystemEventListener systemEventListener;
public TaskServerHandler(TaskService service, SystemEventListener systemEventListener) {
this.service = service;
this.clients = new HashMap<String, SessionWriter>();
this.systemEventListener = systemEventListener;
}
public void exceptionCaught(SessionWriter session, Throwable cause) throws Exception {
systemEventListener.exception("Uncaught exception on Server", cause);
}
public void messageReceived(SessionWriter session, Object message) throws Exception {
Command cmd = (Command) message;
TaskServiceSession taskSession = service.createSession();
CommandName response = null;
try {
systemEventListener.debug("Message receieved on server : " + cmd.getName());
systemEventListener.debug("Arguments : " + Arrays.toString(cmd.getArguments().toArray()));
switch (cmd.getName()) {
case OperationRequest: {
response = CommandName.OperationResponse;
Operation operation = (Operation) cmd.getArguments().get(0);
systemEventListener.debug("Command receieved on server was operation of type: " + operation);
long taskId = (Long) cmd.getArguments().get(1);
String userId = (String) cmd.getArguments().get(2);
String targetEntityId = null;
ContentData data = null;
List<String> groupIds = null;
if (cmd.getArguments().size() > 3) {
targetEntityId = (String) cmd.getArguments().get(3);
if (cmd.getArguments().size() > 4) {
data = (ContentData) cmd.getArguments().get(4);
if (cmd.getArguments().size() > 5) {
groupIds = (List<String>) cmd.getArguments().get(5);
}
}
}
taskSession.taskOperation(operation, taskId, userId, targetEntityId, data, groupIds);
List args = Collections.emptyList();
Command resultsCmnd = new Command(cmd.getId(), CommandName.OperationResponse, args);
session.write(resultsCmnd);
break;
}
case GetTaskRequest: {
response = CommandName.GetTaskResponse;
long taskId = (Long) cmd.getArguments().get(0);
Task task = taskSession.getTask(taskId);
List args = new ArrayList(1);
args.add(task);
Command resultsCmnd = new Command(cmd.getId(), CommandName.GetTaskResponse, args);
session.write(resultsCmnd);
break;
}
case AddTaskRequest: {
response = CommandName.AddTaskResponse;
Task task = (Task) cmd.getArguments().get(0);
ContentData content = (ContentData) cmd.getArguments().get(1);
taskSession.addTask(task, content);
List args = new ArrayList(1);
args.add(task.getId());
Command resultsCmnd = new Command(cmd.getId(),
CommandName.AddTaskResponse,
args);
session.write(resultsCmnd);
break;
}
case AddCommentRequest: {
response = CommandName.AddCommentResponse;
Comment comment = (Comment) cmd.getArguments().get(1);
taskSession.addComment((Long) cmd.getArguments().get(0),
comment);
List args = new ArrayList(1);
args.add(comment.getId());
Command resultsCmnd = new Command(cmd.getId(),
CommandName.AddCommentResponse,
args);
session.write(resultsCmnd);
break;
}
case DeleteCommentRequest: {
response = CommandName.DeleteCommentResponse;
long taskId = (Long) cmd.getArguments().get(0);
long commentId = (Long) cmd.getArguments().get(1);
taskSession.deleteComment(taskId,
commentId);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.DeleteCommentResponse,
Collections.emptyList());
session.write(resultsCmnd);
break;
}
case AddAttachmentRequest: {
response = CommandName.AddAttachmentResponse;
Attachment attachment = (Attachment) cmd.getArguments().get(1);
Content content = (Content) cmd.getArguments().get(2);
taskSession.addAttachment((Long) cmd.getArguments().get(0),
attachment,
content);
List args = new ArrayList(2);
args.add(attachment.getId());
args.add(content.getId());
Command resultsCmnd = new Command(cmd.getId(),
CommandName.AddAttachmentResponse,
args);
session.write(resultsCmnd);
break;
}
case DeleteAttachmentRequest: {
response = CommandName.DeleteAttachmentResponse;
long taskId = (Long) cmd.getArguments().get(0);
long attachmentId = (Long) cmd.getArguments().get(1);
long contentId = (Long) cmd.getArguments().get(2);
taskSession.deleteAttachment(taskId,
attachmentId,
contentId);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.DeleteAttachmentResponse,
Collections.emptyList());
session.write(resultsCmnd);
break;
}
case SetDocumentContentRequest: {
response = CommandName.SetDocumentContentResponse;
long taskId = (Long) cmd.getArguments().get(0);
Content content = (Content) cmd.getArguments().get(1);
taskSession.setDocumentContent(taskId,
content);
List args = new ArrayList(1);
args.add(content.getId());
Command resultsCmnd = new Command(cmd.getId(),
CommandName.SetDocumentContentResponse,
args);
session.write(resultsCmnd);
break;
}
case GetContentRequest: {
response = CommandName.GetContentResponse;
long contentId = (Long) cmd.getArguments().get(0);
Content content = taskSession.getContent(contentId);
List args = new ArrayList(1);
args.add(content);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.GetContentResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTaskByWorkItemId: {
response = CommandName.QueryTaskByWorkItemIdResponse;
Task result = taskSession.getTaskByWorkItemId((Long) cmd.getArguments().get(0));
List args = new ArrayList(1);
args.add(result);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskByWorkItemIdResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTasksOwned: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getTasksOwned((String) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTasksAssignedAsBusinessAdministrator: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getTasksAssignedAsBusinessAdministrator((String) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTasksAssignedAsPotentialOwner: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getTasksAssignedAsPotentialOwner((String) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTasksAssignedAsPotentialOwnerWithGroup: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getTasksAssignedAsPotentialOwner(
(String) cmd.getArguments().get(0),
(List<String>) cmd.getArguments().get(1),
(String) cmd.getArguments().get(2));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTasksAssignedAsPotentialOwnerByGroup: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getTasksAssignedAsPotentialOwnerByGroup((String) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QuerySubTasksAssignedAsPotentialOwner: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getSubTasksAssignedAsPotentialOwner((Long) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1),
(String) cmd.getArguments().get(2));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryGetSubTasksByParentTaskId: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getSubTasksByParent((Long) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTasksAssignedAsTaskInitiator: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getTasksAssignedAsTaskInitiator((String) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTasksAssignedAsExcludedOwner: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getTasksAssignedAsExcludedOwner((String) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTasksAssignedAsRecipient: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getTasksAssignedAsRecipient((String) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case QueryTasksAssignedAsTaskStakeholder: {
response = CommandName.QueryTaskSummaryResponse;
List<TaskSummary> results = taskSession.getTasksAssignedAsTaskStakeholder((String) cmd.getArguments().get(0),
(String) cmd.getArguments().get(1));
List args = new ArrayList(1);
args.add(results);
Command resultsCmnd = new Command(cmd.getId(),
CommandName.QueryTaskSummaryResponse,
args);
session.write(resultsCmnd);
break;
}
case RegisterForEventRequest: {
response = CommandName.EventTriggerResponse;
EventKey key = (EventKey) cmd.getArguments().get(0);
boolean remove = (Boolean) cmd.getArguments().get(1);
String uuid = (String) cmd.getArguments().get(2);
clients.put(uuid, session);
EventTransport transport = new EventTransport(uuid,
cmd.getId(),
clients,
remove);
service.getEventKeys().register(key, transport);
break;
}
case RegisterClient: {
String uuid = (String) cmd.getArguments().get(0);
clients.put(uuid, session);
break;
}
default: {
systemEventListener.debug("Unknown command recieved on server");
}
}
} catch (RuntimeException e) {
systemEventListener.exception(e.getMessage(),e);
e.printStackTrace(System.err);
List<Object> list = new ArrayList<Object>(1);
list.add(e);
Command resultsCmnd = new Command(cmd.getId(), response, list);
session.write(resultsCmnd);
} finally {
taskSession.dispose();
}
}
// public void sessionIdle(SessionWriter session, IdleStatus status) throws Exception {
// systemEventListener.debug("Server IDLE " + session.getIdleCount(status));
// }
}