wangxj3 commented on a change in pull request #6722:
URL: https://github.com/apache/dolphinscheduler/pull/6722#discussion_r756686265
##########
File path:
dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessService.java
##########
@@ -2568,6 +2581,174 @@ private void addAuthorizedResources(List<Resource>
ownResources, int userId) {
return processTaskMap;
}
+ /**
+ * @param taskId task id
+ * @param taskName
+ * @param groupId
+ * @param processId
+ * @param priority
+ * @return
+ */
+ public boolean acquireTaskGroup(int taskId,
+ String taskName, int groupId,
+ int processId, int priority) {
+ TaskGroup taskGroup = taskGroupMapper.selectById(groupId);
+ if (taskGroup == null) {
+ return true;
+ }
+ // if task group is not applicable
+ if (taskGroup.getStatus() == Flag.NO.getCode()) {
+ return true;
+ }
+ TaskGroupQueue taskGroupQueue =
this.taskGroupQueueMapper.queryByTaskId(taskId);
+ if (taskGroupQueue == null) {
+ taskGroupQueue = insertIntoTaskGroupQueue(taskId, taskName,
groupId, processId, priority, TaskGroupQueueStatus.WAIT_QUEUE);
+ } else {
+ if (taskGroupQueue.getStatus() ==
TaskGroupQueueStatus.ACQUIRE_SUCCESS) {
+ return true;
+ }
+ taskGroupQueue.setInQueue(Flag.NO.getCode());
+ taskGroupQueue.setStatus(TaskGroupQueueStatus.WAIT_QUEUE);
+ this.taskGroupQueueMapper.updateById(taskGroupQueue);
+ }
+ //check priority
+ List<TaskGroupQueue> highPriorityTasks =
taskGroupQueueMapper.queryHighPriorityTasks(groupId, priority,
TaskGroupQueueStatus.WAIT_QUEUE.getCode());
+ if (CollectionUtils.isNotEmpty(highPriorityTasks)) {
+ this.taskGroupQueueMapper.updateInQueue(Flag.NO.getCode(),
taskGroupQueue.getId());
+ return false;
+ }
+ //try to get taskGroup
+ int count = taskGroupMapper.selectAvailableCountById(groupId);
+ if (count == 1 && robTaskGroupResouce(taskGroupQueue)) {
+ return true;
+ }
+ this.taskGroupQueueMapper.updateInQueue(Flag.NO.getCode(),
taskGroupQueue.getId());
+ return false;
+ }
+
+ public boolean robTaskGroupResouce(TaskGroupQueue taskGroupQueue) {
Review comment:
Get the resources of the task group
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
