feat: refactor the cache

2022-03-04 10:15:30 +08:00
parent b6b0a0c5ef
commit acd007fb71
28 changed files with 622 additions and 341 deletions
@@ -1,73 +1,79 @@
-import { fetchBaseQuery } from '@reduxjs/toolkit/query';
-import toast from 'react-hot-toast';
-import { updateToken, clearAuthData } from '../slices/auth.data';
-import BASE_URL, { tokenHeader } from '../config';
-const whiteList = ['login', 'checkInviteTokenValid'];
+import { fetchBaseQuery } from "@reduxjs/toolkit/query";
+import toast from "react-hot-toast";
+import { updateToken, clearAuthData } from "../slices/auth.data";
+import BASE_URL, { tokenHeader } from "../config";
+const whiteList = ["login", "checkInviteTokenValid", "getServer", "getOpenid"];
 const baseQuery = fetchBaseQuery({
- baseUrl: BASE_URL,
- prepareHeaders: (headers, { getState, endpoint }) => {
-  console.log('req', endpoint);
-  const { token } = getState().authData;
-  if (token && !whiteList.includes(endpoint)) {
-   headers.set(tokenHeader, token);
-  }
-  // 发送channel msg (临时举措)
-  // if (endpoint == "sendChannelMsg") {
-  //   headers.set("Content-Type", "text/plain");
-  // }
-  return headers;
- }
+  baseUrl: BASE_URL,
+  prepareHeaders: (headers, { getState, endpoint }) => {
+    console.log("req", endpoint);
+    const { token } = getState().authData;
+    if (token && !whiteList.includes(endpoint)) {
+      headers.set(tokenHeader, token);
+    }
+    // 发送channel msg (临时举措)
+    // if (endpoint == "sendChannelMsg") {
+    //   headers.set("Content-Type", "text/plain");
+    // }
+    return headers;
+  },
 });
 const baseQueryWithReauth = async (args, api, extraOptions) => {
- let result = await baseQuery(args, api, extraOptions);
- if (result.error) {
-  console.log('api error', result.error);
-  switch (result.error.originalStatus) {
-   case 404:
-    {
-     toast.error('Request Not Found');
-    }
-    break;
-   case 500:
-    {
-     toast.error(result.error.data || 'server error');
-    }
-    break;
-   case 401:
-    {
-     // try to get a new token with refreshToken
-     const { token, refreshToken } = api.getState().authData;
-     const refreshResult = await baseQuery(
-      {
-       url: '/token/renew',
-       method: 'POST',
-       body: {
-        token,
-        refresh_token: refreshToken
-       }
-      },
-      api,
-      extraOptions
-     );
-     console.log({ refreshResult });
-     if (refreshResult.data) {
-      // store the new token
-      api.dispatch(updateToken(refreshResult.data));
-      // retry the initial query
-      result = await baseQuery(args, api, extraOptions);
-     } else {
-      toast.error('token expired, please login again');
-      api.dispatch(clearAuthData());
-     }
-    }
+  let result = await baseQuery(args, api, extraOptions);
+  if (result?.error) {
+    console.log("api error", result.error.originalStatus, api.endpoint);
+    switch (result.error.originalStatus || result.error.status) {
+      case 404:
+        {
+          toast.error("Request Not Found");
+        }
+        break;
+      case 500:
+        {
+          toast.error(result.error.data || "server error");
+        }
+        break;
+      case 401:
+        {
+          if (api.endpoint === "renew") {
+            toast.error("token expired, please login again");
+            api.dispatch(clearAuthData());
+            location.href = "/#/login";
+            return;
+          }
+          // try to get a new token with refreshToken
+          const { token, refreshToken } = api.getState().authData;
+          const refreshResult = await baseQuery(
+            {
+              url: "/token/renew",
+              method: "POST",
+              body: {
+                token,
+                refresh_token: refreshToken,
+              },
+            },
+            api,
+            extraOptions
+          );
+          console.log({ refreshResult });
+          if (refreshResult.data) {
+            // store the new token
+            api.dispatch(updateToken(refreshResult.data));
+            // retry the initial query
+            result = await baseQuery(args, api, extraOptions);
+          } else {
+            toast.error("token expired, please login again");
+            api.dispatch(clearAuthData());
+          }
+        }

-    break;
+        break;

-   default:
-    break;
+      default:
+        break;
+    }
  }
- }
- return result;
+  return result;
 };

 export default baseQueryWithReauth;