UXの高い「いいね」ボタンをRxJSで実装する

ちきさんです。

[榊原氏](https://twitter.com/rdlabo)からヒントをもらってSNSアプリとかでよくある「いいね」ボタンをRxJS(Angular)で実現するコードを書いたので共有します。

## 概要

ユーザーの操作に素早く応答する画面を作りつつ、サーバーとクライアント間で状態の整合性を保つ。

## 特徴

- サーバーから初期状態を受け取るまでは「いいね」ボタンを押しても何もリクエストを飛ばさない。
- 初期状態を受け取った後はクライアントで状態を管理し、ユーザーの操作に応じてすぐに画面を更新する。
- 画面の更新をしてからバックグラウンドでサーバーの状態を更新する。(Lazy Sync)

つまりサーバーから初期状態を取得したらあとはクライアントが状態管理の主導権を握るということ。

デモ → https://stackblitz.com/edit/angular-qwfwkm

---

## コード解説

### サービス

サーバー側の状態管理をまとめています。本質ではない部分は省略しました。

```ts:FakeStateService.ts
export interface FakeState {
  iine: boolean
}

@Injectable()
export class FakeStateService {
  private state: FakeState
  private requester$ = new Subject<boolean>()
  private dispatcher$ = new Subject<void>()
  private provider$ = new ReplaySubject<FakeState>(1)

  constructor() {
    this.state = {
      iine: Math.random() > 0.5
    }

    // サーバーにPOSTリクエストを送る。
    this.requester$
      .concatMap(iine => this.httpPostRequest(iine))
      .subscribe(() => {
        this.dispatcher$.next()
      })

    // dispatcherがnextされる度にサーバーから状態を取得し直す。
    Observable.merge(this.dispatcher$)
      .switchMap(() => this.httpGetRequest())
      .subscribe(state => {
        this.provider$.next(state)
      })

    this.dispatcher$.next()
  }

  private httpGetRequest(): Promise<FakeState> { /* 省略 */ }

  private httpPostRequest(iine: boolean): Promise<void> { /* 省略 */ }

  getState$(): Observable<FakeState> {
    return this.provider$;
  }

  updateIine(iine: boolean): void {
    this.requester$.next(iine)
  }
}
```

「いいね」の状態を更新する流れは下記のようになります。

コンポーネントから`updateIine()`を呼び出されます。

```ts
updateIine(iine: boolean): void {
  this.requester$.next(iine)
}
```

`reqester$`のストリームが`httpPostRequest()`を呼び出すことでサーバーの状態を更新します。

```ts
this.requester$
  .concatMap(iine => this.httpPostRequest(iine))
  .subscribe(() => {
    this.dispatcher$.next()
  })
```

`concatMap`オペレーターによりアクションの順番が担保されているので「いいね」ボタンを連打されても状態の矛盾が生じないようになっています。(参照: [RxJSのconcatMap, mergeMap, switchMapの違いを理解する](https://qiita.com/ovrmrw/items/b45d7bf29c8d29415bd7))

`dispatcher$`のストリームを合成したストリームが`httpGetRequest()`を呼び出すことでサーバーの状態を取得します。

```ts
Observable.merge(this.dispatcher$)
  .switchMap(() => this.httpGetRequest())
  .subscribe(state => {
    this.provider$.next(state)
  })
```

わざわざ`merge`オペレーターを使っているのは、今後`dispatcher$`が複数になってもコードを書き換えなくていいようにするためです。
`switchMap`オペレーターによりストリームは適切にキャンセルされます。(参照: [RxJSのconcatMap, mergeMap, switchMapの違いを理解する](https://qiita.com/ovrmrw/items/b45d7bf29c8d29415bd7))

`provider$`のストリームは`getState$()`を通じてコンポーネントが取得できるようにします。

```ts
getState$(): Observable<FakeState> {
  return this.provider$;
}
```

### コンポーネント

クライアント側の状態管理をまとめています。本質ではない部分は省略しました。

```ts:AppComponent.ts
@Component({ /* 省略 */ })
export class AppComponent {
  private isLoaded = false
  state: FakeState | null = null

  constructor(private fakeStateService: FakeStateService) {
    // サーバーから状態を取得する。これ以後はクライアントの状態を優先してViewに反映させる。
    this.fakeStateService.getState$()
      .take(1)
      .subscribe(state => {
        this.state = state
        this.isLoaded = true
      })
  }

  get serverState$(): Observable<FakeState> {
    return this.fakeStateService.getState$()
  }

  updateIine(): void {
    Observable.of(this.state)
      .takeWhile(() => this.isLoaded)
      .map(state => !state.iine)
      .subscribe(iine => {
        this.updateComponentState({ iine })
        this.fakeStateService.updateIine(iine) // サーバーの状態変更は裏で処理される。
      })
  }

  private updateComponentState(state: Partial<FakeState>): void {
    this.state = { ...this.state, ...state }
  }
}
```

コンポーネント初期化時に、一度だけサーバーの状態を取得します。
このとき`isLoaded`フラグを`true`にしているのがコツです。

```ts
    this.fakeStateService.getState$()
      .take(1)
      .subscribe(state => {
        this.state = state
        this.isLoaded = true
      })
```

「いいね」ボタンを押されたら、`updateIine()`を呼び出すことでクライアントの状態を更新してすぐに画面に反映させます。そのあと遅れてサーバーの状態を更新します。
*なのでサーバーの状態が正しく更新されないときがあるかもしれませんが、そこを厳密にやるべきかどうかは要件次第でしょう。*

```ts
  updateIine(): void {
    Observable.of(this.state)
      .takeWhile(() => this.isLoaded)
      .map(state => !state.iine)
      .subscribe(iine => {
        this.updateComponentState({ iine })
        this.fakeStateService.updateIine(iine)
      })
  }
```

`takeWhile`オペレーターによって`isLoaded`フラグが`true`のときだけストリームが流れるようになっています。
サーバーから「いいね」の状態を取得していないのに「いいね」ボタンを押されて更新リクエストを飛ばしてしまうよりよっぽどいいですね。

その後は`serverState$()`プロパティを通じて更新後のサーバーの状態を取得しているので、クライアントとサーバー間で状態の矛盾が生じないことをデモで確認していだけると思います。

```ts
  get serverState$(): Observable<FakeState> {
    return this.fakeStateService.getState$()
  }
```

---

RxJSをちょっと使うだけでユーザーにとって直感的なUIを簡単に構築できますね。ではまた。

P.S.
似たようなことを以前書いたことがあるなと思ったらこれでした。 → [Firebase-as-a-Store ～RxJSで作るFirebaseバックエンドのRedux～](https://qiita.com/ovrmrw/items/27d06475f405fd4ca9b4)