IfObservable.ts 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. import { Observable, SubscribableOrPromise } from '../Observable';
  2. import { Subscriber } from '../Subscriber';
  3. import { TeardownLogic } from '../Subscription';
  4. import { subscribeToResult } from '../util/subscribeToResult';
  5. import { OuterSubscriber } from '../OuterSubscriber';
  6. /**
  7. * We need this JSDoc comment for affecting ESDoc.
  8. * @extends {Ignored}
  9. * @hide true
  10. */
  11. export class IfObservable<T, R> extends Observable<T> {
  12. static create<T, R>(condition: () => boolean | void,
  13. thenSource?: SubscribableOrPromise<T> | void,
  14. elseSource?: SubscribableOrPromise<R> | void): Observable<T|R> {
  15. return new IfObservable(condition, thenSource, elseSource);
  16. }
  17. constructor(private condition: () => boolean | void,
  18. private thenSource?: SubscribableOrPromise<T> | void,
  19. private elseSource?: SubscribableOrPromise<R> | void) {
  20. super();
  21. }
  22. /** @deprecated internal use only */ _subscribe(subscriber: Subscriber<T|R>): TeardownLogic {
  23. const { condition, thenSource, elseSource } = this;
  24. return new IfSubscriber(subscriber, condition, thenSource, elseSource);
  25. }
  26. }
  27. class IfSubscriber<T, R> extends OuterSubscriber<T, T> {
  28. constructor(destination: Subscriber<T>,
  29. private condition: () => boolean | void,
  30. private thenSource?: SubscribableOrPromise<T> | void,
  31. private elseSource?: SubscribableOrPromise<R> | void) {
  32. super(destination);
  33. this.tryIf();
  34. }
  35. private tryIf(): void {
  36. const { condition, thenSource, elseSource } = this;
  37. let result: boolean;
  38. try {
  39. result = <boolean>condition();
  40. const source = result ? thenSource : elseSource;
  41. if (source) {
  42. this.add(subscribeToResult(this, source));
  43. } else {
  44. this._complete();
  45. }
  46. } catch (err) {
  47. this._error(err);
  48. }
  49. }
  50. }