rxjs 自定義 operator

查看 rxjs 源碼,我們取一個比較簡單的操作符 every 作為例子。從 every 的實現(xiàn)可以看到,every 函數(shù)調用 source 的 lift 函數(shù),傳入自定義的兩個類 EveryOperator 和 EverySubscriber。接下來看代碼注釋。

// Observable.js
import { canReportError } from "./util/canReportError";
import { toSubscriber } from "./util/toSubscriber";
import { observable as Symbol_observable } from "./symbol/observable";
import { pipeFromArray } from "./util/pipe";
import { config } from "./config";
export class Observable {
  constructor(subscribe) {
    this._isScalar = false;
    if (subscribe) {
      this._subscribe = subscribe;
    }
  }
  // 返回一個新的Observable,將source置為自身,operator置為輸入的operator
  lift(operator) {
    const observable = new Observable();
    observable.source = this;
    observable.operator = operator;
    return observable;
  }
  subscribe(observerOrNext, error, complete) {
    const { operator } = this;
    const sink = toSubscriber(observerOrNext, error, complete);
    if (operator) {
      // 下面設置的EveryOperator call方法就是在這里使用
      // 這里傳入Subscriber 并且把自身的上一個Observable傳過去
      sink.add(operator.call(sink, this.source));
    } else {
      sink.add(
        this.source ||
          (config.useDeprecatedSynchronousErrorHandling &&
            !sink.syncErrorThrowable)
          ? this._subscribe(sink)
          : this._trySubscribe(sink)
      );
    }
    if (config.useDeprecatedSynchronousErrorHandling) {
      if (sink.syncErrorThrowable) {
        sink.syncErrorThrowable = false;
        if (sink.syncErrorThrown) {
          throw sink.syncErrorValue;
        }
      }
    }
    return sink;
  }
  ...
}

// every.js

import { Subscriber } from "../Subscriber";
export function every(predicate, thisArg) {
  // 返回一個生成新Observable(source為傳入?yún)?shù),operator為every)的函數(shù)
  return (source) => source.lift(new EveryOperator(predicate, thisArg, source));
}
class EveryOperator {
  constructor(predicate, thisArg, source) {
    this.predicate = predicate;
    this.thisArg = thisArg;
    this.source = source;
  }
  // 當擁有當前Operator的Observable被subscribe的時候 會調用call(subscriber, this.source)
  // observer subscribe函數(shù)傳入?yún)?shù)構建的
  // 這里的source是當前Observable的上一個Observable
  call(observer, source) {
    // 這里訂閱了上一個Observable
    return source.subscribe(
      // 這里實際上就會生成一個標準的Observer/Subscriber 有next error complete
      // 這里的this.source 跟source是同一個source(應該)
      new EverySubscriber(observer, this.predicate, this.thisArg, this.source)
    );
  }
}
// 繼承Subscriber
class EverySubscriber extends Subscriber {
  // destination 就是給我們調用next 傳入處理后的值
  // 從source進行訂閱
  constructor(destination, predicate, thisArg, source) {
    super(destination);
    this.predicate = predicate;
    this.thisArg = thisArg;
    this.source = source;
    this.index = 0;
    this.thisArg = thisArg || this;
  }
  notifyComplete(everyValueMatch) {
    // every的性質導致他只會調用一次next 就馬上到complete
    this.destination.next(everyValueMatch);
    this.destination.complete();
  }
  // 每次有值過來 就會調用到_next
  _next(value) {
    let result = false;
    try {
      result = this.predicate.call(
        this.thisArg,
        value,
        this.index++,
        this.source
      );
    } catch (err) {
      this.destination.error(err);
      return;
    }
    if (!result) {
      this.notifyComplete(false);
    }
  }
  // 上游complete 觸發(fā)_complete
  _complete() {
    this.notifyComplete(true);
  }
}
//# sourceMappingURL=every.js.map

當然,我們自己去封裝操作符的時候不一定要按照它的邏輯。我們通過上面代碼的分析,我們可以一個操作符函數(shù)為

const myoperator = (myargs) => (source) => new Observable();

其中,你要返回一個 Observable,在構造 Observable 時通過 subscribe 上一個 Observable 即 source,進行 next 向下游發(fā)出值。

這個 Observable 的構造方式可以是通過 Observable.prototype.lift (source.lift)構建,通過傳入一個帶有 call 方法的 Operator 類,而這個類的 call 方法返回一個 Subscription。

下面做一個簡單的 square 函數(shù)示例。

// 使用Observable.create
const square = () => (source) =>
  Observable.create((subscriber) => {
    const subscription = source.subscribe(
      (value) => {
        subscriber.next(value * value);
      },
      (err) => {
        subscriber.error(err);
      },
      () => {
        subscriber.complete();
      }
    );
    // 這里要返回subscription 參考TeardownLogic
    return subscription;
  });
// 使用lift
class SquareOperator {
  constructor(thisArg, source) {
    this.source = source;
  }
  call(subscriber, source) {
    return source.subscribe({
      next: (value) => {
        subscriber.next(value * value);
      },
      error: (error) => {
        subscriber.error(error);
      },
      complete: () => {
        subscriber.complete;
      },
    });
  }
}
const square = () => (source, thisArg) =>
  source.lift(new SquareOperator(thisArg, source));
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 230,501評論 6 544
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 99,673評論 3 429
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 178,610評論 0 383
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,939評論 1 318
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,668評論 6 412
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 56,004評論 1 329
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 44,001評論 3 449
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 43,173評論 0 290
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 49,705評論 1 336
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,426評論 3 359
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,656評論 1 374
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 39,139評論 5 364
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,833評論 3 350
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,247評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,580評論 1 295
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,371評論 3 400
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,621評論 2 380