RpcService.js 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. "use strict";
  2. var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
  3. function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
  4. return new (P || (P = Promise))(function (resolve, reject) {
  5. function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
  6. function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
  7. function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
  8. step((generator = generator.apply(thisArg, _arguments || [])).next());
  9. });
  10. };
  11. Object.defineProperty(exports, "__esModule", { value: true });
  12. const RpcMessage_1 = require("./RpcMessage");
  13. function createRpcService(port) {
  14. const handlers = new Map();
  15. let isListenerRegistered = false;
  16. const callListener = (message) => __awaiter(this, void 0, void 0, function* () {
  17. if (RpcMessage_1.isRpcCallMessage(message)) {
  18. const handler = handlers.get(message.procedure);
  19. try {
  20. if (!handler) {
  21. throw new Error(`No handler found for procedure ${message.procedure}.`);
  22. }
  23. const result = yield handler(message.payload);
  24. yield port.dispatchMessage(RpcMessage_1.createRpcReturn(message.procedure, message.id, result));
  25. }
  26. catch (error) {
  27. yield port.dispatchMessage(RpcMessage_1.createRpcThrow(message.procedure, message.id, {
  28. message: error.toString(),
  29. stack: error.stack,
  30. }));
  31. }
  32. }
  33. });
  34. return {
  35. isOpen: () => port.isOpen() && isListenerRegistered,
  36. open: () => __awaiter(this, void 0, void 0, function* () {
  37. if (!port.isOpen()) {
  38. yield port.open();
  39. }
  40. if (!isListenerRegistered) {
  41. port.addMessageListener(callListener);
  42. isListenerRegistered = true;
  43. }
  44. }),
  45. close: () => __awaiter(this, void 0, void 0, function* () {
  46. if (isListenerRegistered) {
  47. port.removeMessageListener(callListener);
  48. isListenerRegistered = false;
  49. }
  50. if (port.isOpen()) {
  51. yield port.close();
  52. }
  53. }),
  54. addCallHandler: (procedure, handler) => {
  55. if (handlers.has(procedure)) {
  56. throw new Error(`Handler for '${procedure}' procedure has been already registered`);
  57. }
  58. handlers.set(procedure, handler);
  59. },
  60. removeCallHandler: (procedure) => handlers.delete(procedure),
  61. };
  62. }
  63. exports.createRpcService = createRpcService;