- /** {@inheritDoc} */
- public remove (workerKey: number): boolean {
- const workerDeleted = this.workerLastVirtualTaskTimestamp.delete(workerKey)
- for (const [key, value] of this.workerLastVirtualTaskTimestamp.entries()) {
- if (key > workerKey) {
- this.workerLastVirtualTaskTimestamp.set(key - 1, value)
+ /** @inheritDoc */
+ public choose (): number | undefined {
+ this.nextWorkerNodeKey = this.fairShareNextWorkerNodeKey()
+ return this.nextWorkerNodeKey
+ }
+
+ /** @inheritDoc */
+ public remove (workerNodeKey: number): boolean {
+ this.workersVirtualTaskEndTimestamp.splice(workerNodeKey, 1)
+ return true
+ }
+
+ private fairShareNextWorkerNodeKey (): number | undefined {
+ let minWorkerVirtualTaskEndTimestamp = Infinity
+ let chosenWorkerNodeKey: number | undefined
+ for (const [workerNodeKey] of this.pool.workerNodes.entries()) {
+ if (!this.isWorkerNodeEligible(workerNodeKey)) {
+ continue
+ }
+ if (this.workersVirtualTaskEndTimestamp[workerNodeKey] == null) {
+ this.computeWorkerVirtualTaskEndTimestamp(workerNodeKey)
+ }
+ const workerVirtualTaskEndTimestamp =
+ this.workersVirtualTaskEndTimestamp[workerNodeKey]
+ if (workerVirtualTaskEndTimestamp < minWorkerVirtualTaskEndTimestamp) {
+ minWorkerVirtualTaskEndTimestamp = workerVirtualTaskEndTimestamp
+ chosenWorkerNodeKey = workerNodeKey