【JavaScript・Node.js】エラーが発生したPromiseをリトライするソースコードを簡潔に書きたい

Question

非同期処理を実行し，エラーが発生した場合は最大３回までリトライを繰り返す，という動作を実現させたいと思い，以下のようなコードを実装しました．

こちらのコードで意図した動作は達成できるのですが，ネストが深く見辛く感じています．

// 一部省略

var requestPromise = require('request-promise');

var resp = await requestPromise.put(options).catch(async () => {
    return await requestPromise.put(options).catch(async () => {
        return await requestPromise.put(options).catch(async () => {
            return await requestPromise.put(options).catch((e) => {
                throw new Error("データ更新に失敗しました。 ErrorMessage : " + e.message);
            });
        });
    });
});

// 一部省略

動作内容を変えずに，簡潔に書き換えることは可能でしょうか？
詳しい方がおりましたらご教授いただきたいです．よろしくお願いいたします．

Answer 1

awaitはPromiseという箱をかぶった非同期の箱を取り出して、
値を取り出して同期に戻すという役割があります。

なので最終的にPromiseなら何でも問題ありません。
純正のPromise流儀の書き方を続けて、
最後にawaitで待つような作りにしましょう。

Promise流儀の書き方は、
完了を.thenか.catchのメソッドで待ち受けるという
一次元配列的な書き方が出来る事を目的としたやり方ですので、
下記のようなコードが成立します。

var resp = await requestPromise.put(options)
  .catch(() => requestPromise.put(options))
  .catch(() => requestPromise.put(options))
  .catch(() => requestPromise.put(options))
  .catch(e => Promise.reject(new Error("データ更新に失敗しました。 ErrorMessage : " + e.message)));

上記のコードはPromiseの中にPromiseという風に
2重のPromiseの箱で包まれていますが問題ありません。
.thenや.catchの戻り値でPromiseがネストしている場合、
それらを自動suna的に全て取り除いて値の状態にしてくれます。

なのでユーザーとしては気にせずPromiseを返すような処理を
がんがん.thenや.catchメソッドでつないで行けば良いのです。

---

さて、上記のコードはベースです。
Promise作って.catchメソッドをぶら下げるだけなので、
いろんな書き方に派生できますね。

同じ処理に着目して代入しながらfor文を使うことも可能です。

var putReq = requestPromise.put(options);
for (var i = 0; i < 3; i++) {
  // 毎回代入させることで1度ずつ挑戦を実現できる
  // 代入しない場合、最初の挑戦に失敗すると同時並行で3個のリクエスト処理が同時に飛んでしまう
  putReq = putReq.catch(() => requestPromise.put(options));
}
putReq.catch(e => Promise.reject(new Error("データ更新に失敗しました。 ErrorMessage : " + e.message)));
var resp = await putReq;

Lodaashなどのライブラリを使った関数型プログラミング的なアプローチならもう少し行数削れますね。
少々大道芸的ですが、reduce使ってワンライナーでやってみました。

// まず1度も成功していない想定でPromise.rejectを作ってから.catchを4回ぶら下げる
var resp = await [1, 2, 3, 4].reduce(
  (p) => p.catch(() => requestPromise.put(options)),
  Promise.reject()
).catch(e =>
  Promise.reject(new Error("データ更新に失敗しました。 ErrorMessage : " + e.message))
);

---

【おまけ】 素直にasync/await構文で頑張る

async/await構文を使ってコードを書くとこんな感じになりますね。
Promiseの糖衣構文という制約上、catchを握りつぶすのでちょっとお行儀が悪いですね。

var resp = null;
for (let i = 0; i < 4; i++) {
  try {
    resp = await requestPromise.put(options);
    break;
  } catch (e) {} // お行儀が悪いがエラーは握りつぶす
}
if (resp == null) {
  throw new Error("データ更新に失敗しました。 ErrorMessage : " + e.message)
}

やはり.cacheメソッドをぶら下げながら、
ある程度の所まではPromiseで引っ張った方が綺麗に書けるでしょうね。

Comments

[dal]

ご丁寧に回答いただき本当にありがとうございました！！