Correct check for asigned enhanced consumer on the lease manager

lifion · Apr 29, 2019 · 83e5d1f · 83e5d1f
1 parent 91c2ebf
commit 83e5d1f
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/lib/lease-manager.js b/lib/lease-manager.js
@@ -120,7 +120,8 @@ class LeaseManager {
       logger,
       stateStore,
       streamName,
-      useAutoShardAssignment
+      useAutoShardAssignment,
+      useEnhancedFanOut
     } = options;
 
     Object.assign(internal(this), {
@@ -130,13 +131,14 @@ class LeaseManager {
       isStandalone: !useAutoShardAssignment,
       logger,
       stateStore,
-      streamName
+      streamName,
+      useEnhancedFanOut
     });
   }
 
   async start() {
     const privateProps = internal(this);
-    const { consumersManager, logger, stateStore } = privateProps;
+    const { consumersManager, logger, stateStore, useEnhancedFanOut } = privateProps;
 
     const acquireLeases = async () => {
       logger.debug('Trying to acquire leases…');
@@ -148,11 +150,13 @@ class LeaseManager {
         return;
       }
 
-      const consumerArn = await stateStore.getAssignedEnhancedConsumer();
-      if (!consumerArn) {
-        logger.debug("Can't acquire leases now as there's no assigned enhanced consumer.");
-        privateProps.timeoutId = setTimeout(acquireLeases, ACQUIRE_LEASES_INTERVAL);
-        return;
+      if (useEnhancedFanOut) {
+        const consumerArn = await stateStore.getAssignedEnhancedConsumer();
+        if (!consumerArn) {
+          logger.debug("Can't acquire leases now as there's no assigned enhanced consumer.");
+          privateProps.timeoutId = setTimeout(acquireLeases, ACQUIRE_LEASES_INTERVAL);
+          return;
+        }
       }
 
       const shards = await getStreamShards(privateProps);