276 lines
8.8 KiB
JavaScript
276 lines
8.8 KiB
JavaScript
/*
|
||
* (c) Copyright Ascensio System SIA 2010-2024
|
||
*
|
||
* This program is a free software product. You can redistribute it and/or
|
||
* modify it under the terms of the GNU Affero General Public License (AGPL)
|
||
* version 3 as published by the Free Software Foundation. In accordance with
|
||
* Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
|
||
* that Ascensio System SIA expressly excludes the warranty of non-infringement
|
||
* of any third-party rights.
|
||
*
|
||
* This program is distributed WITHOUT ANY WARRANTY; without even the implied
|
||
* warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. For
|
||
* details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
|
||
*
|
||
* You can contact Ascensio System SIA at 20A-6 Ernesta Birznieka-Upish
|
||
* street, Riga, Latvia, EU, LV-1050.
|
||
*
|
||
* The interactive user interfaces in modified source and object code versions
|
||
* of the Program must display Appropriate Legal Notices, as required under
|
||
* Section 5 of the GNU AGPL version 3.
|
||
*
|
||
* Pursuant to Section 7(b) of the License you must retain the original Product
|
||
* logo when distributing the program. Pursuant to Section 7(e) we decline to
|
||
* grant you any rights under trademark law for use of our trademarks.
|
||
*
|
||
* All the Product's GUI elements, including illustrations and icon sets, as
|
||
* well as technical writing content are licensed under the terms of the
|
||
* Creative Commons Attribution-ShareAlike 4.0 International. See the License
|
||
* terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
|
||
*
|
||
*/
|
||
|
||
'use strict';
|
||
const config = require('config');
|
||
const events = require('events');
|
||
const util = require('util');
|
||
const co = require('co');
|
||
const constants = require('./../../Common/sources/constants');
|
||
const commonDefines = require('./../../Common/sources/commondefines');
|
||
const rabbitMQCore = require('./../../Common/sources/rabbitMQCore');
|
||
const activeMQCore = require('./../../Common/sources/activeMQCore');
|
||
|
||
const cfgQueueType = config.get('queue.type');
|
||
const cfgRabbitExchangePubSub = config.util.cloneDeep(config.get('rabbitmq.exchangepubsub'));
|
||
const cfgRabbitQueuePubsub = config.util.cloneDeep(config.get('rabbitmq.queuepubsub'));
|
||
const cfgActiveTopicPubSub = constants.ACTIVEMQ_TOPIC_PREFIX + config.get('activemq.topicpubsub');
|
||
|
||
function initRabbit(pubsub, callback) {
|
||
return co(function* () {
|
||
let e = null;
|
||
try {
|
||
const conn = yield rabbitMQCore.connetPromise(() => {
|
||
clear(pubsub);
|
||
if (!pubsub.isClose) {
|
||
setTimeout(() => {
|
||
init(pubsub, null);
|
||
}, rabbitMQCore.RECONNECT_TIMEOUT);
|
||
}
|
||
});
|
||
pubsub.connection = conn;
|
||
pubsub.channelPublish = yield rabbitMQCore.createChannelPromise(conn);
|
||
pubsub.exchangePublish = yield rabbitMQCore.assertExchangePromise(
|
||
pubsub.channelPublish,
|
||
cfgRabbitExchangePubSub.name,
|
||
'fanout',
|
||
cfgRabbitExchangePubSub.options
|
||
);
|
||
|
||
pubsub.channelReceive = yield rabbitMQCore.createChannelPromise(conn);
|
||
const queue = yield rabbitMQCore.assertQueuePromise(pubsub.channelReceive, cfgRabbitQueuePubsub.name, cfgRabbitQueuePubsub.options);
|
||
pubsub.channelReceive.bindQueue(queue, cfgRabbitExchangePubSub.name, '');
|
||
yield rabbitMQCore.consumePromise(
|
||
pubsub.channelReceive,
|
||
queue,
|
||
message => {
|
||
if (null != pubsub.channelReceive) {
|
||
if (message) {
|
||
pubsub.emit('message', message.content.toString());
|
||
}
|
||
pubsub.channelReceive.ack(message);
|
||
}
|
||
},
|
||
{noAck: false}
|
||
);
|
||
//process messages received while reconnection time
|
||
yield repeat(pubsub);
|
||
} catch (err) {
|
||
e = err;
|
||
}
|
||
if (callback) {
|
||
callback(e);
|
||
}
|
||
});
|
||
}
|
||
function initActive(pubsub, callback) {
|
||
return co(function* () {
|
||
let e = null;
|
||
try {
|
||
const conn = yield activeMQCore.connetPromise(() => {
|
||
clear(pubsub);
|
||
if (!pubsub.isClose) {
|
||
setTimeout(() => {
|
||
init(pubsub, null);
|
||
}, activeMQCore.RECONNECT_TIMEOUT);
|
||
}
|
||
});
|
||
pubsub.connection = conn;
|
||
//https://github.com/amqp/rhea/issues/251#issuecomment-535076570
|
||
const optionsPubSubSender = {
|
||
target: {
|
||
address: cfgActiveTopicPubSub,
|
||
capabilities: ['topic']
|
||
}
|
||
};
|
||
pubsub.channelPublish = yield activeMQCore.openSenderPromise(conn, optionsPubSubSender);
|
||
|
||
const optionsPubSubReceiver = {
|
||
source: {
|
||
address: cfgActiveTopicPubSub,
|
||
capabilities: ['topic']
|
||
},
|
||
credit_window: 0,
|
||
autoaccept: false
|
||
};
|
||
const receiver = yield activeMQCore.openReceiverPromise(conn, optionsPubSubReceiver);
|
||
//todo ?consumer.dispatchAsync=false&consumer.prefetchSize=1
|
||
receiver.add_credit(1);
|
||
receiver.on('message', context => {
|
||
if (context) {
|
||
pubsub.emit('message', context.message.body);
|
||
}
|
||
|
||
context.delivery.accept();
|
||
receiver.add_credit(1);
|
||
});
|
||
//process messages received while reconnection time
|
||
yield repeat(pubsub);
|
||
} catch (err) {
|
||
e = err;
|
||
}
|
||
if (callback) {
|
||
callback(e);
|
||
}
|
||
});
|
||
}
|
||
function clear(pubsub) {
|
||
pubsub.channelPublish = null;
|
||
pubsub.exchangePublish = null;
|
||
pubsub.channelReceive = null;
|
||
}
|
||
function repeat(pubsub) {
|
||
return co(function* () {
|
||
for (let i = 0; i < pubsub.publishStore.length; ++i) {
|
||
yield publish(pubsub, pubsub.publishStore[i]);
|
||
}
|
||
pubsub.publishStore.length = 0;
|
||
});
|
||
}
|
||
function publishRabbit(pubsub, data) {
|
||
return new Promise((resolve, _reject) => {
|
||
//Channels act like stream.Writable when you call publish or sendToQueue: they return either true, meaning “keep sending”, or false, meaning “please wait for a ‘drain’ event”.
|
||
const keepSending = pubsub.channelPublish.publish(pubsub.exchangePublish, '', data);
|
||
if (!keepSending) {
|
||
//todo (node:4308) MaxListenersExceededWarning: Possible EventEmitter memory leak detected. 11 drain listeners added to [Sender]. Use emitter.setMaxListeners() to increase limit
|
||
pubsub.channelPublish.once('drain', resolve);
|
||
} else {
|
||
resolve();
|
||
}
|
||
});
|
||
}
|
||
|
||
function publishActive(pubsub, data) {
|
||
return new Promise((resolve, _reject) => {
|
||
//Returns true if the sender has available credits for sending a message. Otherwise it returns false.
|
||
const sendable = pubsub.channelPublish.sendable();
|
||
if (!sendable) {
|
||
//todo (node:4308) MaxListenersExceededWarning: Possible EventEmitter memory leak detected. 11 sendable listeners added to [Sender]. Use emitter.setMaxListeners() to increase limit
|
||
pubsub.channelPublish.once('sendable', () => {
|
||
resolve(publishActive(pubsub, data));
|
||
});
|
||
} else {
|
||
pubsub.channelPublish.send({durable: true, body: data});
|
||
resolve();
|
||
}
|
||
});
|
||
}
|
||
function closeRabbit(conn) {
|
||
return rabbitMQCore.closePromise(conn);
|
||
}
|
||
function closeActive(conn) {
|
||
return activeMQCore.closePromise(conn);
|
||
}
|
||
|
||
function healthCheckRabbit(pubsub) {
|
||
return co(function* () {
|
||
if (!pubsub.channelPublish) {
|
||
return false;
|
||
}
|
||
const exchange = yield rabbitMQCore.assertExchangePromise(
|
||
pubsub.channelPublish,
|
||
cfgRabbitExchangePubSub.name,
|
||
'fanout',
|
||
cfgRabbitExchangePubSub.options
|
||
);
|
||
return !!exchange;
|
||
});
|
||
}
|
||
function healthCheckActive(pubsub) {
|
||
return co(function* () {
|
||
if (!pubsub.connection) {
|
||
return false;
|
||
}
|
||
yield null;
|
||
return pubsub.connection.is_open();
|
||
});
|
||
}
|
||
|
||
let init;
|
||
let publish;
|
||
let close;
|
||
let healthCheck;
|
||
if (commonDefines.c_oAscQueueType.rabbitmq === cfgQueueType) {
|
||
init = initRabbit;
|
||
publish = publishRabbit;
|
||
close = closeRabbit;
|
||
healthCheck = healthCheckRabbit;
|
||
} else {
|
||
init = initActive;
|
||
publish = publishActive;
|
||
close = closeActive;
|
||
healthCheck = healthCheckActive;
|
||
}
|
||
|
||
function PubsubRabbitMQ() {
|
||
this.isClose = false;
|
||
this.connection = null;
|
||
this.channelPublish = null;
|
||
this.exchangePublish = null;
|
||
this.channelReceive = null;
|
||
this.publishStore = [];
|
||
}
|
||
util.inherits(PubsubRabbitMQ, events.EventEmitter);
|
||
PubsubRabbitMQ.prototype.init = function (callback) {
|
||
init(this, callback);
|
||
};
|
||
PubsubRabbitMQ.prototype.initPromise = function () {
|
||
const t = this;
|
||
return new Promise((resolve, reject) => {
|
||
init(t, err => {
|
||
if (err) {
|
||
reject(err);
|
||
} else {
|
||
resolve();
|
||
}
|
||
});
|
||
});
|
||
};
|
||
PubsubRabbitMQ.prototype.publish = function (message) {
|
||
const data = Buffer.from(message);
|
||
if (null != this.channelPublish) {
|
||
return publish(this, data);
|
||
} else {
|
||
this.publishStore.push(data);
|
||
return Promise.resolve();
|
||
}
|
||
};
|
||
PubsubRabbitMQ.prototype.close = function () {
|
||
this.isClose = true;
|
||
return close(this.connection);
|
||
};
|
||
PubsubRabbitMQ.prototype.healthCheck = function () {
|
||
return healthCheck(this);
|
||
};
|
||
|
||
module.exports = PubsubRabbitMQ;
|