rocketmq-client-nodejs-beta
Version:
RocketMQ Node.js Client
82 lines • 7.11 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.PublishingLoadBalancer = void 0;
const node_crypto_1 = require("node:crypto");
const definition_pb_1 = require("../../proto/apache/rocketmq/v2/definition_pb");
const util_1 = require("../util");
class PublishingLoadBalancer {
#index;
#messageQueues;
constructor(topicRouteData, index) {
this.#messageQueues = topicRouteData.messageQueues.filter(mq => {
return mq.queueId === util_1.MASTER_BROKER_ID && (mq.permission === definition_pb_1.Permission.WRITE || mq.permission === definition_pb_1.Permission.READ_WRITE);
});
this.#index = index === undefined ? (0, node_crypto_1.randomInt)(this.#messageQueues.length) : index;
if (this.#messageQueues.length === 0) {
throw new Error(`No writable message queue found, topicRouteData=${JSON.stringify(topicRouteData)}`);
}
}
update(topicRouteData) {
return new PublishingLoadBalancer(topicRouteData, this.#index);
}
takeMessageQueues(excluded, count) {
if (this.#index >= this.#messageQueues.length) {
this.#index = 0;
}
let next = this.#index++;
const candidates = [];
const candidateBrokerNames = new Set();
const size = this.#messageQueues.length;
for (let i = 0; i < size; i++) {
const messageQueue = this.#messageQueues[next++ % size];
const broker = messageQueue.broker;
const brokerName = broker.name;
if (!excluded.has(broker.endpoints.facade) && !candidateBrokerNames.has(brokerName)) {
candidateBrokerNames.add(brokerName);
candidates.push(messageQueue);
}
if (candidates.length >= count) {
return candidates;
}
}
// If all endpoints are isolated.
if (candidates.length === 0) {
for (let i = 0; i < size; i++) {
const messageQueue = this.#messageQueues[next++ % size];
const broker = messageQueue.broker;
const brokerName = broker.name;
if (!candidateBrokerNames.has(brokerName)) {
candidateBrokerNames.add(brokerName);
candidates.push(messageQueue);
}
if (candidates.length >= count) {
return candidates;
}
}
}
return candidates;
}
takeMessageQueueByMessageGroup(messageGroup) {
const hashCode = (0, util_1.calculateStringSipHash24)(messageGroup);
const index = parseInt(`${hashCode % BigInt(this.#messageQueues.length)}`);
return this.#messageQueues[index];
}
}
exports.PublishingLoadBalancer = PublishingLoadBalancer;
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiUHVibGlzaGluZ0xvYWRCYWxhbmNlci5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9wcm9kdWNlci9QdWJsaXNoaW5nTG9hZEJhbGFuY2VyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7QUFBQTs7Ozs7Ozs7Ozs7Ozs7O0dBZUc7OztBQUVILDZDQUF3QztBQUN4QyxnRkFBMEU7QUFFMUUsa0NBQXFFO0FBRXJFLE1BQWEsc0JBQXNCO0lBQ2pDLE1BQU0sQ0FBUztJQUNmLGNBQWMsQ0FBaUI7SUFFL0IsWUFBWSxjQUE4QixFQUFFLEtBQWM7UUFDeEQsSUFBSSxDQUFDLGNBQWMsR0FBRyxjQUFjLENBQUMsYUFBYSxDQUFDLE1BQU0sQ0FBQyxFQUFFLENBQUMsRUFBRTtZQUM3RCxPQUFPLEVBQUUsQ0FBQyxPQUFPLEtBQUssdUJBQWdCLElBQUksQ0FBQyxFQUFFLENBQUMsVUFBVSxLQUFLLDBCQUFVLENBQUMsS0FBSyxJQUFJLEVBQUUsQ0FBQyxVQUFVLEtBQUssMEJBQVUsQ0FBQyxVQUFVLENBQUMsQ0FBQztRQUM1SCxDQUFDLENBQUMsQ0FBQztRQUNILElBQUksQ0FBQyxNQUFNLEdBQUcsS0FBSyxLQUFLLFNBQVMsQ0FBQyxDQUFDLENBQUMsSUFBQSx1QkFBUyxFQUFDLElBQUksQ0FBQyxjQUFjLENBQUMsTUFBTSxDQUFDLENBQUMsQ0FBQyxDQUFDLEtBQUssQ0FBQztRQUNsRixJQUFJLElBQUksQ0FBQyxjQUFjLENBQUMsTUFBTSxLQUFLLENBQUMsRUFBRSxDQUFDO1lBQ3JDLE1BQU0sSUFBSSxLQUFLLENBQUMsbURBQW1ELElBQUksQ0FBQyxTQUFTLENBQUMsY0FBYyxDQUFDLEVBQUUsQ0FBQyxDQUFDO1FBQ3ZHLENBQUM7SUFDSCxDQUFDO0lBRUQsTUFBTSxDQUFDLGNBQThCO1FBQ25DLE9BQU8sSUFBSSxzQkFBc0IsQ0FBQyxjQUFjLEVBQUUsSUFBSSxDQUFDLE1BQU0sQ0FBQyxDQUFDO0lBQ2pFLENBQUM7SUFFRCxpQkFBaUIsQ0FBQyxRQUFnQyxFQUFFLEtBQWE7UUFDL0QsSUFBSSxJQUFJLENBQUMsTUFBTSxJQUFJLElBQUksQ0FBQyxjQUFjLENBQUMsTUFBTSxFQUFFLENBQUM7WUFDOUMsSUFBSSxDQUFDLE1BQU0sR0FBRyxDQUFDLENBQUM7UUFDbEIsQ0FBQztRQUNELElBQUksSUFBSSxHQUFHLElBQUksQ0FBQyxNQUFNLEVBQUUsQ0FBQztRQUN6QixNQUFNLFVBQVUsR0FBbUIsRUFBRSxDQUFDO1FBQ3RDLE1BQU0sb0JBQW9CLEdBQUcsSUFBSSxHQUFHLEVBQVUsQ0FBQztRQUUvQyxNQUFNLElBQUksR0FBRyxJQUFJLENBQUMsY0FBYyxDQUFDLE1BQU0sQ0FBQztRQUN4QyxLQUFLLElBQUksQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLEdBQUcsSUFBSSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7WUFDOUIsTUFBTSxZQUFZLEdBQUcsSUFBSSxDQUFDLGNBQWMsQ0FBQyxJQUFJLEVBQUUsR0FBRyxJQUFJLENBQUMsQ0FBQztZQUN4RCxNQUFNLE1BQU0sR0FBRyxZQUFZLENBQUMsTUFBTSxDQUFDO1lBQ25DLE1BQU0sVUFBVSxHQUFHLE1BQU0sQ0FBQyxJQUFJLENBQUM7WUFDL0IsSUFBSSxDQUFDLFFBQVEsQ0FBQyxHQUFHLENBQUMsTUFBTSxDQUFDLFNBQVMsQ0FBQyxNQUFNLENBQUMsSUFBSSxDQUFDLG9CQUFvQixDQUFDLEdBQUcsQ0FBQyxVQUFVLENBQUMsRUFBRSxDQUFDO2dCQUNwRixvQkFBb0IsQ0FBQyxHQUFHLENBQUMsVUFBVSxDQUFDLENBQUM7Z0JBQ3JDLFVBQVUsQ0FBQyxJQUFJLENBQUMsWUFBWSxDQUFDLENBQUM7WUFDaEMsQ0FBQztZQUNELElBQUksVUFBVSxDQUFDLE1BQU0sSUFBSSxLQUFLLEVBQUUsQ0FBQztnQkFDL0IsT0FBTyxVQUFVLENBQUM7WUFDcEIsQ0FBQztRQUNILENBQUM7UUFDRCxpQ0FBaUM7UUFDakMsSUFBSSxVQUFVLENBQUMsTUFBTSxLQUFLLENBQUMsRUFBRSxDQUFDO1lBQzVCLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxJQUFJLEVBQUUsQ0FBQyxFQUFFLEVBQUUsQ0FBQztnQkFDOUIsTUFBTSxZQUFZLEdBQUcsSUFBSSxDQUFDLGNBQWMsQ0FBQyxJQUFJLEVBQUUsR0FBRyxJQUFJLENBQUMsQ0FBQztnQkFDeEQsTUFBTSxNQUFNLEdBQUcsWUFBWSxDQUFDLE1BQU0sQ0FBQztnQkFDbkMsTUFBTSxVQUFVLEdBQUcsTUFBTSxDQUFDLElBQUksQ0FBQztnQkFDL0IsSUFBSSxDQUFDLG9CQUFvQixDQUFDLEdBQUcsQ0FBQyxVQUFVLENBQUMsRUFBRSxDQUFDO29CQUMxQyxvQkFBb0IsQ0FBQyxHQUFHLENBQUMsVUFBVSxDQUFDLENBQUM7b0JBQ3JDLFVBQVUsQ0FBQyxJQUFJLENBQUMsWUFBWSxDQUFDLENBQUM7Z0JBQ2hDLENBQUM7Z0JBQ0QsSUFBSSxVQUFVLENBQUMsTUFBTSxJQUFJLEtBQUssRUFBRSxDQUFDO29CQUMvQixPQUFPLFVBQVUsQ0FBQztnQkFDcEIsQ0FBQztZQUNILENBQUM7UUFDSCxDQUFDO1FBQ0QsT0FBTyxVQUFVLENBQUM7SUFDcEIsQ0FBQztJQUVELDhCQUE4QixDQUFDLFlBQW9CO1FBQ2pELE1BQU0sUUFBUSxHQUFHLElBQUEsK0JBQXdCLEVBQUMsWUFBWSxDQUFDLENBQUM7UUFDeEQsTUFBTSxLQUFLLEdBQUcsUUFBUSxDQUFDLEdBQUcsUUFBUSxHQUFHLE1BQU0sQ0FBQyxJQUFJLENBQUMsY0FBYyxDQUFDLE1BQU0sQ0FBQyxFQUFFLENBQUMsQ0FBQztRQUMzRSxPQUFPLElBQUksQ0FBQyxjQUFjLENBQUMsS0FBSyxDQUFDLENBQUM7SUFDcEMsQ0FBQztDQUNGO0FBOURELHdEQThEQyJ9