能否通过CloudFormation或CDK配置Athena数据源？还有哪些可行方式？

Athena的数据源配置核心是创建**数据目录（Data Catalog）**并关联对应的联邦Lambda连接器，以下是几种可行的配置方式：

一、AWS SDK（直接调用API） #

这是最直接的程序化配置方式，通过AWS SDK调用CreateDataCatalog接口即可完成数据源创建，支持所有类型的联邦数据源。

示例（Python Boto3） #

import boto3

athena_client = boto3.client('athena')

# 创建关联MySQL RDS的数据源
athena_client.create_data_catalog(
    Name='MySQLRDSCatalog',
    Type='LAMBDA',
    Description='Data catalog for MySQL RDS',
    Parameters={
        'lambdaFunctionArn': 'arn:aws:lambda:us-east-1:123456789012:function:aws-athena-mysql-connector'
    },
    Tags=[{'Key': 'Environment', 'Value': 'Production'}]
)

# 创建自定义联邦Lambda的数据源
athena_client.create_data_catalog(
    Name='CustomLambdaCatalog',
    Type='LAMBDA',
    Description='Custom federated data catalog',
    Parameters={
        'lambdaFunctionArn': 'arn:aws:lambda:us-east-1:123456789012:function:your-custom-federated-lambda'
    }
)

注意：执行操作的IAM角色需要拥有athena:CreateDataCatalog权限，以及调用目标Lambda的lambda:InvokeFunction权限。

二、控制台配置后导出CloudFormation模板 #

虽然原生CloudFormation没有直接定义Athena数据目录的资源类型，但可以通过以下方式生成可复用的模板：

1. 在Athena控制台手动创建目标数据目录（关联RDS或自定义Lambda）。
2. 进入CloudFormation控制台，找到自动生成的对应栈（名称通常包含AthenaDataCatalog）。
3. 导出该栈的模板，按需修改后即可用于批量部署。

导出的模板会包含Lambda权限、IAM角色等关联资源，需确保这些配置匹配你的环境需求。

三、使用AWS CDK #

AWS CDK提供了CfnDataCatalog构造，可通过代码定义Athena数据目录，最终编译为CloudFormation栈部署。

示例（TypeScript CDK） #

import * as cdk from 'aws-cdk-lib';
import * as athena from 'aws-cdk-lib/aws-athena';

const app = new cdk.App();
const stack = new cdk.Stack(app, 'AthenaDataCatalogStack');

// 创建Postgres RDS数据源
new athena.CfnDataCatalog(stack, 'PostgresRDSCatalog', {
    name: 'PostgresRDSCatalog',
    type: 'LAMBDA',
    description: 'Data catalog for Postgres RDS',
    parameters: {
        lambdaFunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:aws-athena-postgresql-connector'
    },
    tags: [{ key: 'Environment', value: 'Production' }]
});

// 创建自定义联邦Lambda数据源
new athena.CfnDataCatalog(stack, 'CustomLambdaCatalog', {
    name: 'CustomLambdaCatalog',
    type: 'LAMBDA',
    description: 'Custom federated data catalog',
    parameters: {
        lambdaFunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:your-custom-federated-lambda'
    }
});

app.synth();

针对不同数据源的额外注意事项 #

• MySQL/Postgres RDS：使用AWS官方联邦连接器Lambda时，需确保Lambda配置了正确的VPC权限（可访问RDS实例），并拥有RDS的访问权限（通过IAM角色或数据库凭证）。
• 自定义联邦Lambda：需遵循Athena联邦接口规范开发，确保能处理Athena的查询请求并返回符合格式的结果。

内容的提问来源于stack exchange，提问作者Mridang Agarwalla