round-robin.js 873 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. /**
  2. * Deps.
  3. */
  4. var slice = require('../utils').slice;
  5. /**
  6. * Round-robin plugin.
  7. *
  8. * Provides a `send` method which will
  9. * write the `msg` to all connected peers.
  10. *
  11. * @param {Object} options
  12. * @api private
  13. */
  14. module.exports = function(options){
  15. options = options || {};
  16. var fallback = options.fallback || function(){};
  17. return function(sock){
  18. /**
  19. * Bind callback to `sock`.
  20. */
  21. fallback = fallback.bind(sock);
  22. /**
  23. * Initialize counter.
  24. */
  25. var n = 0;
  26. /**
  27. * Sends `msg` to all connected peers round-robin.
  28. */
  29. sock.send = function(){
  30. var socks = this.socks;
  31. var len = socks.length;
  32. var sock = socks[n++ % len];
  33. var msg = slice(arguments);
  34. if (sock && sock.writable) {
  35. sock.write(this.pack(msg));
  36. } else {
  37. fallback(msg);
  38. }
  39. };
  40. };
  41. };