/**
* Copyright 2010 JBoss Inc
*
* 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 org.drools.task.service;
import org.drools.SystemEventListener;
import org.drools.eventmessaging.EventKeys;
import org.drools.task.*;
import org.drools.task.event.MessagingTaskEventListener;
import org.drools.task.event.TaskEventListener;
import org.drools.task.event.TaskEventSupport;
import org.drools.task.query.DeadlineSummary;
import org.drools.task.query.TaskSummary;
import org.mvel2.MVEL;
import org.mvel2.ParserContext;
import org.mvel2.compiler.ExpressionCompiler;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Callable;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
public class TaskService {
EntityManagerFactory emf;
EntityManager em;
ScheduledThreadPoolExecutor scheduler;
private EscalatedDeadlineHandler escalatedDeadlineHandler;
private UserInfo userInfo;
private TaskEventSupport eventSupport;
private EventKeys eventKeys;
/**
* Listener used for logging
*/
private SystemEventListener systemEventListener;
Map<Operation, List<OperationCommand>> operations;
public TaskService(EntityManagerFactory emf,SystemEventListener systemEventListener) {
this(emf, systemEventListener, null);
}
public TaskService(EntityManagerFactory emf, SystemEventListener systemEventListener, EscalatedDeadlineHandler escalationHandler) {
this.emf = emf;
this.systemEventListener = systemEventListener;
this.em = emf.createEntityManager();
if (escalationHandler != null) {
this.escalatedDeadlineHandler = escalationHandler;
}
eventSupport = new TaskEventSupport();
eventKeys = new EventKeys();
eventSupport.addEventListener(new MessagingTaskEventListener(eventKeys));
scheduler = new ScheduledThreadPoolExecutor(3);
long now = System.currentTimeMillis();
for (Object object : em.createNamedQuery("UnescalatedDeadlines").getResultList()) {
DeadlineSummary summary = (DeadlineSummary) object;
schedule(new ScheduledTaskDeadline(summary.getTaskId(),
summary.getDeadlineId(),
this),
summary.getDate().getTime() - now);
}
Map vars = new HashMap();
//Reader reader;
Reader reader = new InputStreamReader(getClass().getResourceAsStream("operations-dsl.mvel"));
try {
operations = (Map<Operation, List<OperationCommand>>) eval(toString(reader), vars);
} catch (IOException e) {
throw new RuntimeException("Unable To initialise TaskService, could not load Operations DSL");
}
}
public TaskServiceSession createSession() {
return new TaskServiceSession(this, emf.createEntityManager());
}
public void schedule(ScheduledTaskDeadline deadline,
long delay) {
scheduler.schedule(deadline,
delay,
TimeUnit.MILLISECONDS);
}
public Map<Operation, List<OperationCommand>> getOperations() {
return operations;
}
public List<OperationCommand> getCommandsForOperation(Operation operation) {
return operations.get(operation);
}
public EventKeys getEventKeys() {
return eventKeys;
}
public void addEventListener(final TaskEventListener listener) {
this.eventSupport.addEventListener(listener);
}
public void removeEventListener(final TaskEventListener listener) {
this.eventSupport.removeEventListener(listener);
}
public List<TaskEventListener> getWorkingMemoryEventListeners() {
return this.eventSupport.getEventListeners();
}
public TaskEventSupport getEventSupport() {
return eventSupport;
}
public UserInfo getUserinfo() {
return userInfo;
}
public void setUserinfo(UserInfo userInfo) {
this.userInfo = userInfo;
}
public EntityManager getEntityManager() {
return em;
}
public void setEscalatedDeadlineHandler(EscalatedDeadlineHandler escalatedDeadlineHandler) {
this.escalatedDeadlineHandler = escalatedDeadlineHandler;
}
public void executeEscalatedDeadline(long taskId,
long deadlineId) {
EntityManager localEm = emf.createEntityManager();
Task task = localEm.find(Task.class,
taskId);
Deadline deadline = localEm.find(Deadline.class,
deadlineId);
if (escalatedDeadlineHandler == null) {
escalatedDeadlineHandler = new DefaultEscalatedDeadlineHandler();
}
escalatedDeadlineHandler.executeEscalatedDeadline(task,
deadline,
localEm,
this);
localEm.close();
}
public static String toString(Reader reader) throws IOException {
int charValue ;
StringBuffer sb = new StringBuffer(1024);
while ((charValue = reader.read()) != -1) {
sb.append((char) charValue);
}
return sb.toString();
}
private static Map<String, Class> inputs = new HashMap<String, Class>();
public static Map<String, Class> getInputs() {
synchronized (inputs) {
if (inputs.isEmpty()) {
// org.drools.task
inputs.put("AccessType", AccessType.class);
inputs.put("AllowedToDelegate", AllowedToDelegate.class);
inputs.put("Attachment", Attachment.class);
inputs.put("BooleanExpression", BooleanExpression.class);
inputs.put("Comment", Comment.class);
inputs.put("Content", Content.class);
inputs.put("Deadline", Deadline.class);
inputs.put("Deadlines", Deadlines.class);
inputs.put("Delegation", Delegation.class);
inputs.put("EmailNotification", EmailNotification.class);
inputs.put("EmailNotificationHeader", EmailNotificationHeader.class);
inputs.put("Escalation", Escalation.class);
inputs.put("Group", Group.class);
inputs.put("I18NText", I18NText.class);
inputs.put("Notification", Notification.class);
inputs.put("NotificationType", NotificationType.class);
inputs.put("OrganizationalEntity", OrganizationalEntity.class);
inputs.put("PeopleAssignments", PeopleAssignments.class);
inputs.put("Reassignment", Reassignment.class);
inputs.put("Status", Status.class);
inputs.put("StatusChange", StatusChange.class);
inputs.put("Task", Task.class);
inputs.put("TaskData", TaskData.class);
inputs.put("User", User.class);
inputs.put("UserInfo", UserInfo.class);
inputs.put("WorkItemNotification", WorkItemNotification.class);
// org.drools.task.service
inputs.put("Allowed", Allowed.class);
inputs.put("Command", Command.class);
inputs.put("CommandName", CommandName.class);
inputs.put("ContentData", ContentData.class);
inputs.put("Operation", Operation.class);
inputs.put("Operation.Claim", Operation.class);
inputs.put("OperationCommand", OperationCommand.class);
// org.drools.task.query
inputs.put("DeadlineSummary", DeadlineSummary.class);
inputs.put("TaskSummary", TaskSummary.class);
}
return inputs;
}
}
public Object eval(String str,
Map vars) {
ExpressionCompiler compiler = new ExpressionCompiler(str.trim());
ParserContext context = new ParserContext();
context.addPackageImport("org.drools.task");
context.addPackageImport("org.drools.task.service");
context.addPackageImport("org.drools.task.query");
context.addPackageImport("java.util");
return MVEL.executeExpression(compiler.compile(context),
vars);
}
public static class ScheduledTaskDeadline
implements
Callable {
private long taskId;
private long deadlineId;
private TaskService service;
public ScheduledTaskDeadline(long taskId,
long deadlineId,
TaskService service) {
this.taskId = taskId;
this.deadlineId = deadlineId;
this.service = service;
}
public long getTaskId() {
return taskId;
}
public long getDeadlineId() {
return deadlineId;
}
public Object call() throws Exception {
try {
service.executeEscalatedDeadline(taskId,
deadlineId);
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + (int) (deadlineId ^ (deadlineId >>> 32));
result = prime * result + (int) (taskId ^ (taskId >>> 32));
return result;
}
@Override
public boolean equals(Object obj) {
if (this == obj) return true;
if (obj == null) return false;
if (!(obj instanceof ScheduledTaskDeadline)) return false;
ScheduledTaskDeadline other = (ScheduledTaskDeadline) obj;
if (deadlineId != other.deadlineId) return false;
if (taskId != other.taskId) return false;
return true;
}
}
}