bpmn-elements 6.0.1 → 7.0.0

package/dist/src/eventDefinitions/CompensateEventDefinition.js

@@ -3,186 +3,219 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.default = CompensationEventDefinition;
+exports.default = CompensateEventDefinition;
 
 var _shared = require("../shared");
 
 var _messageHelper = require("../messageHelper");
 
-function CompensationEventDefinition(activity, eventDefinition, context) {
+const completedSymbol = Symbol.for('completed');
+const executeMessageSymbol = Symbol.for('executeMessage');
+const messageQSymbol = Symbol.for('messageQ');
+const compensateQSymbol = Symbol.for('compensateQ');
+const associationsSymbol = Symbol.for('associations');
+
+function CompensateEventDefinition(activity, eventDefinition, context) {
   const {
     id,
     broker,
     environment,
     isThrowing
   } = activity;
-  const {
-    type
-  } = eventDefinition;
-  const {
-    debug
-  } = environment.Logger(type.toLowerCase());
-  const compensationQueueName = `compensate-${(0, _shared.brokerSafeId)(id)}-q`;
-  const associations = context.getOutboundAssociations(id) || [];
-  if (!isThrowing) setupCatch();
-  const source = {
-    id,
-    type,
-    reference: {
-      referenceType: 'compensate'
-    },
-    execute: isThrowing ? executeThrow : executeCatch
+  this.id = id;
+  const type = this.type = eventDefinition.type;
+  const reference = this.reference = {
+    referenceType: 'compensate'
   };
-  return source;
-
-  function executeCatch(executeMessage) {
-    let completed;
-    const messageContent = (0, _messageHelper.cloneContent)(executeMessage.content);
-    const {
-      executionId,
-      parent
-    } = messageContent;
-    const parentExecutionId = parent && parent.executionId;
-    broker.consume(compensationQueueName, onCompensateApiMessage, {
-      noAck: true,
-      consumerTag: `_oncompensate-${executionId}`
-    });
-    if (completed) return;
-    broker.subscribeTmp('api', `activity.#.${executionId}`, onApiMessage, {
-      noAck: true,
-      consumerTag: `_api-${executionId}`
+  this.isThrowing = isThrowing;
+  this.activity = activity;
+  this.broker = broker;
+  this.logger = environment.Logger(type.toLowerCase());
+
+  if (!isThrowing) {
+    this[completedSymbol] = false;
+    this[associationsSymbol] = context.getOutboundAssociations(id) || [];
+    const messageQueueName = `${reference.referenceType}-${(0, _shared.brokerSafeId)(id)}-q`;
+    this[messageQSymbol] = broker.assertQueue(messageQueueName, {
+      autoDelete: false,
+      durable: true
     });
-    if (completed) return stop();
-    debug(`<${executionId} (${id})> expect compensate`);
-    broker.assertExchange('compensate', 'topic');
-    const compensateQ = broker.assertQueue('compensate-q', {
+    broker.bindQueue(messageQueueName, 'api', `*.${reference.referenceType}.#`, {
       durable: true,
-      autoDelete: false
-    });
-    broker.subscribeTmp('compensate', 'execute.#', onCollect, {
-      noAck: true,
-      consumerTag: '_oncollect-messages'
-    });
-    broker.publish('execution', 'execute.detach', (0, _messageHelper.cloneContent)(messageContent, {
-      bindExchange: 'compensate'
-    }));
-    broker.publish('event', 'activity.detach', { ...messageContent,
-      executionId: parentExecutionId,
-      parent: (0, _messageHelper.shiftParent)(parent),
-      bindExchange: 'compensate'
+      priority: 400
     });
+  }
+}
 
-    function onCollect(routingKey, message) {
-      switch (routingKey) {
-        case 'execute.error':
-        case 'execute.completed':
-          {
-            return compensateQ.queueMessage(message.fields, (0, _messageHelper.cloneContent)(message.content), message.properties);
-          }
-      }
-    }
-
-    function onCompensateApiMessage(routingKey, message) {
-      const output = message.content.message;
-      completed = true;
-      stop();
-      debug(`<${executionId} (${id})> caught compensate event`);
-      broker.publish('event', 'activity.catch', { ...messageContent,
-        message: { ...output
-        },
-        executionId: parentExecutionId,
-        parent: (0, _messageHelper.shiftParent)(executeMessage.content.parent)
-      }, {
-        type: 'catch'
-      });
-      compensateQ.on('depleted', onDepleted);
-      compensateQ.consume(onCollected, {
-        noAck: true,
-        consumerTag: '_convey-messages'
-      });
-      associations.forEach(association => {
-        association.complete((0, _messageHelper.cloneMessage)(message));
-      });
-
-      function onDepleted() {
-        compensateQ.off('depleted', onDepleted);
-        return broker.publish('execution', 'execute.completed', { ...messageContent,
-          output,
-          state: 'catch'
-        });
-      }
-    }
-
-    function onCollected(routingKey, message) {
-      associations.forEach(association => {
-        association.take((0, _messageHelper.cloneMessage)(message));
-      });
-    }
-
-    function onApiMessage(routingKey, message) {
-      const messageType = message.properties.type;
-
-      switch (messageType) {
-        case 'compensate':
-          {
-            return onCompensateApiMessage(routingKey, message);
-          }
-
-        case 'discard':
-          {
-            completed = true;
-            stop();
-            associations.forEach(association => {
-              association.discard((0, _messageHelper.cloneMessage)(message));
-            });
-            return broker.publish('execution', 'execute.discard', { ...messageContent
-            });
-          }
-
-        case 'stop':
-          {
-            stop();
-            break;
-          }
+const proto = CompensateEventDefinition.prototype;
+Object.defineProperty(proto, 'executionId', {
+  get() {
+    const message = this[executeMessageSymbol];
+    return message && message.content.executionId;
+  }
+
+});
+
+proto.execute = function execute(executeMessage) {
+  return this.isThrowing ? this.executeThrow(executeMessage) : this.executeCatch(executeMessage);
+};
+
+proto.executeCatch = function executeCatch(executeMessage) {
+  this[executeMessageSymbol] = executeMessage;
+  this[completedSymbol] = false;
+  const executeContent = executeMessage.content;
+  const {
+    executionId,
+    parent
+  } = executeContent;
+
+  this._debug('expect compensate');
+
+  const broker = this.broker;
+  broker.assertExchange('compensate', 'topic');
+  this[compensateQSymbol] = broker.assertQueue('compensate-q', {
+    durable: true,
+    autoDelete: false
+  });
+  broker.subscribeTmp('compensate', 'execute.#', this._onCollect.bind(this), {
+    noAck: true,
+    consumerTag: '_oncollect-messages'
+  });
+  broker.publish('execution', 'execute.detach', (0, _messageHelper.cloneContent)(executeContent, {
+    sourceExchange: 'execution',
+    bindExchange: 'compensate'
+  }));
+  this[messageQSymbol].consume(this._onCompensateApiMessage.bind(this), {
+    noAck: true,
+    consumerTag: `_oncompensate-${executionId}`
+  });
+  if (this[completedSymbol]) return;
+
+  const onApiMessage = this._onApiMessage.bind(this);
+
+  broker.subscribeTmp('api', `activity.#.${executionId}`, onApiMessage, {
+    noAck: true,
+    consumerTag: `_api-${executionId}`
+  });
+  const detachContent = (0, _messageHelper.cloneContent)(executeContent, {
+    executionId: parent.executionId,
+    bindExchange: 'compensate'
+  });
+  detachContent.parent = (0, _messageHelper.shiftParent)(parent);
+  broker.publish('event', 'activity.detach', detachContent);
+};
+
+proto.executeThrow = function executeThrow(executeMessage) {
+  const executeContent = executeMessage.content;
+  const {
+    executionId,
+    parent
+  } = executeContent;
+  const parentExecutionId = parent && parent.executionId;
+  this.logger.debug(`<${executionId} (${this.activity.id})> throw compensate`);
+  const broker = this.broker;
+  const throwContent = (0, _messageHelper.cloneContent)(executeContent, {
+    executionId: parentExecutionId,
+    state: 'throw'
+  });
+  throwContent.parent = (0, _messageHelper.shiftParent)(parent);
+  broker.publish('event', 'activity.compensate', throwContent, {
+    type: 'compensate',
+    delegate: true
+  });
+  return broker.publish('execution', 'execute.completed', (0, _messageHelper.cloneContent)(executeContent));
+};
+
+proto._onCollect = function onCollect(routingKey, message) {
+  switch (routingKey) {
+    case 'execute.error':
+    case 'execute.completed':
+      {
+        return this[compensateQSymbol].queueMessage(message.fields, (0, _messageHelper.cloneContent)(message.content), message.properties);
       }
-    }
-
-    function stop() {
-      broker.cancel(`_api-${executionId}`);
-      broker.cancel(`_oncompensate-${executionId}`);
-      broker.cancel('_oncollect-messages');
-      broker.cancel('_convey-messages');
-    }
   }
+};
 
-  function executeThrow(executeMessage) {
-    const messageContent = (0, _messageHelper.cloneContent)(executeMessage.content);
-    const {
-      executionId,
-      parent
-    } = messageContent;
-    const parentExecutionId = parent && parent.executionId;
-    debug(`<${executionId} (${id})> throw compensate`);
-    broker.publish('event', 'activity.compensate', { ...(0, _messageHelper.cloneContent)(messageContent),
-      executionId: parentExecutionId,
-      parent: (0, _messageHelper.shiftParent)(parent),
-      state: 'throw'
-    }, {
-      type: 'compensate',
-      delegate: true
-    });
-    return broker.publish('execution', 'execute.completed', { ...messageContent
-    });
+proto._onCompensateApiMessage = function onCompensateApiMessage(routingKey, message) {
+  const output = message.content.message;
+  this[completedSymbol] = true;
+
+  this._stop();
+
+  this._debug('caught compensate event');
+
+  const broker = this.broker;
+  const executeContent = this[executeMessageSymbol].content;
+  const catchContent = (0, _messageHelper.cloneContent)(executeContent, {
+    message: { ...output
+    },
+    executionId: executeContent.parent.executionId
+  });
+  catchContent.parent = (0, _messageHelper.shiftParent)(catchContent.parent);
+  broker.publish('event', 'activity.catch', catchContent, {
+    type: 'catch'
+  });
+  const compensateQ = this[compensateQSymbol];
+  compensateQ.on('depleted', onDepleted);
+  compensateQ.consume(this._onCollected.bind(this), {
+    noAck: true,
+    consumerTag: '_convey-messages'
+  });
+
+  for (const association of this[associationsSymbol]) association.complete((0, _messageHelper.cloneMessage)(message));
+
+  function onDepleted() {
+    compensateQ.off('depleted', onDepleted);
+    return broker.publish('execution', 'execute.completed', (0, _messageHelper.cloneContent)(executeContent, {
+      output,
+      state: 'catch'
+    }));
   }
+};
 
-  function setupCatch() {
-    broker.assertQueue(compensationQueueName, {
-      autoDelete: false,
-      durable: true
-    });
-    broker.bindQueue(compensationQueueName, 'api', '*.compensate.#', {
-      durable: true,
-      priority: 400
-    });
+proto._onCollected = function onCollected(routingKey, message) {
+  for (const association of this[associationsSymbol]) association.take((0, _messageHelper.cloneMessage)(message));
+};
+
+proto._onApiMessage = function onApiMessage(routingKey, message) {
+  const messageType = message.properties.type;
+
+  switch (messageType) {
+    case 'compensate':
+      {
+        return this._onCompensateApiMessage(routingKey, message);
+      }
+
+    case 'discard':
+      {
+        this[completedSymbol] = true;
+
+        this._stop();
+
+        for (const association of this[associationsSymbol]) association.discard((0, _messageHelper.cloneMessage)(message));
+
+        return this.broker.publish('execution', 'execute.discard', (0, _messageHelper.cloneContent)(this[executeMessageSymbol].content));
+      }
+
+    case 'stop':
+      {
+        this._stop();
+
+        break;
+      }
   }
-}
+};
+
+proto._stop = function stop() {
+  const broker = this.broker,
+        executionId = this.executionId;
+  broker.cancel(`_api-${executionId}`);
+  broker.cancel(`_oncompensate-${executionId}`);
+  broker.cancel('_oncollect-messages');
+  broker.cancel('_convey-messages');
+  this[messageQSymbol].purge();
+};
+
+proto._debug = function debug(msg) {
+  this.logger.debug(`<${this.executionId} (${this.activity.id})> ${msg}`);
+};