pubnub
Version:
Publish & Subscribe Real-time Messaging with PubNub
210 lines (209 loc) • 10.2 kB
JavaScript
"use strict";
/**
* Subscribe Event Engine module.
*
* @internal
*/
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.EventEngine = void 0;
const receiving_1 = require("./states/receiving");
const dispatcher_1 = require("./dispatcher");
const utils_1 = require("../core/utils");
const unsubscribed_1 = require("./states/unsubscribed");
const core_1 = require("./core");
const categories_1 = __importDefault(require("../core/constants/categories"));
const utils = __importStar(require("../core/utils"));
const events = __importStar(require("./events"));
/**
* Subscribe Event Engine Core.
*
* @internal
*/
class EventEngine {
get _engine() {
return this.engine;
}
constructor(dependencies) {
this.channels = [];
this.groups = [];
this.dependencies = dependencies;
this.engine = new core_1.Engine(dependencies.config.logger());
this.dispatcher = new dispatcher_1.EventEngineDispatcher(this.engine, dependencies);
dependencies.config.logger().debug('EventEngine', 'Create subscribe event engine.');
this._unsubscribeEngine = this.engine.subscribe((change) => {
if (change.type === 'invocationDispatched') {
this.dispatcher.dispatch(change.invocation);
}
});
this.engine.start(unsubscribed_1.UnsubscribedState, undefined);
}
/**
* Subscription-based current timetoken.
*
* @returns Timetoken based on current timetoken plus diff between current and loop start time.
*/
get subscriptionTimetoken() {
const currentState = this.engine.currentState;
if (!currentState)
return undefined;
let referenceTimetoken;
let currentTimetoken = '0';
if (currentState.label === receiving_1.ReceivingState.label) {
const context = this.engine.currentContext;
currentTimetoken = context.cursor.timetoken;
referenceTimetoken = context.referenceTimetoken;
}
return (0, utils_1.subscriptionTimetokenFromReference)(currentTimetoken, referenceTimetoken !== null && referenceTimetoken !== void 0 ? referenceTimetoken : '0');
}
subscribe({ channels, channelGroups, timetoken, withPresence, }) {
var _a;
// check if the channels and groups are already subscribed
const hasNewChannels = channels === null || channels === void 0 ? void 0 : channels.some((channel) => !this.channels.includes(channel));
const hasNewGroups = channelGroups === null || channelGroups === void 0 ? void 0 : channelGroups.some((group) => !this.groups.includes(group));
const hasNewSubscriptions = hasNewChannels || hasNewGroups;
this.channels = [...this.channels, ...(channels !== null && channels !== void 0 ? channels : [])];
this.groups = [...this.groups, ...(channelGroups !== null && channelGroups !== void 0 ? channelGroups : [])];
if (withPresence) {
this.channels.map((c) => this.channels.push(`${c}-pnpres`));
this.groups.map((g) => this.groups.push(`${g}-pnpres`));
}
if (timetoken) {
this.engine.transition(events.restore(Array.from(new Set([...this.channels, ...(channels !== null && channels !== void 0 ? channels : [])])), Array.from(new Set([...this.groups, ...(channelGroups !== null && channelGroups !== void 0 ? channelGroups : [])])), timetoken));
}
else {
if (hasNewSubscriptions) {
this.engine.transition(events.subscriptionChange(Array.from(new Set([...this.channels, ...(channels !== null && channels !== void 0 ? channels : [])])), Array.from(new Set([...this.groups, ...(channelGroups !== null && channelGroups !== void 0 ? channelGroups : [])]))));
}
else {
this.dependencies.config
.logger()
.debug('EventEngine', 'Skipping state transition - all channels/groups already subscribed. Emitting SubscriptionChanged event.');
// Get current timetoken from state context
const currentState = this.engine.currentState;
const currentContext = this.engine.currentContext;
let currentTimetoken = '0';
if ((currentState === null || currentState === void 0 ? void 0 : currentState.label) === receiving_1.ReceivingState.label && currentContext) {
const receivingContext = currentContext;
currentTimetoken = (_a = receivingContext.cursor) === null || _a === void 0 ? void 0 : _a.timetoken;
}
// Manually emit SubscriptionChanged status event
this.dependencies.emitStatus({
category: categories_1.default.PNSubscriptionChangedCategory,
affectedChannels: Array.from(new Set(this.channels)),
affectedChannelGroups: Array.from(new Set(this.groups)),
currentTimetoken,
});
}
}
if (this.dependencies.join) {
this.dependencies.join({
channels: Array.from(new Set(this.channels.filter((c) => !c.endsWith('-pnpres')))),
groups: Array.from(new Set(this.groups.filter((g) => !g.endsWith('-pnpres')))),
});
}
}
unsubscribe({ channels = [], channelGroups = [] }) {
const filteredChannels = utils.removeSingleOccurrence(this.channels, [
...channels,
...channels.map((c) => `${c}-pnpres`),
]);
const filteredGroups = utils.removeSingleOccurrence(this.groups, [
...channelGroups,
...channelGroups.map((c) => `${c}-pnpres`),
]);
if (new Set(this.channels).size !== new Set(filteredChannels).size ||
new Set(this.groups).size !== new Set(filteredGroups).size) {
const channelsToLeave = utils.findUniqueCommonElements(this.channels, channels);
const groupsToLeave = utils.findUniqueCommonElements(this.groups, channelGroups);
if (this.dependencies.presenceState) {
channelsToLeave === null || channelsToLeave === void 0 ? void 0 : channelsToLeave.forEach((c) => delete this.dependencies.presenceState[c]);
groupsToLeave === null || groupsToLeave === void 0 ? void 0 : groupsToLeave.forEach((g) => delete this.dependencies.presenceState[g]);
}
this.channels = filteredChannels;
this.groups = filteredGroups;
this.engine.transition(events.subscriptionChange(Array.from(new Set(this.channels.slice(0))), Array.from(new Set(this.groups.slice(0)))));
if (this.dependencies.leave) {
this.dependencies.leave({
channels: channelsToLeave.slice(0),
groups: groupsToLeave.slice(0),
});
}
}
}
unsubscribeAll(isOffline = false) {
const channelGroups = this.getSubscribedChannelGroups();
const channels = this.getSubscribedChannels();
this.channels = [];
this.groups = [];
if (this.dependencies.presenceState) {
Object.keys(this.dependencies.presenceState).forEach((objectName) => {
delete this.dependencies.presenceState[objectName];
});
}
this.engine.transition(events.subscriptionChange(this.channels.slice(0), this.groups.slice(0), isOffline));
if (this.dependencies.leaveAll)
this.dependencies.leaveAll({ channels, groups: channelGroups, isOffline });
}
reconnect({ timetoken, region }) {
const channelGroups = this.getSubscribedChannels();
const channels = this.getSubscribedChannels();
this.engine.transition(events.reconnect(timetoken, region));
if (this.dependencies.presenceReconnect)
this.dependencies.presenceReconnect({ channels, groups: channelGroups });
}
disconnect(isOffline = false) {
const channelGroups = this.getSubscribedChannels();
const channels = this.getSubscribedChannels();
this.engine.transition(events.disconnect(isOffline));
if (this.dependencies.presenceDisconnect)
this.dependencies.presenceDisconnect({ channels, groups: channelGroups, isOffline });
}
getSubscribedChannels() {
return Array.from(new Set(this.channels.slice(0)));
}
getSubscribedChannelGroups() {
return Array.from(new Set(this.groups.slice(0)));
}
dispose() {
this.disconnect(true);
this._unsubscribeEngine();
this.dispatcher.dispose();
}
}
exports.EventEngine = EventEngine;