]>
Commit | Line | Data |
---|---|---|
c53853ca C |
1 | import { join } from 'path' |
2 | import Piscina from 'piscina' | |
daa0226b | 3 | import { processImage } from '@server/helpers/image-utils' |
c53853ca | 4 | import { WORKER_THREADS } from '@server/initializers/constants' |
405c83f9 | 5 | import { httpBroadcast } from './workers/http-broadcast' |
c53853ca C |
6 | import { downloadImage } from './workers/image-downloader' |
7 | ||
daa0226b | 8 | let downloadImageWorker: Piscina |
c53853ca C |
9 | |
10 | function downloadImageFromWorker (options: Parameters<typeof downloadImage>[0]): Promise<ReturnType<typeof downloadImage>> { | |
daa0226b C |
11 | if (!downloadImageWorker) { |
12 | downloadImageWorker = new Piscina({ | |
13 | filename: join(__dirname, 'workers', 'image-downloader.js'), | |
14 | concurrentTasksPerWorker: WORKER_THREADS.DOWNLOAD_IMAGE.CONCURRENCY, | |
15 | maxThreads: WORKER_THREADS.DOWNLOAD_IMAGE.MAX_THREADS | |
16 | }) | |
17 | } | |
18 | ||
19 | return downloadImageWorker.run(options) | |
c53853ca C |
20 | } |
21 | ||
3a54605d C |
22 | // --------------------------------------------------------------------------- |
23 | ||
daa0226b | 24 | let processImageWorker: Piscina |
3a54605d C |
25 | |
26 | function processImageFromWorker (options: Parameters<typeof processImage>[0]): Promise<ReturnType<typeof processImage>> { | |
daa0226b C |
27 | if (!processImageWorker) { |
28 | processImageWorker = new Piscina({ | |
29 | filename: join(__dirname, 'workers', 'image-processor.js'), | |
30 | concurrentTasksPerWorker: WORKER_THREADS.PROCESS_IMAGE.CONCURRENCY, | |
31 | maxThreads: WORKER_THREADS.PROCESS_IMAGE.MAX_THREADS | |
32 | }) | |
33 | } | |
34 | ||
3a54605d C |
35 | return processImageWorker.run(options) |
36 | } | |
37 | ||
405c83f9 C |
38 | // --------------------------------------------------------------------------- |
39 | ||
40 | let parallelHTTPBroadcastWorker: Piscina | |
41 | ||
42 | function parallelHTTPBroadcastFromWorker (options: Parameters<typeof httpBroadcast>[0]): Promise<ReturnType<typeof httpBroadcast>> { | |
43 | if (!parallelHTTPBroadcastWorker) { | |
44 | parallelHTTPBroadcastWorker = new Piscina({ | |
45 | filename: join(__dirname, 'workers', 'http-broadcast.js'), | |
46 | concurrentTasksPerWorker: WORKER_THREADS.PARALLEL_HTTP_BROADCAST.CONCURRENCY, | |
47 | maxThreads: WORKER_THREADS.PARALLEL_HTTP_BROADCAST.MAX_THREADS | |
48 | }) | |
49 | } | |
50 | ||
51 | return parallelHTTPBroadcastWorker.run(options) | |
52 | } | |
53 | ||
54 | // --------------------------------------------------------------------------- | |
55 | ||
56 | let sequentialHTTPBroadcastWorker: Piscina | |
57 | ||
58 | function sequentialHTTPBroadcastFromWorker (options: Parameters<typeof httpBroadcast>[0]): Promise<ReturnType<typeof httpBroadcast>> { | |
59 | if (!sequentialHTTPBroadcastWorker) { | |
60 | sequentialHTTPBroadcastWorker = new Piscina({ | |
61 | filename: join(__dirname, 'workers', 'http-broadcast.js'), | |
62 | concurrentTasksPerWorker: WORKER_THREADS.SEQUENTIAL_HTTP_BROADCAST.CONCURRENCY, | |
63 | maxThreads: WORKER_THREADS.SEQUENTIAL_HTTP_BROADCAST.MAX_THREADS | |
64 | }) | |
65 | } | |
66 | ||
67 | return sequentialHTTPBroadcastWorker.run(options) | |
68 | } | |
69 | ||
c53853ca | 70 | export { |
3a54605d | 71 | downloadImageFromWorker, |
405c83f9 C |
72 | processImageFromWorker, |
73 | parallelHTTPBroadcastFromWorker, | |
74 | sequentialHTTPBroadcastFromWorker | |
c53853ca | 75 | } |