UNPKG

@jonaskello-forks/amqp-client

Version:

AMQP 0-9-1 client, both for browsers (WebSocket) and node (TCP Socket)

845 lines (798 loc) 32.9 kB
import { AMQPError } from './amqp-error.js' import { AMQPView } from './amqp-view.js' import { AMQPQueue } from './amqp-queue.js' import { AMQPConsumer } from './amqp-consumer.js' import { AMQPMessage } from './amqp-message.js' import { AMQPBaseClient } from './amqp-base-client.js' import { AMQPProperties } from './amqp-properties.js' /** * Represents an AMQP Channel. Almost all actions in AMQP are performed on a Channel. */ export class AMQPChannel { readonly connection: AMQPBaseClient readonly id: number readonly consumers = new Map<string, AMQPConsumer>() readonly promises: [(arg0: any) => void, (err?: Error) => void][] = [] private readonly unconfirmedPublishes: [number, (confirmId: number) => void, (err?: Error) => void][] = [] private closed = false /** Used for string -> arraybuffer when publishing */ private static textEncoder = new TextEncoder() /** Frame buffer, reuse when publishes to avoid repated allocations */ private readonly buffer: AMQPView confirmId = 0 delivery?: AMQPMessage getMessage?: AMQPMessage returned?: AMQPMessage /** * @param connection - The connection this channel belongs to * @param id - ID of the channel */ constructor(connection: AMQPBaseClient, id: number) { this.connection = connection this.id = id this.buffer = new AMQPView(new ArrayBuffer(connection.frameMax)) } /** * Declare a queue and return an AMQPQueue instance. */ queue(name = "", {passive = false, durable = name !== "", autoDelete = name === "", exclusive = name === ""} = {}, args = {}): Promise<AMQPQueue> { return new Promise((resolve, reject) => { this.queueDeclare(name, {passive, durable, autoDelete, exclusive}, args) .then(({name}) => resolve(new AMQPQueue(this, name))) .catch(reject) }) } /** * Alias for basicQos * @param prefetchCount - max inflight messages */ prefetch(prefetchCount: number) { return this.basicQos(prefetchCount) } /** * Default handler for Returned messages * @param message returned from server */ onReturn(message: AMQPMessage) { console.error("Message returned from server", message) } /** * Close the channel gracefully * @param [reason] might be logged by the server */ close(reason = "", code = 200) { if (this.closed) return this.rejectClosed() this.closed = true let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel frame.setUint32(j, 0); j += 4 // frameSize frame.setUint16(j, 20); j += 2 // class: channel frame.setUint16(j, 40); j += 2 // method: close frame.setUint16(j, code); j += 2 // reply code j += frame.setShortString(j, reason) // reply reason frame.setUint16(j, 0); j += 2 // failing-class-id frame.setUint16(j, 0); j += 2 // failing-method-id frame.setUint8(j, 206); j += 1 // frame end byte frame.setUint32(3, j - 8) // update frameSize return this.sendRpc(frame, j) } /** * Synchronously receive a message from a queue * @param queue - name of the queue to poll * @param param * @param [param.noAck=true] - if message is removed from the server upon delivery, or have to be acknowledged * @return - returns null if the queue is empty otherwise a single message */ basicGet(queue: string, { noAck = true } = {}): Promise<AMQPMessage|null> { if (this.closed) return this.rejectClosed() let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel: 1 frame.setUint32(j, 11); j += 4 // frameSize frame.setUint16(j, 60); j += 2 // class: basic frame.setUint16(j, 70); j += 2 // method: get frame.setUint16(j, 0); j += 2 // reserved1 j += frame.setShortString(j, queue) // queue frame.setUint8(j, noAck ? 1 : 0); j += 1 // noAck frame.setUint8(j, 206); j += 1 // frame end byte frame.setUint32(3, j - 8) // update frameSize return this.sendRpc(frame, j) } /** * Consume from a queue. Messages will be delivered asynchronously. * @param queue - name of the queue to poll * @param param * @param [param.tag=""] - tag of the consumer, will be server generated if left empty * @param [param.noAck=true] - if messages are removed from the server upon delivery, or have to be acknowledged * @param [param.exclusive=false] - if this can be the only consumer of the queue, will return an Error if there are other consumers to the queue already * @param [param.args={}] - custom arguments * @param {function(AMQPMessage) : void} callback - will be called for each message delivered to this consumer */ basicConsume(queue: string, {tag = "", noAck = true, exclusive = false, args = {}} = {}, callback: (msg: AMQPMessage) => void): Promise<AMQPConsumer> { if (this.closed) return this.rejectClosed() let j = 0 const noWait = false const noLocal = false const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel: 1 frame.setUint32(j, 0); j += 4 // frameSize frame.setUint16(j, 60); j += 2 // class: basic frame.setUint16(j, 20); j += 2 // method: consume frame.setUint16(j, 0); j += 2 // reserved1 j += frame.setShortString(j, queue) // queue j += frame.setShortString(j, tag) // tag let bits = 0 if (noLocal) bits = bits | (1 << 0) if (noAck) bits = bits | (1 << 1) if (exclusive) bits = bits | (1 << 2) if (noWait) bits = bits | (1 << 3) frame.setUint8(j, bits); j += 1 // noLocal/noAck/exclusive/noWait j += frame.setTable(j, args) // arguments table frame.setUint8(j, 206); j += 1 // frame end byte frame.setUint32(3, j - 8) // update frameSize return new Promise((resolve, reject) => { this.sendRpc(frame, j).then((consumerTag) => { const consumer = new AMQPConsumer(this, consumerTag, callback) this.consumers.set(consumerTag, consumer) resolve(consumer) }).catch(reject) }) } /** * Cancel/stop a consumer * @param tag - consumer tag */ basicCancel(tag: string): Promise<AMQPChannel> { if (this.closed) return this.rejectClosed() const noWait = false let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel: 1 frame.setUint32(j, 0); j += 4 // frameSize frame.setUint16(j, 60); j += 2 // class: basic frame.setUint16(j, 30); j += 2 // method: cancel j += frame.setShortString(j, tag) // tag frame.setUint8(j, noWait ? 1 : 0); j += 1 // noWait frame.setUint8(j, 206); j += 1 // frame end byte frame.setUint32(3, j - 8) // update frameSize return new Promise((resolve, reject) => { this.sendRpc(frame, j).then((consumerTag) => { const consumer = this.consumers.get(consumerTag) if (consumer) { consumer.setClosed() this.consumers.delete(consumerTag) } resolve(this) }).catch(reject) }) } /** * Acknowledge a delivered message * @param deliveryTag - tag of the message * @param [multiple=false] - batch confirm all messages up to this delivery tag */ basicAck(deliveryTag: number, multiple = false): Promise<void> { if (this.closed) return this.rejectClosed() let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel frame.setUint32(j, 13); j += 4 // frameSize frame.setUint16(j, 60); j += 2 // class: basic frame.setUint16(j, 80); j += 2 // method: ack frame.setUint64(j, deliveryTag); j += 8 frame.setUint8(j, multiple ? 1 : 0); j += 1 frame.setUint8(j, 206); j += 1 // frame end byte return this.connection.send(new Uint8Array(frame.buffer, 0, 21)) } /** * Acknowledge a delivered message * @param deliveryTag - tag of the message * @param [requeue=false] - if the message should be requeued or removed * @param [multiple=false] - batch confirm all messages up to this delivery tag */ basicNack(deliveryTag: number, requeue = false, multiple = false): Promise<void> { if (this.closed) return this.rejectClosed() let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel frame.setUint32(j, 13); j += 4 // frameSize frame.setUint16(j, 60); j += 2 // class: basic frame.setUint16(j, 120); j += 2 // method: nack frame.setUint64(j, deliveryTag); j += 8 let bits = 0 if (multiple) bits = bits | (1 << 0) if (requeue) bits = bits | (1 << 1) frame.setUint8(j, bits); j += 1 frame.setUint8(j, 206); j += 1 // frame end byte return this.connection.send(new Uint8Array(frame.buffer, 0, 21)) } /** * Acknowledge a delivered message * @param deliveryTag - tag of the message * @param [requeue=false] - if the message should be requeued or removed */ basicReject(deliveryTag: number, requeue = false) { if (this.closed) return this.rejectClosed() let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel frame.setUint32(j, 13); j += 4 // frameSize frame.setUint16(j, 60); j += 2 // class: basic frame.setUint16(j, 90); j += 2 // method: reject frame.setUint64(j, deliveryTag); j += 8 frame.setUint8(j, requeue ? 1 : 0); j += 1 frame.setUint8(j, 206); j += 1 // frame end byte return this.connection.send(new Uint8Array(frame.buffer, 0, 21)) } /** * Tell the server to redeliver all unacknowledged messages again, or reject and requeue them. * @param [requeue=false] - if the message should be requeued or redeliviered to this channel */ basicRecover(requeue = false) { if (this.closed) return this.rejectClosed() let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel frame.setUint32(j, 5); j += 4 // frameSize frame.setUint16(j, 60); j += 2 // class: basic frame.setUint16(j, 110); j += 2 // method: recover frame.setUint8(j, requeue ? 1 : 0); j += 1 frame.setUint8(j, 206); j += 1 // frame end byte return this.sendRpc(frame, j) } /** * Publish a message * @param exchange - the exchange to publish to, the exchange must exists * @param routingKey - routing key * @param data - the data to be published, can be a string or an uint8array * @param [mandatory] - if the message should be returned if there's no queue to be delivered to * @param [immediate] - if the message should be returned if it can't be delivered to a consumer immediately (not supported in RabbitMQ) * @return - fulfilled when the message is enqueue on the socket, or if publish confirm is enabled when the message is confirmed by the server */ async basicPublish(exchange: string, routingKey: string, data: string|Uint8Array|ArrayBuffer|Buffer|null, properties: AMQPProperties = {}, mandatory = false, immediate = false): Promise<number> { if (this.closed) return this.rejectClosed() if (this.connection.blocked) return Promise.reject(new AMQPError(`Connection blocked by server: ${this.connection.blocked}`, this.connection)) let body : Uint8Array if (typeof Buffer !== "undefined" && data instanceof Buffer) { body = data } else if (data instanceof Uint8Array) { body = data } else if (data instanceof ArrayBuffer) { body = new Uint8Array(data) } else if (data === null) { body = new Uint8Array(0) } else if (typeof data === "string") { body = AMQPChannel.textEncoder.encode(data) } else { throw new TypeError(`Invalid type ${typeof data} for parameter data`) } let j = 0 const buffer = this.buffer buffer.setUint8(j, 1); j += 1 // type: method buffer.setUint16(j, this.id); j += 2 // channel j += 4 // frame size, update later buffer.setUint16(j, 60); j += 2 // class: basic buffer.setUint16(j, 40); j += 2 // method: publish buffer.setUint16(j, 0); j += 2 // reserved1 j += buffer.setShortString(j, exchange) // exchange j += buffer.setShortString(j, routingKey) // routing key let bits = 0 if (mandatory) bits = bits | (1 << 0) if (immediate) bits = bits | (1 << 1) buffer.setUint8(j, bits); j += 1 // mandatory/immediate buffer.setUint8(j, 206); j += 1 // frame end byte buffer.setUint32(3, j - 8) // update frameSize const headerStart = j buffer.setUint8(j, 2); j += 1 // type: header buffer.setUint16(j, this.id); j += 2 // channel j += 4 // frame size, update later buffer.setUint16(j, 60); j += 2 // class: basic buffer.setUint16(j, 0); j += 2 // weight buffer.setUint32(j, 0); j += 4 // bodysize (upper 32 of 64 bits) buffer.setUint32(j, body.byteLength); j += 4 // bodysize j += buffer.setProperties(j, properties); // properties buffer.setUint8(j, 206); j += 1 // frame end byte buffer.setUint32(headerStart + 3, j - headerStart - 8) // update frameSize // Send current frames if there's no body to send if (body.byteLength === 0) { await this.connection.send(new Uint8Array(buffer.buffer, 0, j)) } else if (j >= buffer.byteLength - 8) { // Send current frames if a body frame can't fit in the rest of the frame buffer await this.connection.send(new Uint8Array(buffer.buffer, 0, j)) j = 0 } // split body into multiple frames if body > frameMax for (let bodyPos = 0; bodyPos < body.byteLength;) { const frameSize = Math.min(body.byteLength - bodyPos, buffer.byteLength - 8 - j) // frame overhead is 8 bytes const dataSlice = body.subarray(bodyPos, bodyPos + frameSize) buffer.setUint8(j, 3); j += 1 // type: body buffer.setUint16(j, this.id); j += 2 // channel buffer.setUint32(j, frameSize); j += 4 // frameSize const bodyView = new Uint8Array(buffer.buffer, j, frameSize) bodyView.set(dataSlice); j += frameSize // body content buffer.setUint8(j, 206); j += 1 // frame end byte await this.connection.send(new Uint8Array(buffer.buffer, 0, j)) bodyPos += frameSize j = 0 } // if publish confirm is enabled, put a promise on a queue if the sends were ok // the promise on the queue will be fullfilled by the read loop when an ack/nack // comes from the server if (this.confirmId) { return new Promise((resolve, reject) => this.unconfirmedPublishes.push([this.confirmId++, resolve, reject])) } else { return Promise.resolve(0) } } /** * Set prefetch limit. * Recommended to set as each unacknowledge message will be store in memory of the client. * The server won't deliver more messages than the limit until messages are acknowledged. * @param prefetchCount - number of messages to limit to * @param prefetchSize - number of bytes to limit to (not supported by RabbitMQ) * @param global - if the prefetch is limited to the channel, or if false to each consumer */ basicQos(prefetchCount: number, prefetchSize = 0, global = false) { if (this.closed) return this.rejectClosed() let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel: 1 frame.setUint32(j, 11); j += 4 // frameSize frame.setUint16(j, 60); j += 2 // class: basic frame.setUint16(j, 10); j += 2 // method: qos frame.setUint32(j, prefetchSize); j += 4 // prefetch size frame.setUint16(j, prefetchCount); j += 2 // prefetch count frame.setUint8(j, global ? 1 : 0); j += 1 // glocal frame.setUint8(j, 206); j += 1 // frame end byte return this.sendRpc(frame, j) } /** * Enable or disable flow. Disabling flow will stop the server from delivering messages to consumers. * Not supported in RabbitMQ * @param active - false to stop the flow, true to accept messages */ basicFlow(active = true) { if (this.closed) return this.rejectClosed() let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel: 1 frame.setUint32(j, 5); j += 4 // frameSize frame.setUint16(j, 20); j += 2 // class: channel frame.setUint16(j, 20); j += 2 // method: flow frame.setUint8(j, active ? 1 : 0); j += 1 // active flow frame.setUint8(j, 206); j += 1 // frame end byte return this.sendRpc(frame, j) } /** * Enable publish confirm. The server will then confirm each publish with an Ack or Nack when the message is enqueued. */ confirmSelect() { if (this.closed) return this.rejectClosed() let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel frame.setUint32(j, 5); j += 4 // frame size frame.setUint16(j, 85); j += 2 // class: confirm frame.setUint16(j, 10); j += 2 // method: select frame.setUint8(j, 0); j += 1 // noWait frame.setUint8(j, 206); j += 1 // frame end byte return this.sendRpc(frame, j) // parseFrames in base will set channel.confirmId = 0 } /** * Declare a queue * @param name - name of the queue, if empty the server will generate a name * @param params * @param [params.passive=false] - if the queue name doesn't exists the channel will be closed with an error, fulfilled if the queue name does exists * @param [params.durable=true] - if the queue should survive server restarts * @param [params.autoDelete=false] - if the queue should be deleted when the last consumer of the queue disconnects * @param [params.exclusive=false] - if the queue should be deleted when the channel is closed * @param args - optional custom queue arguments * @return fulfilled when confirmed by the server */ queueDeclare(name = "", {passive = false, durable = name !== "", autoDelete = name === "", exclusive = name === ""} = {}, args = {}) { if (this.closed) return this.rejectClosed() const noWait = false let j = 0 const declare = this.buffer declare.setUint8(j, 1); j += 1 // type: method declare.setUint16(j, this.id); j += 2 // channel: 1 declare.setUint32(j, 0); j += 4 // frameSize declare.setUint16(j, 50); j += 2 // class: queue declare.setUint16(j, 10); j += 2 // method: declare declare.setUint16(j, 0); j += 2 // reserved1 j += declare.setShortString(j, name) // name let bits = 0 if (passive) bits = bits | (1 << 0) if (durable) bits = bits | (1 << 1) if (exclusive) bits = bits | (1 << 2) if (autoDelete) bits = bits | (1 << 3) if (noWait) bits = bits | (1 << 4) declare.setUint8(j, bits); j += 1 j += declare.setTable(j, args) // arguments declare.setUint8(j, 206); j += 1 // frame end byte declare.setUint32(3, j - 8) // update frameSize return this.sendRpc(declare, j) } /** * Delete a queue * @param name - name of the queue, if empty it will delete the last declared queue * @param params * @param [params.ifUnused=false] - only delete if the queue doesn't have any consumers * @param [params.ifEmpty=false] - only delete if the queue is empty */ queueDelete(name = "", { ifUnused = false, ifEmpty = false } = {}) { if (this.closed) return this.rejectClosed() const noWait = false let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel: 1 frame.setUint32(j, 0); j += 4 // frameSize frame.setUint16(j, 50); j += 2 // class: queue frame.setUint16(j, 40); j += 2 // method: delete frame.setUint16(j, 0); j += 2 // reserved1 j += frame.setShortString(j, name) // name let bits = 0 if (ifUnused) bits = bits | (1 << 0) if (ifEmpty) bits = bits | (1 << 1) if (noWait) bits = bits | (1 << 2) frame.setUint8(j, bits); j += 1 frame.setUint8(j, 206); j += 1 // frame end byte frame.setUint32(3, j - 8) // update frameSize return this.sendRpc(frame, j) } /** * Bind a queue to an exchange * @param queue - name of the queue * @param exchange - name of the exchange * @param routingKey - key to bind with * @param args - optional arguments, e.g. for header exchanges * @return fulfilled when confirmed by the server */ queueBind(queue: string, exchange: string, routingKey: string, args = {}) { if (this.closed) return this.rejectClosed() const noWait = false let j = 0 const bind = this.buffer bind.setUint8(j, 1); j += 1 // type: method bind.setUint16(j, this.id); j += 2 // channel: 1 bind.setUint32(j, 0); j += 4 // frameSize bind.setUint16(j, 50); j += 2 // class: queue bind.setUint16(j, 20); j += 2 // method: bind bind.setUint16(j, 0); j += 2 // reserved1 j += bind.setShortString(j, queue) j += bind.setShortString(j, exchange) j += bind.setShortString(j, routingKey) bind.setUint8(j, noWait ? 1 : 0); j += 1 // noWait j += bind.setTable(j, args) bind.setUint8(j, 206); j += 1 // frame end byte bind.setUint32(3, j - 8) // update frameSize return this.sendRpc(bind, j) } /** * Unbind a queue from an exchange * @param queue - name of the queue * @param exchange - name of the exchange * @param routingKey - key that was bound * @param args - arguments, e.g. for header exchanges * @return fulfilled when confirmed by the server */ queueUnbind(queue: string, exchange: string, routingKey: string, args = {}) { if (this.closed) return this.rejectClosed() let j = 0 const unbind = this.buffer unbind.setUint8(j, 1); j += 1 // type: method unbind.setUint16(j, this.id); j += 2 // channel: 1 unbind.setUint32(j, 0); j += 4 // frameSize unbind.setUint16(j, 50); j += 2 // class: queue unbind.setUint16(j, 50); j += 2 // method: unbind unbind.setUint16(j, 0); j += 2 // reserved1 j += unbind.setShortString(j, queue) j += unbind.setShortString(j, exchange) j += unbind.setShortString(j, routingKey) j += unbind.setTable(j, args) unbind.setUint8(j, 206); j += 1 // frame end byte unbind.setUint32(3, j - 8) // update frameSize return this.sendRpc(unbind, j) } /** * Purge a queue * @param queue - name of the queue * @return fulfilled when confirmed by the server */ queuePurge(queue: string): Promise<{ messageCount: number }> { if (this.closed) return this.rejectClosed() const noWait = false let j = 0 const purge = this.buffer purge.setUint8(j, 1); j += 1 // type: method purge.setUint16(j, this.id); j += 2 // channel: 1 purge.setUint32(j, 0); j += 4 // frameSize purge.setUint16(j, 50); j += 2 // class: queue purge.setUint16(j, 30); j += 2 // method: purge purge.setUint16(j, 0); j += 2 // reserved1 j += purge.setShortString(j, queue) purge.setUint8(j, noWait ? 1 : 0); j += 1 // noWait purge.setUint8(j, 206); j += 1 // frame end byte purge.setUint32(3, j - 8) // update frameSize return this.sendRpc(purge, j) } /** * Declare an exchange * @param name - name of the exchange * @param type - type of exchange (direct, fanout, topic, header, or a custom type) * @param param * @param [param.passive=false] - if the exchange name doesn't exists the channel will be closed with an error, fulfilled if the exchange name does exists * @param [param.durable=true] - if the exchange should survive server restarts * @param [param.autoDelete=false] - if the exchange should be deleted when the last binding from it is deleted * @param [param.internal=false] - if exchange is internal to the server. Client's can't publish to internal exchanges. * @param args - optional arguments * @return Fulfilled when the exchange is created or if it already exists */ exchangeDeclare(name: string, type: string, { passive = false, durable = true, autoDelete = false, internal = false } = {}, args = {}) { const noWait = false let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel frame.setUint32(j, 0); j += 4 // frame size frame.setUint16(j, 40); j += 2 // class: exchange frame.setUint16(j, 10); j += 2 // method: declare frame.setUint16(j, 0); j += 2 // reserved1 j += frame.setShortString(j, name) j += frame.setShortString(j, type) let bits = 0 if (passive) bits = bits | (1 << 0) if (durable) bits = bits | (1 << 1) if (autoDelete) bits = bits | (1 << 2) if (internal) bits = bits | (1 << 3) if (noWait) bits = bits | (1 << 4) frame.setUint8(j, bits); j += 1 j += frame.setTable(j, args) frame.setUint8(j, 206); j += 1 // frame end byte frame.setUint32(3, j - 8) // update frameSize return this.sendRpc(frame, j) } /** * Delete an exchange * @param name - name of the exchange * @param param * @param [param.ifUnused=false] - only delete if the exchange doesn't have any bindings * @return Fulfilled when the exchange is deleted or if it's already deleted */ exchangeDelete(name: string, { ifUnused = false } = {}) { const noWait = false let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel frame.setUint32(j, 0); j += 4 // frame size frame.setUint16(j, 40); j += 2 // class: exchange frame.setUint16(j, 20); j += 2 // method: declare frame.setUint16(j, 0); j += 2 // reserved1 j += frame.setShortString(j, name) let bits = 0 if (ifUnused) bits = bits | (1 << 0) if (noWait) bits = bits | (1 << 1) frame.setUint8(j, bits); j += 1 frame.setUint8(j, 206); j += 1 // frame end byte frame.setUint32(3, j - 8) // update frameSize return this.sendRpc(frame, j) } /** * Exchange to exchange binding. * @param destination - name of the destination exchange * @param source - name of the source exchange * @param routingKey - key to bind with * @param args - optional arguments, e.g. for header exchanges * @return fulfilled when confirmed by the server */ exchangeBind(destination: string, source: string, routingKey = "", args = {}) { if (this.closed) return this.rejectClosed() let j = 0 const bind = this.buffer bind.setUint8(j, 1); j += 1 // type: method bind.setUint16(j, this.id); j += 2 // channel: 1 bind.setUint32(j, 0); j += 4 // frameSize bind.setUint16(j, 40); j += 2 // class: exchange bind.setUint16(j, 30); j += 2 // method: bind bind.setUint16(j, 0); j += 2 // reserved1 j += bind.setShortString(j, destination) j += bind.setShortString(j, source) j += bind.setShortString(j, routingKey) bind.setUint8(j, 0); j += 1 // noWait j += bind.setTable(j, args) bind.setUint8(j, 206); j += 1 // frame end byte bind.setUint32(3, j - 8) // update frameSize return this.sendRpc(bind, j) } /** * Delete an exchange-to-exchange binding * @param destination - name of destination exchange * @param source - name of the source exchange * @param routingKey - key that was bound * @param args - arguments, e.g. for header exchanges * @return fulfilled when confirmed by the server */ exchangeUnbind(destination: string, source: string, routingKey = "", args = {}) { if (this.closed) return this.rejectClosed() let j = 0 const unbind = this.buffer unbind.setUint8(j, 1); j += 1 // type: method unbind.setUint16(j, this.id); j += 2 // channel: 1 unbind.setUint32(j, 0); j += 4 // frameSize unbind.setUint16(j, 40); j += 2 // class: exchange unbind.setUint16(j, 40); j += 2 // method: unbind unbind.setUint16(j, 0); j += 2 // reserved1 j += unbind.setShortString(j, destination) j += unbind.setShortString(j, source) j += unbind.setShortString(j, routingKey) unbind.setUint8(j, 0); j += 1 // noWait j += unbind.setTable(j, args) unbind.setUint8(j, 206); j += 1 // frame end byte unbind.setUint32(3, j - 8) // update frameSize return this.sendRpc(unbind, j) } /** * Set this channel in Transaction mode. * Rember to commit the transaction, overwise the server will eventually run out of memory. */ txSelect() { return this.txMethod(10) } /** * Commit a transaction */ txCommit() { return this.txMethod(20) } /** * Rollback a transaction */ txRollback() { return this.txMethod(30) } private txMethod(methodId: number) { if (this.closed) return this.rejectClosed() let j = 0 const frame = this.buffer frame.setUint8(j, 1); j += 1 // type: method frame.setUint16(j, this.id); j += 2 // channel: 1 frame.setUint32(j, 4); j += 4 // frameSize frame.setUint16(j, 90); j += 2 // class: Tx frame.setUint16(j, methodId); j += 2 frame.setUint8(j, 206); j += 1 // frame end byte return this.sendRpc(frame, j) } /** * Resolves the next RPC promise * @ignore */ resolvePromise(value?: any) { const promise = this.promises.shift() if (promise) { const [resolve, ] = promise resolve(value) return true } return false } /** * Rejects the next RPC promise * @return true if a promise was rejected, otherwise false */ private rejectPromise(err?: Error) { const promise = this.promises.shift() if (promise) { const [, reject] = promise reject(err) return true } return false } /** * Send a RPC request, will resolve a RPC promise when RPC response arrives * @param frame with data * @param frameSize - bytes the frame actually is */ private sendRpc(frame: AMQPView, frameSize: number): Promise<any> { return new Promise((resolve, reject) => { this.connection.send(new Uint8Array(frame.buffer, 0, frameSize)) .then(() => this.promises.push([resolve, reject])) .catch(reject) }) } /** * Marks the channel as closed * All outstanding RPC requests will be rejected * All outstanding publish confirms will be rejected * All consumers will be marked as closed * @ignore * @param [err] - why the channel was closed */ setClosed(err?: Error) { if (!this.closed) { this.closed = true this.consumers.forEach((consumer) => consumer.setClosed(err)) this.consumers.clear() // Empty and reject all RPC promises while(this.rejectPromise(err)) { 1 } this.unconfirmedPublishes.forEach(([, , reject]) => reject(err)) } } /** * @return Rejected promise with an error */ private rejectClosed() { return Promise.reject(new AMQPError("Channel is closed", this.connection)) } /** * Called from AMQPBaseClient when a publish is confirmed by the server. * Will fulfill one or more (if multiple) Unconfirmed Publishes. * @ignore * @param deliveryTag * @param multiple - true if all unconfirmed publishes up to this deliveryTag should be resolved or just this one * @param nack - true if negative confirm, hence reject the unconfirmed publish(es) */ publishConfirmed(deliveryTag: number, multiple: boolean, nack: boolean) { // is queueMicrotask() needed here? const idx = this.unconfirmedPublishes.findIndex(([tag,]) => tag === deliveryTag) if (idx !== -1) { const confirmed = multiple ? this.unconfirmedPublishes.splice(0, idx + 1) : this.unconfirmedPublishes.splice(idx, 1) confirmed.forEach(([tag, resolve, reject]) => { if (nack) reject(new Error("Message rejected")) else resolve(tag) }) } else { console.warn("Cant find unconfirmed deliveryTag", deliveryTag, "multiple:", multiple, "nack:", nack) } } /** * Called from AMQPBaseClient when a message is ready * @ignore * @param message */ onMessageReady(message: AMQPMessage) { if (this.delivery) { delete this.delivery this.deliver(message) } else if (this.getMessage) { delete this.getMessage this.resolvePromise(message) } else { delete this.returned this.onReturn(message) } } /** * Deliver a message to a consumer * @ignore */ deliver(message: AMQPMessage) { queueMicrotask(() => { const consumer = this.consumers.get(message.consumerTag) if (consumer) { consumer.onMessage(message) } else { console.warn("Consumer", message.consumerTag, "not available on channel", this.id) } }) } }