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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
This file is part of GNU Taler
(C) 2023 Taler Systems S.A.
GNU Taler is free software; you can redistribute it and/or modify it under the
terms of the GNU General Public License as published by the Free Software
Foundation; either version 3, or (at your option) any later version.
GNU Taler is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE. See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with
GNU Taler; see the file COPYING. If not, see <http://www.gnu.org/licenses/>
*/
import { AbsoluteTime } from "@gnu-taler/taler-util";
import test from "ava";
import { CryptoDispatcher, CryptoWorkerFactory } from "./crypto-dispatcher.js";
import {
CryptoWorker,
CryptoWorkerResponseMessage,
} from "./cryptoWorkerInterface.js";
export class MyCryptoWorker implements CryptoWorker {
/**
* Function to be called when we receive a message from the worker thread.
*/
onmessage: undefined | ((m: any) => void) = undefined;
/**
* Function to be called when we receive an error from the worker thread.
*/
onerror: undefined | ((m: any) => void) = undefined;
/**
* Add an event listener for either an "error" or "message" event.
*/
addEventListener(event: "message" | "error", fn: (x: any) => void): void {
switch (event) {
case "message":
this.onmessage = fn;
break;
case "error":
this.onerror = fn;
break;
}
}
private dispatchMessage(msg: any): void {
if (this.onmessage) {
this.onmessage(msg);
}
}
/**
* Send a message to the worker thread.
*/
postMessage(msg: any): void {
const handleRequest = async () => {
let responseMsg: CryptoWorkerResponseMessage;
if (msg.operation === "testSuccess") {
responseMsg = {
id: msg.id,
type: "success",
result: {
testResult: 42,
},
};
} else if (msg.operation === "testError") {
responseMsg = {
id: msg.id,
type: "error",
error: {
code: 42,
when: AbsoluteTime.now(),
hint: "bla",
},
};
} else if (msg.operation === "testTimeout") {
// Don't respond
return;
}
try {
setTimeout(() => this.dispatchMessage(responseMsg), 0);
} catch (e) {
console.error("got error during dispatch", e);
}
};
handleRequest().catch((e) => {
console.error("Error while handling crypto request:", e);
});
}
/**
* Forcibly terminate the worker thread.
*/
terminate(): void {
// This is a no-op.
}
}
export class MyCryptoWorkerFactory implements CryptoWorkerFactory {
startWorker(): CryptoWorker {
return new MyCryptoWorker();
}
getConcurrency(): number {
return 1;
}
}
test("continues after error", async (t) => {
const cryptoDisp = new CryptoDispatcher(new MyCryptoWorkerFactory());
const resp1 = await cryptoDisp.doRpc("testSuccess", 0, {});
t.assert((resp1 as any).testResult === 42);
const exc = await t.throwsAsync(async () => {
const resp2 = await cryptoDisp.doRpc("testError", 0, {});
});
// Check that it still works after one error.
const resp2 = await cryptoDisp.doRpc("testSuccess", 0, {});
t.assert((resp2 as any).testResult === 42);
// Check that it still works after timeout.
const resp3 = await cryptoDisp.doRpc("testSuccess", 0, {});
t.assert((resp3 as any).testResult === 42);
});
|
