refactor(catchError): simplify implementation

ReactiveX · Aug 26, 2020 · 5893948 · 5893948
1 parent 837ed8d
commit 5893948
Showing 1 changed file with 14 additions and 26 deletions.
diff --git a/src/internal/operators/catchError.ts b/src/internal/operators/catchError.ts
@@ -112,41 +112,29 @@ export function catchError<T, O extends ObservableInput<any>>(
     lift(source, function (this: Subscriber<T>, source: Observable<T>) {
       const subscriber = this;
       const subscription = new Subscription();
-      let innerSub: Subscription | null = null;
-      let syncUnsub = false;
-      let handledResult: Observable<ObservedValueOf<O>>;
+      let nextSource: Observable<T> | null = null;
+      let sourceSubscription: Subscription | null = null;
 
-      const handleError = (err: any) => {
-        try {
-          handledResult = from(selector(err, catchError(selector)(source)));
-        } catch (err) {
-          subscriber.error(err);
-          return;
+      const tryToSwapSubscription = () => {
+        if (sourceSubscription && nextSource && !subscriber.closed) {
+          subscription.add(nextSource.subscribe(subscriber));
+          sourceSubscription = nextSource = null;
         }
       };
 
-      innerSub = source.subscribe(
+      sourceSubscription = source.subscribe(
         new CatchErrorSubscriber(subscriber, (err) => {
-          handleError(err);
-          if (handledResult) {
-            if (innerSub) {
-              innerSub.unsubscribe();
-              innerSub = null;
-              subscription.add(handledResult.subscribe(subscriber));
-            } else {
-              syncUnsub = true;
-            }
+          try {
+            nextSource = from(selector(err, catchError(selector)(source)));
+            tryToSwapSubscription();
+          } catch (selectorErr) {
+            subscriber.error(selectorErr);
           }
         })
       );
 
-      if (syncUnsub) {
-        innerSub.unsubscribe();
-        innerSub = null;
-        subscription.add(handledResult!.subscribe(subscriber));
-      } else {
-        subscription.add(innerSub);
-      }
+      subscription.add(sourceSubscription);
+      tryToSwapSubscription();
 
       return subscription;
     });