如何在3次请求后中断此Javascript HTTP无限请求循环？

Posted 2021-04-03

所以我有2个组件，Token.jsx和httpRequest.jsx。当令牌到期时它们都互相调用并进入无限循环，我想在3个http错误之后打破循环。

export default function GenerateToken() {
  return new Promise(((resolve) => {
    const refreshToken = GetRefreshToken();
    const url = `${cloudFunctionURL}/users/auth/idtoken/refresh`;
    const headers = {
      'Content-Type': 'application/json',
    };
    const params = {
      refreshToken,
    };
    httpRequest.makeHTTPCall('post', url, headers, params).then((tokenObject) => {
      // Storing the idToken in localstorage
      // reactLocalStorage.set('PL_IdToken', tokenObject.idToken);
      StoreIdToken(`Id ${tokenObject.id_token}`);
      StoreUserId(tokenObject.user_id);
      StoreRefreshToken(tokenObject.refresh_token);
      resolve(tokenObject.id_token);
    });
  }));
}

//第二个文件

function makeHTTPCall(method, url, headers, params = null) {
  return new Promise((resolve, reject) => {
    headers.Authorization = GetIdToken();
    headers['Content-Type'] = 'application/json';
    qwest.setDefaultDataType('json');
    qwest.setDefaultOptions({
      headers,
    });

    // Make http request
    qwest[`${method}`](url, params)
      .then((xhr, response) => {
        resolve(response);
      })
      .catch((error, xhr) => {
        if (xhr.status === 401) { // IdToken expired
          GenerateToken().then(() => {
            resolve(GET(url));
          });
        }
        else {
          reject(error); // error
        }
      });
  });
}

答案

你可以给它们两个像tryCount这样的参数，并在其中一个函数中递增它，一旦它达到2就拒绝它：

export default function GenerateToken(tryCount = 0) {
  return new Promise((resolve) => {
    const refreshToken = GetRefreshToken();
    const url = `${cloudFunctionURL}/users/auth/idtoken/refresh`;
    const headers = {
      'Content-Type': 'application/json',
    };
    const params = {
      refreshToken,
    };
    httpRequest.makeHTTPCall('post', url, headers, params, tryCount).then((tokenObject) => {
      // Storing the idToken in localstorage
      // reactLocalStorage.set('PL_IdToken', tokenObject.idToken);
      StoreIdToken(`Id ${tokenObject.id_token}`);
      StoreUserId(tokenObject.user_id);
      StoreRefreshToken(tokenObject.refresh_token);
      resolve(tokenObject.id_token);
    });
  });
}
// 2nd File

function makeHTTPCall(method, url, headers, params, tryCount) {
  return new Promise((resolve, reject) => {
    headers.Authorization = GetIdToken();
    headers['Content-Type'] = 'application/json';
    qwest.setDefaultDataType('json');
    qwest.setDefaultOptions({
      headers,
    });

    // Make http request
    qwest[`${method}`](url, params)
      .then((xhr, response) => {
        resolve(response);
      })
      .catch((error, xhr) => {
        if (xhr.status === 401 && tryCount <= 1) { // IdToken expired, and we've recursed less than twice so far
          GenerateToken(tryCount + 1).then(() => {
            resolve(GET(url));
          });
        } else {
          reject(error); // error
        }
      });
  });
}

另请注意，您应该避免使用explicit Promise construction antipattern：

export default function GenerateToken(tryCount = 0) {
  const refreshToken = GetRefreshToken();
  const url = `${cloudFunctionURL}/users/auth/idtoken/refresh`;
  const headers = {
    'Content-Type': 'application/json',
  };
  const params = {
    refreshToken,
  };
  return httpRequest.makeHTTPCall('post', url, headers, params, tryCount).then((tokenObject) => {
    // Storing the idToken in localstorage
    // reactLocalStorage.set('PL_IdToken', tokenObject.idToken);
    StoreIdToken(`Id ${tokenObject.id_token}`);
    StoreUserId(tokenObject.user_id);
    StoreRefreshToken(tokenObject.refresh_token);
    return tokenObject.id_token;
  });
}
// 2nd File

function makeHTTPCall(method, url, headers, params, tryCount) {
  headers.Authorization = GetIdToken();
  headers['Content-Type'] = 'application/json';
  qwest.setDefaultDataType('json');
  qwest.setDefaultOptions({
    headers,
  });

  // Make http request
  return qwest[`${method}`](url, params)
    .then((xhr, response) => {
      return response;
    })
    .catch((error, xhr) => {
    if (xhr.status === 401 && tryCount <= 1) { // IdToken expired, and we've recursed less than twice so far
      return GenerateToken(tryCount + 1).then(() => {
        return GET(url);
      });
    } else {
      throw new Error(error);
    }
  });
}