const WorkQueue = require('wrappitmq').WorkQueue;

// Set-up.
const queue = new WorkQueue({
  // All these options are optional.
  prefetch: 1, // How many messages to work on asynchronously (Default: 1)
  queue: 'workqueue', // Name of the queue to use. (Default: workqueue)
  url: 'amqp://localhost:5672' // Can also be specified on connect()
});

// URL here is optional, can also be set as above.
await queue.connect('amqp://localhost:5672');

queue.on('error', (err) => {
  // Something went wrong and you should log it.
});
queue.on('close', (err) => {
  // The connection was closed and the queue is no longer usable.
  // An err is given if the connection was closed due to an error.
});

// Enqueue a task.
await queue.enqueue({ do: 'it' });

// Consume tasks.
const cancel = await queue.consume(async (task) => {
  await work(task);
  // Will be acknowledged when work() is done.
});

// Cancel consumer to stop receiving tasks.
await cancel();

// Close connection.
await queue.close();

const PubSub = require('wrappitmq').PubSub;

// Set-up.
const pubsub = new PubSub({
  // All these options are optional.
  exchange: 'pubsub', // Name of the queue to use. (Default: pubsub)
  url: 'amqp://localhost:5672' // Can also be specified on connect()
});

// URL here is optional, can also be set as above.
await pubsub.connect('amqp://localhost:5672');

pubsub.on('error', (err) => {
  // Something went wrong and you should log it.
});
pubsub.on('close', (err) => {
  // The connection was closed and the queue is no longer usable.
  // An err is given if the connection was closed due to an error.
});

// Publish a message.
await pubsub.publish('login', { userId: 1 });

// Subscribe to a topic.
const unsub = await pubsub.subscribe('login', async (message) => {
  await loggedIn(message.userId);
});

// Cancel subscription to stop receiving messages.
await unsub();

// Close connection.
await pubsub.close();