要在AWS Amplify中成功登录后保留会话，可以使用本地存储（Local Storage）或Cookie来存储用户的身份验证令牌。以下是一个基本的解决方案，其中使用了Local Storage来保存令牌。

首先，确保你已经配置好了AWS Amplify并成功实现了登录功能。

在用户成功登录后，将令牌存储在Local Storage中。以下是一个示例代码：

import Amplify, { Auth } from 'aws-amplify';

// 登录函数
async function signIn(username, password) {
  try {
    // 调用AWS Amplify的登录方法
    const user = await Auth.signIn(username, password);
    
    // 获取身份验证令牌
    const token = user.signInUserSession.idToken.jwtToken;
    
    // 将令牌存储在Local Storage中
    localStorage.setItem('token', token);
    
    // 重定向到其他页面或执行其他操作
    // ...
  } catch (error) {
    console.log('登录失败', error);
  }
}

接下来，在应用程序加载时，检查Local Storage中是否存在令牌，并在必要时将其设置为默认的请求头。以下是一个示例代码：

import Amplify, { Auth } from 'aws-amplify';

// 检查Local Storage中是否存在令牌
const token = localStorage.getItem('token');

// 如果令牌存在，将其设置为默认的请求头
if (token) {
  Amplify.configure({
    API: {
      graphql_headers: async () => ({
        Authorization: token
      })
    }
  });
}

// 在应用程序加载后初始化AWS Amplify
Amplify.configure({
  // 配置...
});

这样，当用户成功登录后，令牌将被存储在Local Storage中，并在每次请求时自动添加到请求头中。在用户关闭浏览器或注销后，令牌将被删除，从而实现会话的保留与销毁。