-
Notifications
You must be signed in to change notification settings - Fork 5
Expand file tree
/
Copy path3-counting.js
More file actions
49 lines (42 loc) · 1.19 KB
/
3-counting.js
File metadata and controls
49 lines (42 loc) · 1.19 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
'use strict';
const fs = require('node:fs');
const threads = require('node:worker_threads');
const { Worker, isMainThread } = threads;
class CountingSemaphore {
constructor(shared, offset = 0, initial) {
this.counter = new Int32Array(shared, offset, 1);
if (typeof initial === 'number') {
this.counter[0] = initial;
}
}
enter() {
while (this.counter[0] === 0);
this.counter[0]--;
}
leave() {
this.counter[0]++;
}
}
// Usage
if (isMainThread) {
const buffer = new SharedArrayBuffer(4);
// Try to change 10 to 2 at next line to see the problem
const semaphore = new CountingSemaphore(buffer, 0, 10);
console.dir({ semaphore: semaphore.counter[0] });
for (let i = 0; i < 20; i++) {
new Worker(__filename, { workerData: buffer });
}
} else {
const { threadId, workerData } = threads;
const semaphore = new CountingSemaphore(workerData);
console.dir({ threadId, semaphore: semaphore.counter[0] });
const REPEAT_COUNT = 1000000;
const file = `file-${threadId}.dat`;
semaphore.enter();
const data = `Data from ${threadId}`.repeat(REPEAT_COUNT);
fs.writeFile(file, data, () => {
fs.unlink(file, () => {
semaphore.leave();
});
});
}