feat: add retry for articles

cooderl · Mar 26, 2024 · 4e866c6 · 4e866c6
1 parent 4dd115a
commit 4e866c6
Showing 1 changed file with 32 additions and 22 deletions.
diff --git a/apps/server/src/trpc/trpc.service.ts b/apps/server/src/trpc/trpc.service.ts
@@ -75,8 +75,8 @@ export class TrpcService {
         } else if (errMsg.includes('WeReadError400')) {
           this.logger.error(`账号（${id}）处理请求参数出错`);
           this.logger.error('WeReadError400: ', errMsg);
-          // 30s 后重试
-          await new Promise((resolve) => setTimeout(resolve, 30 * 1e3));
+          // 10s 后重试
+          await new Promise((resolve) => setTimeout(resolve, 10 * 1e3));
         } else {
           this.logger.error("Can't handle this error: ", errMsg);
         }
@@ -125,28 +125,38 @@ export class TrpcService {
     return account;
   }
 
-  async getMpArticles(mpId: string) {
+  async getMpArticles(mpId: string, retryCount = 3) {
     const account = await this.getAvailableAccount();
 
-    return this.request
-      .get<
-        {
-          id: string;
-          title: string;
-          picUrl: string;
-          publishTime: number;
-        }[]
-      >(`/api/v2/platform/mps/${mpId}/articles`, {
-        headers: {
-          xid: account.id,
-          Authorization: `Bearer ${account.token}`,
-        },
-      })
-      .then((res) => res.data)
-      .then((res) => {
-        this.logger.log(`getMpArticles(${mpId}): ${res.length} articles`);
-        return res;
-      });
+    try {
+      const res = await this.request
+        .get<
+          {
+            id: string;
+            title: string;
+            picUrl: string;
+            publishTime: number;
+          }[]
+        >(`/api/v2/platform/mps/${mpId}/articles`, {
+          headers: {
+            xid: account.id,
+            Authorization: `Bearer ${account.token}`,
+          },
+        })
+        .then((res) => res.data)
+        .then((res) => {
+          this.logger.log(`getMpArticles(${mpId}): ${res.length} articles`);
+          return res;
+        });
+      return res;
+    } catch (err) {
+      this.logger.error(`retry(${4 - retryCount}) getMpArticles  error: `, err);
+      if (retryCount > 0) {
+        return this.getMpArticles(mpId, retryCount - 1);
+      } else {
+        throw err;
+      }
+    }
   }
 
   async refreshMpArticlesAndUpdateFeed(mpId: string) {