aboutsummaryrefslogtreecommitdiff
path: root/src/crypto/nodeProcessWorker.ts
blob: c5d0f2e7177eb747f21898e8587be0d945c06799 (plain)
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
import { CryptoWorkerFactory } from "./cryptoApi";

/*
 This file is part of TALER
 (C) 2016 GNUnet e.V.

 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.

 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
 TALER; see the file COPYING.  If not, see <http://www.gnu.org/licenses/>
 */


// tslint:disable:no-var-requires

import { CryptoWorker } from "./cryptoWorker";

import path = require("path");
import child_process = require("child_process");

const nodeWorkerEntry = path.join(__dirname, "nodeWorkerEntry.js");


export class NodeCryptoWorkerFactory implements CryptoWorkerFactory {
  startWorker(): CryptoWorker {
    if (typeof require === "undefined") {
      throw Error("cannot make worker, require(...) not defined");
    }
    const workerCtor = require("./nodeProcessWorker").Worker;
    const workerPath = __dirname + "/cryptoWorker.js";
    return new workerCtor(workerPath);
  }

  getConcurrency(): number {
    return 4;
  }
}

/**
 * Worker implementation that uses node subprocesses.
 */
export class Worker {
  private child: any;

  /**
   * Function to be called when we receive a message from the worker thread.
   */
  onmessage: undefined | ((m: any) => void);

  /**
   * Function to be called when we receive an error from the worker thread.
   */
  onerror: undefined | ((m: any) => void);

  private dispatchMessage(msg: any) {
    if (this.onmessage) {
      this.onmessage({ data: msg });
    } else {
      console.warn("no handler for worker event 'message' defined")
    }
  }

  private dispatchError(msg: any) {
    if (this.onerror) {
      this.onerror({ data: msg });
    } else {
      console.warn("no handler for worker event 'error' defined")
    }
  }

  constructor() {
    this.child = child_process.fork(nodeWorkerEntry);
    this.onerror = undefined;
    this.onmessage = undefined;

    this.child.on("error", (e: any) => {
      this.dispatchError(e);
    });

    this.child.on("message", (msg: any) => {
      console.log("nodeProcessWorker got child message", msg);
      this.dispatchMessage(msg);
    });
  }

  /**
   * 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;
    }
  }

  /**
   * Send a message to the worker thread.
   */
  postMessage (msg: any) {
    this.child.send(JSON.stringify({data: msg}));
  }

  /**
   * Forcibly terminate the worker thread.
   */
  terminate () {
    console.log("terminating node.js worker");
    this.child.kill("SIGINT");
  }
}