main.js 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. // Copyright (c) 2014, Ben Noordhuis <info@bnoordhuis.nl>
  2. //
  3. // Permission to use, copy, modify, and/or distribute this software for any
  4. // purpose with or without fee is hereby granted, provided that the above
  5. // copyright notice and this permission notice appear in all copies.
  6. //
  7. // THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  8. // WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  9. // MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  10. // ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  11. // WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  12. // ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  13. // OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  14. try {
  15. var addon = require('../build/Release/addon');
  16. } catch (e) {
  17. if (e.code !== 'MODULE_NOT_FOUND') throw e;
  18. var addon = require('../build/Debug/addon');
  19. }
  20. var kForkFlag = addon.kForkFlag;
  21. var kSignalFlag = addon.kSignalFlag;
  22. var flags = kSignalFlag;
  23. var options = (process.env.NODE_HEAPDUMP_OPTIONS || '').split(/\s*,\s*/);
  24. for (var i = 0, n = options.length; i < n; i += 1) {
  25. var option = options[i];
  26. if (option === '') continue;
  27. else if (option === 'fork') flags |= kForkFlag;
  28. else if (option === 'signal') flags |= kSignalFlag;
  29. else if (option === 'nofork') flags &= ~kForkFlag;
  30. else if (option === 'nosignal') flags &= ~kSignalFlag;
  31. else console.error('node-heapdump: unrecognized option:', option);
  32. }
  33. addon.configure(flags);
  34. exports.writeSnapshot = function() {
  35. return addon.writeSnapshot.apply(null, arguments);
  36. };