- const chosenWorkerKey = this.currentWorkerId
- if (this.isDynamicPool && !this.workersTaskRunTime.has(chosenWorkerKey)) {
- this.initWorkerTaskRunTime(chosenWorkerKey)
- }
- const workerTaskRunTime =
- this.workersTaskRunTime.get(chosenWorkerKey)?.runTime ?? 0
- const workerTaskWeight =
- this.workersTaskRunTime.get(chosenWorkerKey)?.weight ??
- this.defaultWorkerWeight
- if (workerTaskRunTime < workerTaskWeight) {
- this.setWorkerTaskRunTime(
- chosenWorkerKey,
- workerTaskWeight,
- workerTaskRunTime +
- (this.getWorkerVirtualTaskRunTime(chosenWorkerKey) ?? 0)
- )
+ const chosenWorkerNodeKey = this.currentWorkerNodeId
+ const workerVirtualTaskRunTime = this.workerVirtualTaskRunTime
+ const workerWeight =
+ this.opts.weights?.[chosenWorkerNodeKey] ?? this.defaultWorkerWeight
+ if (workerVirtualTaskRunTime < workerWeight) {
+ this.workerVirtualTaskRunTime =
+ workerVirtualTaskRunTime +
+ this.getWorkerTaskRunTime(chosenWorkerNodeKey)