lambda-mock/child.js at master · sagely/lambda-mock · GitHub

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
'use strict';

var nconf = require('nconf');
nconf.argv().env();

var lambda;
process.on('message', function(msg) {
  if (msg.func) {

    // Just load the lambda once
    if (!lambda) {
      try {
        lambda = require('../' + msg.func);
      } catch (err) {
        process.send({ id: msg.id, code: 500, errorMessage: err.message, stack: err.stack });
        return;
      }
    }

    if (!lambda) {
      process.send({ id: msg.id, code: 500, errorMessage: 'Could not load lambda' });
      return;
    }

    var handler = lambda[msg.handler || 'handler'];
    var sent = false;

    var succeed = function (result) {
      if (!sent) {
        sent = true;
        process.send({ id: msg.id, code: 200, result: result });
      }
      return;
    };

    var fail = function (result) {
      if (!sent) {
        sent = true;
        process.send({ id: msg.id, code: 500, result: result });
      }
      return;
    };

    var promise = handler(msg.body, {
      getRemainingTimeInMillis: function () {
        return 0;
      },
      succeed: succeed,
      fail: fail
    }, function (err, result) {
      if (err) {
        fail(err);
      }
      else {
        succeed(result);
      }
    });
    if (promise) {
      promise.then(function (result) {
        succeed(result);
      }).catch(function (err) {
        fail(err);
      });
    }
  } else {
    process.send({ id: msg.id, code: 500, errorMessage: 'Could not load lambda', msg: msg });
    return;
  }
});