rocketmq-client-nodejs-beta
Version:
RocketMQ Node.js Client
64 lines • 5.67 kB
JavaScript
;
/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
Object.defineProperty(exports, "__esModule", { value: true });
exports.Transaction = void 0;
const definition_pb_1 = require("../../proto/apache/rocketmq/v2/definition_pb");
const message_1 = require("../message");
class Transaction {
static MAX_MESSAGE_NUM = 1;
#producer;
#messageMap = new Map();
#messageSendReceiptMap = new Map();
constructor(producer) {
this.#producer = producer;
}
tryAddMessage(message) {
if (this.#messageMap.size >= Transaction.MAX_MESSAGE_NUM) {
throw new TypeError(`Message in transaction has exceeded the threshold=${Transaction.MAX_MESSAGE_NUM}`);
}
const publishingMessage = new message_1.PublishingMessage(message, this.#producer.publishingSettings, true);
this.#messageMap.set(publishingMessage.messageId, publishingMessage);
return publishingMessage;
}
tryAddReceipt(publishingMessage, sendReceipt) {
if (!this.#messageMap.has(publishingMessage.messageId)) {
throw new TypeError('Message not in transaction');
}
this.#messageSendReceiptMap.set(publishingMessage.messageId, sendReceipt);
}
async commit() {
if (this.#messageSendReceiptMap.size === 0) {
throw new TypeError('Transactional message has not been sent yet');
}
for (const [messageId, sendReceipt] of this.#messageSendReceiptMap.entries()) {
const publishingMessage = this.#messageMap.get(messageId);
await this.#producer.endTransaction(sendReceipt.endpoints, publishingMessage, sendReceipt.messageId, sendReceipt.transactionId, definition_pb_1.TransactionResolution.COMMIT);
}
}
async rollback() {
if (this.#messageSendReceiptMap.size === 0) {
throw new TypeError('Transactional message has not been sent yet');
}
for (const [messageId, sendReceipt] of this.#messageSendReceiptMap.entries()) {
const publishingMessage = this.#messageMap.get(messageId);
await this.#producer.endTransaction(sendReceipt.endpoints, publishingMessage, sendReceipt.messageId, sendReceipt.transactionId, definition_pb_1.TransactionResolution.ROLLBACK);
}
}
}
exports.Transaction = Transaction;
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiVHJhbnNhY3Rpb24uanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvcHJvZHVjZXIvVHJhbnNhY3Rpb24udHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IjtBQUFBOzs7Ozs7Ozs7Ozs7Ozs7R0FlRzs7O0FBRUgsZ0ZBQXFGO0FBQ3JGLHdDQUErRDtBQUkvRCxNQUFhLFdBQVc7SUFDdEIsTUFBTSxDQUFVLGVBQWUsR0FBRyxDQUFDLENBQUM7SUFDM0IsU0FBUyxDQUFXO0lBQ3BCLFdBQVcsR0FBRyxJQUFJLEdBQUcsRUFBNkIsQ0FBQztJQUNuRCxzQkFBc0IsR0FBRyxJQUFJLEdBQUcsRUFBdUIsQ0FBQztJQUVqRSxZQUFZLFFBQWtCO1FBQzVCLElBQUksQ0FBQyxTQUFTLEdBQUcsUUFBUSxDQUFDO0lBQzVCLENBQUM7SUFFRCxhQUFhLENBQUMsT0FBdUI7UUFDbkMsSUFBSSxJQUFJLENBQUMsV0FBVyxDQUFDLElBQUksSUFBSSxXQUFXLENBQUMsZUFBZSxFQUFFLENBQUM7WUFDekQsTUFBTSxJQUFJLFNBQVMsQ0FBQyxxREFBcUQsV0FBVyxDQUFDLGVBQWUsRUFBRSxDQUFDLENBQUM7UUFDMUcsQ0FBQztRQUNELE1BQU0saUJBQWlCLEdBQUcsSUFBSSwyQkFBaUIsQ0FBQyxPQUFPLEVBQUUsSUFBSSxDQUFDLFNBQVMsQ0FBQyxrQkFBa0IsRUFBRSxJQUFJLENBQUMsQ0FBQztRQUNsRyxJQUFJLENBQUMsV0FBVyxDQUFDLEdBQUcsQ0FBQyxpQkFBaUIsQ0FBQyxTQUFTLEVBQUUsaUJBQWlCLENBQUMsQ0FBQztRQUNyRSxPQUFPLGlCQUFpQixDQUFDO0lBQzNCLENBQUM7SUFFRCxhQUFhLENBQUMsaUJBQW9DLEVBQUUsV0FBd0I7UUFDMUUsSUFBSSxDQUFDLElBQUksQ0FBQyxXQUFXLENBQUMsR0FBRyxDQUFDLGlCQUFpQixDQUFDLFNBQVMsQ0FBQyxFQUFFLENBQUM7WUFDdkQsTUFBTSxJQUFJLFNBQVMsQ0FBQyw0QkFBNEIsQ0FBQyxDQUFDO1FBQ3BELENBQUM7UUFDRCxJQUFJLENBQUMsc0JBQXNCLENBQUMsR0FBRyxDQUFDLGlCQUFpQixDQUFDLFNBQVMsRUFBRSxXQUFXLENBQUMsQ0FBQztJQUM1RSxDQUFDO0lBRUQsS0FBSyxDQUFDLE1BQU07UUFDVixJQUFJLElBQUksQ0FBQyxzQkFBc0IsQ0FBQyxJQUFJLEtBQUssQ0FBQyxFQUFFLENBQUM7WUFDM0MsTUFBTSxJQUFJLFNBQVMsQ0FBQyw2Q0FBNkMsQ0FBQyxDQUFDO1FBQ3JFLENBQUM7UUFDRCxLQUFLLE1BQU0sQ0FBRSxTQUFTLEVBQUUsV0FBVyxDQUFFLElBQUksSUFBSSxDQUFDLHNCQUFzQixDQUFDLE9BQU8sRUFBRSxFQUFFLENBQUM7WUFDL0UsTUFBTSxpQkFBaUIsR0FBRyxJQUFJLENBQUMsV0FBVyxDQUFDLEdBQUcsQ0FBQyxTQUFTLENBQUUsQ0FBQztZQUMzRCxNQUFNLElBQUksQ0FBQyxTQUFTLENBQUMsY0FBYyxDQUFDLFdBQVcsQ0FBQyxTQUFTLEVBQUUsaUJBQWlCLEVBQzFFLFdBQVcsQ0FBQyxTQUFTLEVBQUUsV0FBVyxDQUFDLGFBQWEsRUFBRSxxQ0FBcUIsQ0FBQyxNQUFNLENBQUMsQ0FBQztRQUNwRixDQUFDO0lBQ0gsQ0FBQztJQUVELEtBQUssQ0FBQyxRQUFRO1FBQ1osSUFBSSxJQUFJLENBQUMsc0JBQXNCLENBQUMsSUFBSSxLQUFLLENBQUMsRUFBRSxDQUFDO1lBQzNDLE1BQU0sSUFBSSxTQUFTLENBQUMsNkNBQTZDLENBQUMsQ0FBQztRQUNyRSxDQUFDO1FBQ0QsS0FBSyxNQUFNLENBQUUsU0FBUyxFQUFFLFdBQVcsQ0FBRSxJQUFJLElBQUksQ0FBQyxzQkFBc0IsQ0FBQyxPQUFPLEVBQUUsRUFBRSxDQUFDO1lBQy9FLE1BQU0saUJBQWlCLEdBQUcsSUFBSSxDQUFDLFdBQVcsQ0FBQyxHQUFHLENBQUMsU0FBUyxDQUFFLENBQUM7WUFDM0QsTUFBTSxJQUFJLENBQUMsU0FBUyxDQUFDLGNBQWMsQ0FBQyxXQUFXLENBQUMsU0FBUyxFQUFFLGlCQUFpQixFQUMxRSxXQUFXLENBQUMsU0FBUyxFQUFFLFdBQVcsQ0FBQyxhQUFhLEVBQUUscUNBQXFCLENBQUMsUUFBUSxDQUFDLENBQUM7UUFDdEYsQ0FBQztJQUNILENBQUM7O0FBOUNILGtDQStDQyJ9