AWS Amplify 在 Angular 应用中处理令牌刷新时出现未定义令牌的解决方案

问题背景

在使用 AWS Amplify 与 Cognito 服务集成开发 Angular 应用时，开发者可能会遇到一个间歇性的身份验证问题：当应用界面长时间保持打开状态（通常超过1小时，即默认ID令牌的有效期）后，调用 fetchAuthSession() 方法获取会话时，返回的 tokens 属性偶尔会出现 undefined 的情况。

问题分析

这个问题通常出现在以下场景中：

1. 应用保持打开状态超过ID令牌的默认有效期（1小时）
2. 系统尝试自动刷新令牌时
3. 多个并发请求同时触发 fetchAuthSession() 调用

经过深入分析，这可能是由于以下原因导致的：

• 持久化的刷新令牌在某些情况下变得不可读
• 令牌刷新过程中出现了短暂的会话不完整状态
• 多个并发请求导致令牌刷新逻辑出现竞争条件

解决方案

1. 升级 AWS Amplify 版本

从问题追踪来看，AWS Amplify 团队在 v6.12.2 及更高版本中已经修复了多个并发 fetchAuthSession() 调用导致的问题。建议开发者将项目升级到最新稳定版本（当前为6.13.0或更高）。

升级步骤：

1. 更新 package.json 中的 aws-amplify 依赖版本
2. 删除 node_modules 目录
3. 删除 package-lock.json 文件
4. 重新运行 npm install

2. 优化令牌获取逻辑

在 AuthService 中实现更健壮的令牌获取逻辑：

async getAuthTokenWithRetry(maxRetries = 3, delay = 1000): Promise<string> {
    let attempts = 0;
    
    while (attempts < maxRetries) {
        try {
            const session = await fetchAuthSession({ forceRefresh: true });
            const token = session?.tokens?.idToken?.toString();
            
            if (token) {
                return token;
            }
            
            attempts++;
            if (attempts < maxRetries) {
                await new Promise(resolve => setTimeout(resolve, delay));
            }
        } catch (error) {
            console.error('获取令牌失败:', error);
            attempts++;
            if (attempts < maxRetries) {
                await new Promise(resolve => setTimeout(resolve, delay));
            }
        }
    }
    
    throw new Error('无法获取有效令牌');
}

3. 使用 forceRefresh 参数

在关键操作前，可以强制刷新令牌以确保获取最新的会话信息：

const freshSession = await fetchAuthSession({ forceRefresh: true });

4. 实现令牌状态监控

可以创建一个令牌健康检查服务，定期验证令牌有效性：

@Injectable({ providedIn: 'root' })
export class TokenHealthService {
    private checkInterval = 300000; // 5分钟
    
    constructor() {
        setInterval(() => this.checkTokenHealth(), this.checkInterval);
    }
    
    private async checkTokenHealth(): Promise<void> {
        try {
            const session = await fetchAuthSession();
            if (!session.tokens) {
                console.warn('令牌健康检查失败，尝试刷新...');
                await fetchAuthSession({ forceRefresh: true });
            }
        } catch (error) {
            console.error('令牌健康检查异常:', error);
        }
    }
}

最佳实践建议

1. 错误处理：对所有身份验证操作实现全面的错误处理和重试机制
2. 会话监控：实现定期会话检查，预防令牌失效问题
3. 用户反馈：当检测到身份验证问题时，向用户提供清晰的反馈和重新认证的选项
4. 并发控制：避免在短时间内发起多个并发身份验证请求
5. 离线处理：考虑设备离线情况下的令牌处理策略

总结

AWS Amplify 的身份验证功能在大多数情况下工作良好，但在处理令牌刷新时可能会遇到边缘情况。通过升级到最新版本、优化令牌获取逻辑、实现适当的错误处理和重试机制，开发者可以显著提高应用的稳定性和用户体验。

对于关键业务操作，建议总是先验证令牌有效性，必要时强制刷新令牌，并准备好处理身份验证失败的情况。这些措施将帮助开发者构建更健壮的 Angular 应用，与 AWS Cognito 服务实现无缝集成。