AWS Amplify 在 Angular 应用中处理令牌刷新问题的实践指南

问题背景

在使用 AWS Amplify 与 Cognito 进行身份验证的 Angular 应用中，开发者可能会遇到一个间歇性问题：当应用界面长时间保持打开状态（通常超过1小时）后，调用 fetchAuthSession() 方法获取的会话对象中，tokens 属性偶尔会返回 undefined。这种情况通常发生在令牌过期或刷新过程中。

技术原理分析

AWS Amplify 的身份验证模块通过 Cognito 服务管理用户会话和令牌。默认情况下，Cognito 的 ID 令牌有效期为1小时，而刷新令牌的有效期通常配置为30天。当 ID 令牌过期时，应用需要利用刷新令牌获取新的 ID 令牌。

在令牌刷新过程中，可能会出现以下情况导致 tokens 为 undefined：

1. 多个并发 fetchAuthSession() 调用同时触发刷新流程
2. 持久化的刷新令牌暂时无法读取
3. 网络延迟导致刷新过程未及时完成
4. 会话状态在刷新过程中处于过渡状态

解决方案与实践

1. 升级 Amplify 版本

首先确保使用的是最新版本的 AWS Amplify（当前最新为6.13.0或更高）。早期版本（特别是6.12.2之前）存在并发刷新请求处理的问题。

升级步骤：

npm install aws-amplify@latest

2. 优化令牌获取逻辑

在 AuthService 中实现更健壮的令牌获取机制：

class AuthService {
  private session: any;

  async getAuthTokenWithRetry(retryCount = 3): Promise<string> {
    for (let i = 0; i < retryCount; i++) {
      try {
        this.session = await fetchAuthSession();
        const token = this.session?.tokens?.idToken?.toString();
        
        if (token) {
          return token;
        }
        
        // 等待短暂时间后重试
        await new Promise(resolve => setTimeout(resolve, 500));
      } catch (error) {
        console.error('获取令牌失败:', error);
      }
    }
    throw new Error('无法获取有效令牌');
  }
}

3. 强制刷新选项

在关键操作前，可以使用 forceRefresh 参数确保获取最新令牌：

await fetchAuthSession({ forceRefresh: true });

4. 会话状态监听

实现会话状态监听机制，在令牌即将过期时主动刷新：

import { Hub } from 'aws-amplify';

Hub.listen('auth', ({ payload }) => {
  switch (payload.event) {
    case 'tokenRefresh':
      console.log('令牌正在刷新');
      break;
    case 'tokenRefresh_failure':
      console.error('令牌刷新失败');
      break;
    case 'signIn':
    case 'signOut':
      // 处理登录状态变化
      break;
  }
});

最佳实践建议

1. 错误处理：所有令牌获取操作都应该有完善的错误处理和重试机制
2. 令牌缓存：合理缓存令牌，但要注意及时更新
3. 会话监控：实现会话状态监控，提前处理令牌过期情况
4. 性能考虑：避免不必要的令牌刷新操作
5. 离线处理：考虑网络不可用时的降级方案

总结

AWS Amplify 提供了强大的身份验证功能，但在处理令牌刷新等边缘场景时需要特别注意。通过升级到最新版本、优化令牌获取逻辑、实现强制刷新选项和会话状态监听，可以显著提高应用的稳定性和用户体验。开发者应当根据具体业务场景选择合适的解决方案，并始终考虑异常情况的处理。

记住，身份验证是应用安全的基础，任何相关修改都应该经过充分测试，确保不会引入新的安全问题。