PeerTube/server/core/helpers/promise-cache.ts

40 lines
1.0 KiB
TypeScript
Raw Normal View History

export class CachePromiseFactory <A, R> {
2021-06-09 13:34:40 +02:00
private readonly running = new Map<string, Promise<R>>()
constructor (
private readonly fn: (arg: A) => Promise<R>,
private readonly keyBuilder: (arg: A) => string
) {
}
run (arg: A) {
return this.runWithContext(null, arg)
}
runWithContext (ctx: any, arg: A) {
2021-06-09 13:34:40 +02:00
const key = this.keyBuilder(arg)
if (this.running.has(key)) return this.running.get(key)
const p = this.fn.apply(ctx || this, [ arg ])
2021-06-09 13:34:40 +02:00
this.running.set(key, p)
return p.finally(() => this.running.delete(key))
}
}
export function CachePromise (options: {
keyBuilder: (...args: any[]) => string
}) {
return function (_target, _key, descriptor: PropertyDescriptor) {
const promiseCache = new CachePromiseFactory(descriptor.value, options.keyBuilder)
descriptor.value = function () {
if (arguments.length !== 1) throw new Error('Cache promise only support methods with 1 argument')
return promiseCache.runWithContext(this, arguments[0])
}
}
}