回调和通知

Stable Diffusion on Amazon EKS方案采用异步推理模式，当图片生成或报错后，会通过Amazon SNS通知用户。用户应用可以通过订阅 SNS 主题以获取图片生成完成的通知。

添加订阅

请参考 Amazon SNS文档 以了解 SNS 支持的消息目标类型。

您可以从CloudFormation的输出中找到生成的 SNS 主题 ARN：

AWS 管理控制台AWS CLI

• 进入 AWS CloudFormation 控制台
• 选择 Stacks （堆栈）
• 在列表中，选择 SdOnEKSStack （或您自定义的名称）
• 选择 Output （输出）
• 记录 sdNotificationOutputArn 项的值（格式为 arn:aws:sns:us-east-1:123456789012:SdOnEKSStack-sdNotificationOutputCfn-abcdefgh）

运行以下命令以获取 SNS 主题 ARN：

aws cloudformation describe-stacks --stack-name SdOnEKSStack --output text --query 'Stacks[0].Outputs[?OutputKey==`sdNotificationOutputArn`].OutputValue'

如需接收消息，您需要将您的消息接收端（如Amazon SQS队列，HTTP 终端节点等）作为订阅添加到该SNS主题中。

AWS 管理控制台AWS CLI

• 在左侧导航窗格中，选择Subscriptions （订阅）。
• 在 Subscriptions（订阅）页面上，选择 Create subscription（创建订阅）。
• 在 Create subscription（创建订阅）页上的 Details（详细信息）部分中，执行以下操作：
  • 对于 Topic ARN（主题 ARN），选择您在上一步中记录的ARN。
  • 对于 Protocol（协议），选择您的接收端类型。
  • 对于 Endpoint（终端节点），输入您的接收端地址，例如电子邮件地址或 Amazon SQS 队列的 ARN。
• 选择 Create subscription（创建订阅）

请参考Use Amazon SNS with the AWS CLI 添加对该主题的订阅。

回调消息格式

解决方案会通过以下格式向SNS发送任务完成通知，此通知与请求时所使用的的API版本无关：

{
    "id": "task_id", // 任务ID
    "result": true, // true为成功完成，false为未成功完成
    "image_url": [ // 生成图像的S3 URL，格式为 任务ID+4位随机码+图片序号，如有多张图片则所有图片链接均会附上
        "s3://outputbucket/output/test-t2i/test-t2i-abcd-1.png"
    ],
    "output_url": "s3://outputbucket/output/test-t2i/test-t2i-abcd.out", // 任务返回的S3 URL，包含运行时的完整返回
    "context": { // 请求时附带的Context内容
        "abc": 123
    }
}