rpc_client.js 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. #!/usr/bin/env node
  2. const amqp = require('amqplib');
  3. const { basename } = require('path');
  4. const { v4: uuid } = require('uuid');
  5. const queue = 'rpc_queue';
  6. const n = parseInt(process.argv[2], 10);
  7. if (isNaN(n)) {
  8. console.warn('Usage: %s number', basename(process.argv[1]));
  9. process.exit(1);
  10. }
  11. (async () => {
  12. let connection;
  13. try {
  14. connection = await amqp.connect('amqp://localhost');
  15. const channel = await connection.createChannel();
  16. const correlationId = uuid();
  17. const requestFib = new Promise(async (resolve) => {
  18. const { queue: replyTo } = await channel.assertQueue('', { exclusive: true });
  19. await channel.consume(replyTo, (message) => {
  20. if (!message) console.warn(' [x] Consumer cancelled');
  21. else if (message.properties.correlationId === correlationId) {
  22. resolve(message.content.toString());
  23. }
  24. }, { noAck: true });
  25. await channel.assertQueue(queue, { durable: false });
  26. console.log(' [x] Requesting fib(%d)', n);
  27. channel.sendToQueue(queue, Buffer.from(n.toString()), {
  28. correlationId,
  29. replyTo,
  30. });
  31. });
  32. const fibN = await requestFib;
  33. console.log(' [.] Got %d', fibN);
  34. }
  35. catch (err) {
  36. console.warn(err);
  37. }
  38. finally {
  39. if (connection) await connection.close();
  40. };
  41. })();