use of org.kie.internal.task.api.assignment.Assignment in project jbpm by kiegroup.
the class BusinessRuleAssignmentStrategy method apply.
@SuppressWarnings("unchecked")
@Override
public Assignment apply(Task task, TaskContext context, String excludedUser) {
if (!active) {
logger.debug("{} strategy is not active", this);
return null;
}
logger.debug("Using rules to assign actual owner to task {}", task);
KieSession kieSession = this.kieContainer.newKieSession();
try {
context.loadTaskVariables(task);
kieSession.insert(task);
kieSession.fireAllRules();
Set<Assignment> assignments = new TreeSet<>();
String queryName = System.getProperty("org.jbpm.task.assignment.rules.query");
if (queryName != null) {
logger.debug("Query {} is going to be used to retrieve results from working memory", queryName);
QueryResults results = kieSession.getQueryResults(queryName);
results.forEach(row -> assignments.add((Assignment) row.get("assignment")));
} else {
logger.debug("No query defined, retrieving all facts of type Assignment");
Collection<Assignment> results = (Collection<Assignment>) kieSession.getObjects(new ClassObjectFilter(Assignment.class));
assignments.addAll(results);
}
logger.debug("Rule evaluation completed with selected assignments of {}", assignments);
if (assignments.isEmpty()) {
logger.debug("No assignments found by BusinessRule strategy");
return null;
}
Assignment selected = assignments.iterator().next();
logger.debug("Selected assignment is {} for task {}", selected, task);
return selected;
} finally {
kieSession.dispose();
logger.debug("KieSession in BusinessRule disposed");
}
}
use of org.kie.internal.task.api.assignment.Assignment in project jbpm by kiegroup.
the class LoadBalanceAssignmentStrategy method apply.
@Override
public Assignment apply(Task task, TaskContext taskContext, String excludedUser) {
UserInfo userInfo = (UserInfo) ((org.jbpm.services.task.commands.TaskContext) taskContext).get(EnvironmentName.TASK_USER_INFO);
List<OrganizationalEntity> excluded = (getExcludedEntities(task, userInfo));
// Get the the users from the task's the potential owners, making sure that excluded users are not included
List<OrganizationalEntity> potentialOwners = task.getPeopleAssignments().getPotentialOwners().stream().filter(oe -> oe instanceof User && !excluded.contains(oe) && !oe.getId().equals(excludedUser)).collect(Collectors.toList());
// Get the users belonging to groups that are potential owners
task.getPeopleAssignments().getPotentialOwners().stream().filter(oe -> oe instanceof Group).forEach(oe -> {
Iterator<OrganizationalEntity> groupUsers = userInfo.getMembersForGroup((Group) oe);
if (groupUsers != null) {
groupUsers.forEachRemaining(user -> {
if (user != null && !excluded.contains(user) && !potentialOwners.contains(user) && !user.getId().equals(excludedUser)) {
potentialOwners.add(user);
}
});
}
});
logger.debug("Asking the load calculator [{}] for task loads for the users {}", calculator.getIdentifier(), potentialOwners);
List<User> users = potentialOwners.stream().map(entityToUser).collect(Collectors.toList());
Collection<UserTaskLoad> loads = calculator.getUserTaskLoads(users, taskContext);
UserTaskLoad lightestLoad = loads.stream().min(UserTaskLoad::compareTo).orElse(null);
return lightestLoad != null ? new Assignment(lightestLoad.getUser().getId()) : null;
}
use of org.kie.internal.task.api.assignment.Assignment in project jbpm by kiegroup.
the class PotentialOwnerBusynessAssignmentStrategy method apply.
@Override
public Assignment apply(Task task, TaskContext context, String excludedUser) {
if (task.getPeopleAssignments().getPotentialOwners().isEmpty()) {
logger.debug("No potential owners in the task {} can't auto assign", task);
return null;
}
List<OrganizationalEntity> potentialOwners = new ArrayList<>(task.getPeopleAssignments().getPotentialOwners());
Set<String> resolvedUsers = new TreeSet<>(Collections.reverseOrder());
List<OrganizationalEntity> excludedOwners = ((InternalPeopleAssignments) task.getPeopleAssignments()).getExcludedOwners();
potentialOwners.stream().filter(po -> po instanceof User && !excludedOwners.contains(po)).forEach(po -> resolvedUsers.add(po.getId()));
UserInfo userInfo = (UserInfo) ((org.jbpm.services.task.commands.TaskContext) context).get(EnvironmentName.TASK_USER_INFO);
if (userInfo != null) {
logger.debug("Groups going to be resolved by {}", userInfo);
potentialOwners.stream().filter(po -> po instanceof Group && !excludedOwners.contains(po)).forEach(po -> {
Iterator<OrganizationalEntity> usersOfGroup = userInfo.getMembersForGroup((Group) po);
if (usersOfGroup != null) {
while (usersOfGroup.hasNext()) {
OrganizationalEntity entity = usersOfGroup.next();
if (!excludedOwners.contains(entity)) {
resolvedUsers.add(entity.getId());
}
}
}
});
}
logger.debug("Resolved users eligible for task {} assignments are {}", task, resolvedUsers);
if (excludedUser != null) {
logger.debug("Removing excluded user {} from the list of eligible users", excludedUser);
resolvedUsers.remove(excludedUser);
}
TaskPersistenceContext persistenceContext = ((org.jbpm.services.task.commands.TaskContext) context).getPersistenceContext();
Map<String, Object> params = new HashMap<>();
params.put("owners", resolvedUsers);
logger.debug("DB query to be used for finding assignments :: '{}'", getQuery());
List<Assignment> assignments = persistenceContext.queryStringWithParametersInTransaction(getQuery(), params, ClassUtil.<List<Assignment>>castClass(List.class));
if (assignments.size() < resolvedUsers.size()) {
logger.debug("Not all eligible users found in db, adding missing bits (eligible {}, found in db {})", resolvedUsers, assignments);
// in case not all users have already assigned tasks added them to the list so can get the tasks
resolvedUsers.forEach(user -> {
Assignment assignment = new AssignmentImpl(user);
if (!assignments.contains(assignment)) {
// always add missing users to the top of the list so they get assigned first
assignments.add(0, assignment);
}
});
}
if (assignments.isEmpty()) {
logger.debug("No assignments found for task {}", task);
return null;
}
logger.debug("Following assignments {} were found for task {}", assignments, task);
// select first from the top of the list as it has the least assigned tasks
Assignment selected = assignments.get(0);
logger.debug("Retruning first of found assignments {}", selected);
return new Assignment(selected.getUser());
}
use of org.kie.internal.task.api.assignment.Assignment in project jbpm by kiegroup.
the class RoundRobinAssignmentStrategy method apply.
@Override
public Assignment apply(Task task, TaskContext taskContext, String excludedUser) {
UserInfo userInfo = (UserInfo) ((org.jbpm.services.task.commands.TaskContext) taskContext).get(EnvironmentName.TASK_USER_INFO);
List<OrganizationalEntity> excluded = getExcludedEntities(task, userInfo);
// Get the the users from the task's the potential owners
List<OrganizationalEntity> potentialOwners = task.getPeopleAssignments().getPotentialOwners().stream().filter(oe -> oe instanceof User && !excluded.contains(oe)).collect(Collectors.toList());
// Get the users belonging to groups that are potential owners
task.getPeopleAssignments().getPotentialOwners().stream().filter(oe -> oe instanceof Group).forEach(oe -> {
Iterator<OrganizationalEntity> groupUsers = userInfo.getMembersForGroup((Group) oe);
if (groupUsers != null) {
groupUsers.forEachRemaining(user -> {
if (user != null && !excluded.contains(user) && !potentialOwners.contains(user)) {
potentialOwners.add(user);
}
});
}
});
if (excludedUser != null) {
logger.debug("Removing excluded user {} from the list of eligible users", excludedUser);
potentialOwners.removeIf(entity -> entity.getId().equals(excludedUser));
}
String queueName = getQueueName(task);
CircularQueue<OrganizationalEntity> mappedQueue = synchronizedQueue(queueName, potentialOwners);
OrganizationalEntity owner = mappedQueue.take();
return new Assignment(owner.getId());
}
use of org.kie.internal.task.api.assignment.Assignment in project jbpm by kiegroup.
the class AssignmentServiceImpl method assignTask.
@Override
public void assignTask(Task task, TaskContext context, String excludedUser) {
if (!isEnabled()) {
logger.debug("AssignmentService is not enabled - to enable it set system property '" + ENABLED_PROPERTY + "' to true");
return;
}
Assignment assignTo = this.strategy.apply(task, context, excludedUser);
if (assignTo == null || assignTo.getUser() == null) {
logger.warn("Strategy {} did not return any assignment for task {}", strategy, task);
return;
}
logger.debug("Actual owner returned by strategy {} is {} for task {}", strategy, assignTo, task);
User actualOwner = taskModelFactory.newUser(assignTo.getUser());
((InternalTaskData) task.getTaskData()).setActualOwner(actualOwner);
((InternalTaskData) task.getTaskData()).setStatus(Status.Reserved);
}
Aggregations