use of org.kie.server.services.taskassigning.core.model.solver.realtime.RemoveUserProblemFactChange in project droolsjbpm-integration by kiegroup.
the class SolutionChangesBuilderTest method assertChange.
private void assertChange(List<ProblemFactChange<TaskAssigningSolution>> result, int index, RemoveUserProblemFactChange expected) {
RemoveUserProblemFactChange change = (RemoveUserProblemFactChange) result.get(index);
assertUserEquals(expected.getUser(), change.getUser());
}
use of org.kie.server.services.taskassigning.core.model.solver.realtime.RemoveUserProblemFactChange in project droolsjbpm-integration by kiegroup.
the class SolutionChangesBuilderTest method addRemoveUserWhenHasNoTasksChange.
@Test
public void addRemoveUserWhenHasNoTasksChange() {
List<User> userList = mockUserList(mockUser(USER1_ID));
userList.get(0).setEnabled(false);
TaskAssigningSolution solution = mockSolution(Collections.emptyList(), userList);
List<ProblemFactChange<TaskAssigningSolution>> result = SolutionChangesBuilder.create().withSolution(solution).withTasks(Collections.emptyList()).withUserSystem(userSystemService).withContext(context).build();
RemoveUserProblemFactChange expected = new RemoveUserProblemFactChange(userList.get(0));
assertChangeIsTheChangeSetId(result, 0);
assertChange(result, 1, expected);
}
use of org.kie.server.services.taskassigning.core.model.solver.realtime.RemoveUserProblemFactChange in project droolsjbpm-integration by kiegroup.
the class SolutionChangesBuilder method build.
public List<ProblemFactChange<TaskAssigningSolution>> build() {
final Map<Long, Task> taskById = solution.getTaskList().stream().filter(IS_NOT_DUMMY).collect(Collectors.toMap(Task::getId, Function.identity()));
final Map<String, User> usersById = solution.getUserList().stream().collect(Collectors.toMap(User::getEntityId, Function.identity()));
final List<AddTaskProblemFactChange> newTaskChanges = new ArrayList<>();
final List<ReleaseTaskProblemFactChange> releasedTasksChanges = new ArrayList<>();
final List<RemoveTaskProblemFactChange> removedTaskChanges = new ArrayList<>();
final Set<Task> removedTasksSet = new HashSet<>();
final List<TaskPropertyChangeProblemFactChange> propertyChanges = new ArrayList<>();
final Map<String, List<IndexedElement<AssignTaskProblemFactChange>>> changesByUserId = new HashMap<>();
final List<AddUserProblemFactChange> newUserChanges = new ArrayList<>();
final List<ProblemFactChange<TaskAssigningSolution>> userUpdateChanges = new ArrayList<>();
final List<RemoveUserProblemFactChange> removableUserChanges = new ArrayList<>();
final List<TaskData> filteredTaskDataList = taskDataList.stream().filter(taskData -> !context.isProcessedTaskChange(taskData.getTaskId(), taskData.getLastModificationDate())).collect(Collectors.toList());
Task task;
for (TaskData taskData : filteredTaskDataList) {
task = taskById.remove(taskData.getTaskId());
if (task == null) {
addNewTaskChanges(taskData, usersById, newTaskChanges, changesByUserId);
} else {
addTaskChanges(task, taskData, usersById, releasedTasksChanges, removedTasksSet, propertyChanges, changesByUserId);
}
}
for (Task removedTask : removedTasksSet) {
removedTaskChanges.add(new RemoveTaskProblemFactChange(removedTask));
}
if (executeUsersUpdate) {
addUserChanges(usersById, newUserChanges, userUpdateChanges);
} else {
addRemovableUserChanges(changesByUserId, removableUserChanges);
}
List<ProblemFactChange<TaskAssigningSolution>> totalChanges = new ArrayList<>();
totalChanges.addAll(newUserChanges);
totalChanges.addAll(removedTaskChanges);
totalChanges.addAll(releasedTasksChanges);
changesByUserId.values().forEach(byUserChanges -> byUserChanges.forEach(change -> totalChanges.add(change.getElement())));
totalChanges.addAll(propertyChanges);
totalChanges.addAll(userUpdateChanges);
totalChanges.addAll(newTaskChanges);
totalChanges.addAll(removableUserChanges);
if (LOGGER.isTraceEnabled()) {
if (!totalChanges.isEmpty()) {
traceProgrammedChanges(LOGGER, removedTaskChanges, releasedTasksChanges, changesByUserId, propertyChanges, newTaskChanges, newUserChanges, userUpdateChanges, removableUserChanges);
} else {
LOGGER.trace("No changes has been calculated.");
}
}
applyWorkaroundForPLANNER241(solution, totalChanges);
if (!totalChanges.isEmpty()) {
totalChanges.add(0, scoreDirector -> context.setCurrentChangeSetId(context.nextChangeSetId()));
}
filteredTaskDataList.forEach(taskData -> context.setTaskChangeTime(taskData.getTaskId(), taskData.getLastModificationDate()));
return totalChanges;
}
use of org.kie.server.services.taskassigning.core.model.solver.realtime.RemoveUserProblemFactChange in project droolsjbpm-integration by kiegroup.
the class SolutionChangesBuilderTest method addRemoveUserWhenHasNonPinnedTasksChange.
@Test
public void addRemoveUserWhenHasNonPinnedTasksChange() {
User user = mockUser(USER1_ID);
user.setEnabled(false);
Task nonPinnedTask = mockTask(TASK_ID);
nonPinnedTask.setPinned(false);
nonPinnedTask.setPreviousTaskOrUser(user);
user.setNextTask(nonPinnedTask);
List<User> userList = mockUserList(user);
TaskAssigningSolution solution = mockSolution(Collections.singletonList(nonPinnedTask), userList);
List<ProblemFactChange<TaskAssigningSolution>> result = SolutionChangesBuilder.create().withSolution(solution).withTasks(Collections.emptyList()).withUserSystem(userSystemService).withContext(context).build();
RemoveUserProblemFactChange expected = new RemoveUserProblemFactChange(userList.get(0));
assertChangeIsTheChangeSetId(result, 0);
assertChange(result, 1, expected);
}
Aggregations