AWS Amplify/Next.js 身份验证会话存储问题深度解析

问题背景

在使用AWS Amplify与Next.js框架集成时，开发者经常会遇到用户会话无法正确存储在Cookie中的问题。特别是在服务器端渲染(SSR)场景下，这个问题尤为突出。本文将深入分析这个问题的根源，并提供完整的解决方案。

核心问题表现

当开发者配置了AWS Amplify的认证模块，并启用了SSR模式后，可能会遇到以下现象：

1. 用户登录流程（signIn和confirmSignIn）能够成功执行
2. 登录状态仅保存在localStorage中，而非预期的Cookie存储
3. 中间件无法通过Cookie验证用户会话
4. 在开发环境下表现尤为明显

根本原因分析

经过对AWS Amplify源码和实际案例的研究，我们发现这个问题主要由以下几个因素导致：

1. Cookie域配置不当

浏览器会拒绝存储域配置不匹配的Cookie。常见情况包括：

• 开发环境使用localhost或IP地址，但Cookie配置了生产域名
• 未正确设置secure标志（开发环境应为false）
• 域协议不匹配（http与https混用）

2. 重复配置问题

在Next.js应用中，常见的错误模式是在多个位置重复配置Amplify：

• 服务器组件(layout.ts)
• 客户端组件
• 中间件处理文件 这种重复配置可能导致状态不一致。

3. 动态配置的风险

很多开发者尝试根据运行时环境动态生成配置，特别是动态设置redirectSignIn和redirectSignOut的URL。这种做法在SSR场景下容易导致Cookie存储失败。

解决方案

1. 正确的Amplify配置方式

推荐采用以下配置结构：

// amplify-config.ts
export const getAmplifyConfig = (): ResourcesConfig => ({
  Auth: {
    Cognito: {
      userPoolId: process.env.AWS_COGNITO_USER_POOL_ID,
      userPoolClientId: process.env.AWS_COGNITO_CLIENT_ID,
      loginWith: {
        oauth: {
          domain: process.env.AWS_COGNITO_DOMAIN,
          scopes: ["email", "openid", "profile"],
          redirectSignIn: ["http://localhost:3000/signin"],
          redirectSignOut: ["http://localhost:3000/signout"],
          responseType: "token"
        }
      }
    }
  }
});

2. 专门的客户端配置组件

创建一个客户端组件专门处理Amplify配置：

// ConfigureAmplifyClientSide.tsx
"use client";
import { Amplify } from "aws-amplify";
import { cognitoUserPoolsTokenProvider } from "aws-amplify/auth/cognito";
import { CookieStorage } from "aws-amplify/utils";
import { getAmplifyConfig } from "./amplify-config";

const config = getAmplifyConfig();

Amplify.configure(config, { ssr: true });

cognitoUserPoolsTokenProvider.setKeyValueStorage(
  new CookieStorage({
    domain: window.location.hostname,
    path: "/",
    secure: process.env.NODE_ENV === "production",
    sameSite: "Lax"
  })
);

export default function ConfigureAmplifyClientSide() {
  return null;
}

3. 中间件正确实现

中间件中应该这样处理认证状态：

// middleware.ts
import { NextResponse } from "next/server";
import { createServerRunner } from "@aws-amplify/adapter-nextjs";
import { fetchAuthSession } from "aws-amplify/auth/server";
import { getAmplifyConfig } from "./amplify-config";

export async function middleware(request) {
  const response = NextResponse.next();
  const { runWithAmplifyServerContext } = createServerRunner({
    config: getAmplifyConfig()
  });

  const authenticated = await runWithAmplifyServerContext({
    nextServerContext: { request, response },
    operation: async (contextSpec) => {
      try {
        const session = await fetchAuthSession(contextSpec);
        return !!session?.tokens?.idToken;
      } catch (error) {
        console.error(error);
        return false;
      }
    }
  });

  if (!authenticated) {
    return NextResponse.redirect(new URL("/signin", request.url));
  }

  return response;
}

最佳实践建议

1. 环境区分：为开发和生产环境创建不同的配置，避免动态生成
2. Cookie安全：生产环境必须启用secure标志，开发环境则应禁用
3. 配置单一来源：确保Amplify配置只在一个地方定义
4. 协议一致性：确保所有URL使用相同协议（http或https）
5. 测试验证：使用浏览器开发者工具检查Cookie是否被正确设置

常见问题排查

如果按照上述方案仍然遇到问题，可以检查以下方面：

1. 浏览器开发者工具中的Console和Network标签页，查看是否有错误或警告
2. Application标签页中的Cookies部分，确认Cookie是否被存储
3. 确保没有浏览器插件阻止Cookie的设置
4. 检查AWS Cognito控制台中的应用客户端设置是否正确

通过以上方法和建议，开发者应该能够解决AWS Amplify在Next.js应用中会话存储的问题，实现稳定可靠的用户认证流程。