third_codes/PowerJob
1
0
Fork 1
mirror of https://github.com/PowerJob/PowerJob.git synced 2025-07-17 00:00:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: NPE when not select job's dispatchStrategy #988

Browse Source
This commit is contained in:
tjq 2024-09-10 00:05:28 +08:00
parent 170319444c
commit cb46d7d196
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
powerjob-server/powerjob-server-remote/src/main/java/tech/powerjob/server/remote/worker/selector/TaskTrackerSelectorService.java
View File

@ -3,12 +3,14 @@ package tech.powerjob.server.remote.worker.selector;
import com.google.common.collect.Maps;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import tech.powerjob.common.enums.DispatchStrategy;
import tech.powerjob.server.common.module.WorkerInfo;
import tech.powerjob.server.persistence.remote.model.InstanceInfoDO;
import tech.powerjob.server.persistence.remote.model.JobInfoDO;
import java.util.List;
import java.util.Map;
import java.util.Optional;
/**
* TaskTrackerSelectorService
@ -27,7 +29,8 @@ public class TaskTrackerSelectorService {
}
public WorkerInfo select(JobInfoDO jobInfoDO, InstanceInfoDO instanceInfoDO, List<WorkerInfo> availableWorkers) {
TaskTrackerSelector taskTrackerSelector = taskTrackerSelectorMap.get(jobInfoDO.getDispatchStrategy());
Integer strategy = Optional.ofNullable(jobInfoDO.getDispatchStrategy()).orElse(DispatchStrategy.HEALTH_FIRST.getV());
TaskTrackerSelector taskTrackerSelector = taskTrackerSelectorMap.get(strategy);
return taskTrackerSelector.select(jobInfoDO, instanceInfoDO, availableWorkers);
}
}