Creepso/webstreamr-github
1
0
Fork 0
Code Issues 27 Pull requests 7 Projects Releases 240 Packages Wiki Activity Actions

chore: use mutex per id when resolving streams

Browse source
This commit is contained in:
WebStreamr 2025-08-14 08:20:23 +00:00
parent 6492e6be82
commit c0797f3687
No known key found for this signature in database
1 changed files with 21 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
src/controller/StreamController.ts
View file

@ -1,9 +1,12 @@
import { Mutex } from 'async-mutex';
import { Request, Response, Router } from 'express';
import { ContentType } from 'stremio-addon-sdk';
import winston from 'winston';
import { Source } from '../source';
import { contextFromRequestAndResponse, envIsProd, ImdbId, StreamResolver } from '../utils';
const locks = new Map<string, Mutex>();
export class StreamController {
public readonly router: Router;
@ -32,13 +35,25 @@ export class StreamController {
const sources = this.sources.filter(handler => handler.countryCodes.filter(countryCode => countryCode in ctx.config).length);
const { streams, ttl } = await this.streamResolver.resolve(ctx, sources, type, ImdbId.fromString(id));
if (ttl && envIsProd()) {
res.setHeader('Cache-Control', `max-age=${ttl / 1000}, public`);
let mutex = locks.get(id);
if (!mutex) {
mutex = new Mutex();
locks.set(id, mutex);
}
res.setHeader('Content-Type', 'application/json');
res.send(JSON.stringify({ streams }));
await mutex.runExclusive(async () => {
const { streams, ttl } = await this.streamResolver.resolve(ctx, sources, type, ImdbId.fromString(id));
if (ttl && envIsProd()) {
res.setHeader('Cache-Control', `max-age=${ttl / 1000}, public`);
}
res.setHeader('Content-Type', 'application/json');
res.send(JSON.stringify({ streams }));
});
if (!mutex.isLocked()) {
locks.delete(id);
}
};
}