RpcService.js
2.77 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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
const RpcMessage_1 = require("./RpcMessage");
function createRpcService(port) {
const handlers = new Map();
let isListenerRegistered = false;
const callListener = (message) => __awaiter(this, void 0, void 0, function* () {
if (RpcMessage_1.isRpcCallMessage(message)) {
const handler = handlers.get(message.procedure);
try {
if (!handler) {
throw new Error(`No handler found for procedure ${message.procedure}.`);
}
const result = yield handler(message.payload);
yield port.dispatchMessage(RpcMessage_1.createRpcReturn(message.procedure, message.id, result));
}
catch (error) {
yield port.dispatchMessage(RpcMessage_1.createRpcThrow(message.procedure, message.id, {
message: error.toString(),
stack: error.stack,
}));
}
}
});
return {
isOpen: () => port.isOpen() && isListenerRegistered,
open: () => __awaiter(this, void 0, void 0, function* () {
if (!port.isOpen()) {
yield port.open();
}
if (!isListenerRegistered) {
port.addMessageListener(callListener);
isListenerRegistered = true;
}
}),
close: () => __awaiter(this, void 0, void 0, function* () {
if (isListenerRegistered) {
port.removeMessageListener(callListener);
isListenerRegistered = false;
}
if (port.isOpen()) {
yield port.close();
}
}),
addCallHandler: (procedure, handler) => {
if (handlers.has(procedure)) {
throw new Error(`Handler for '${procedure}' procedure has been already registered`);
}
handlers.set(procedure, handler);
},
removeCallHandler: (procedure) => handlers.delete(procedure),
};
}
exports.createRpcService = createRpcService;