Limit visible AllDebrid start countdowns

This commit is contained in:
Sucukdeluxe 2026-03-08 21:47:11 +01:00
parent fdb575b0c1
commit 8b941a2777
2 changed files with 139 additions and 0 deletions

View File

@ -5541,6 +5541,35 @@ export class DownloadManager extends EventEmitter {
return provider;
}
private countVisiblePacedStarts(paceKey: string, now: number, excludeItemId?: string): number {
let count = 0;
for (const [itemId, reservedAt] of this.pacedStartReservationByItem.entries()) {
if (excludeItemId && itemId === excludeItemId) {
continue;
}
if (reservedAt <= now) {
continue;
}
const item = this.session.items[itemId];
if (!item) {
continue;
}
if (this.getPacedStartKeyForItem(item) !== paceKey) {
continue;
}
if (item.status !== "queued" && item.status !== "reconnect_wait") {
continue;
}
count += 1;
}
return count;
}
private getVisiblePacedStartBudget(): number {
const maxParallel = Math.max(1, Number(this.settings.maxParallel) || 1);
return Math.max(0, maxParallel - this.activeTasks.size);
}
private delayPacedStartForItem(item: DownloadItem, now: number): boolean {
const paceKey = this.getPacedStartKeyForItem(item);
if (!paceKey) {
@ -5569,6 +5598,17 @@ export class DownloadManager extends EventEmitter {
return false;
}
const visibleBudget = this.getVisiblePacedStartBudget();
const visibleReservations = this.countVisiblePacedStarts(paceKey, now, item.id);
if (visibleBudget <= 0 || visibleReservations >= visibleBudget) {
this.pacedStartReservationByItem.delete(item.id);
if ((item.fullStatus || "").startsWith("AllDebrid Start in ")) {
item.fullStatus = "Wartet";
item.updatedAt = now;
}
return true;
}
const scheduledAt = Math.max(existingReadyAt, nextAllowedAt);
if (scheduledAt <= now) {
this.pacedStartReservationByItem.delete(item.id);

View File

@ -4838,6 +4838,105 @@ describe("download manager", () => {
}
}, 20000);
it("shows AllDebrid countdowns only for the next visible start wave", async () => {
const root = fs.mkdtempSync(path.join(os.tmpdir(), "rd-dm-"));
tempDirs.push(root);
const chunk = Buffer.alloc(256 * 1024, 9);
const totalLinks = 8;
const server = http.createServer((req, res) => {
const route = req.url || "";
if (!/^\/ad-web-\d+$/.test(route)) {
res.statusCode = 404;
res.end("not-found");
return;
}
let sent = 0;
const totalChunks = 10;
res.statusCode = 200;
res.setHeader("Accept-Ranges", "bytes");
res.setHeader("Content-Length", String(chunk.length * totalChunks));
const timer = setInterval(() => {
if (sent >= totalChunks) {
clearInterval(timer);
res.end();
return;
}
res.write(chunk);
sent += 1;
}, 700);
res.on("close", () => clearInterval(timer));
});
server.listen(0, "127.0.0.1");
await once(server, "listening");
const address = server.address();
if (!address || typeof address === "string") {
throw new Error("server address unavailable");
}
const links = Array.from({ length: totalLinks }, (_, index) => `https://rapidgator.net/file/web-${index + 1}/sample.part${index + 1}.rar.html`);
try {
const manager = new DownloadManager(
{
...defaultSettings(),
allDebridToken: "ad-token",
allDebridUseWebLogin: true,
providerOrder: [],
providerPrimary: "alldebrid",
providerSecondary: "none",
providerTertiary: "none",
outputDir: path.join(root, "downloads"),
extractDir: path.join(root, "extract"),
autoExtract: false,
autoReconnect: false,
enableIntegrityCheck: false,
maxParallel: 5
},
emptySession(),
createStoragePaths(path.join(root, "state")),
{
allDebridWebUnrestrict: async (link) => {
const match = link.match(/web-(\d+)/);
const slot = Number(match?.[1] || 1);
return {
fileName: `ad-web-${slot}.bin`,
directUrl: `http://127.0.0.1:${address.port}/ad-web-${slot}`,
fileSize: chunk.length * 10,
retriesUsed: 0
};
}
}
);
manager.addPackages([{ name: "ad-web-visibility", links }]);
await manager.start();
await waitFor(() => {
const items = Object.values(manager.getSnapshot().session.items);
return items.some((item) => item.status === "downloading");
}, 10000);
await new Promise((resolve) => setTimeout(resolve, 500));
const items = Object.values(manager.getSnapshot().session.items);
const activeCount = items.filter((item) => item.status === "downloading" || item.status === "validating").length;
const countdownItems = items.filter((item) => /^AllDebrid Start in \d+s$/.test(item.fullStatus || ""));
const plainQueuedItems = items.filter((item) => (item.status === "queued" || item.status === "reconnect_wait") && item.fullStatus === "Wartet");
expect(countdownItems.length).toBeLessThanOrEqual(Math.max(0, 5 - activeCount));
expect(plainQueuedItems.length).toBeGreaterThan(0);
manager.stop();
await waitFor(() => !manager.getSnapshot().session.running, 15000);
} finally {
server.close();
await once(server, "close");
}
}, 20000);
it("staggeres AllDebrid starts by 2.5 seconds per active download", async () => {
const root = fs.mkdtempSync(path.join(os.tmpdir(), "rd-dm-"));
tempDirs.push(root);