Introduction > SAM入门 > 集成DynamoDB Table

集成DynamoDB Table

本节我们使用SAM和DynamoDB进行集成。SAM将自动创建一张DynamoDB表，在lambda中对这张表的数据进行查询，并将结果返回给API Gateway

创建项目

执行sam init，使用Python + zip模式，为项目命令名dynamodb-app:

创建完成后，进入到dynamodb-app目录，更新hello_world/app.py文件内容如下：

import boto3
import json
import os

print('Loading function')

# create the client outside of the handler
region_name = os.environ['REGION_NAME']  # 从env获取REGION_NAME，这个环境变量由template.yaml传入
dynamo = boto3.client('dynamodb', region_name=region_name)
table_name = os.environ['TABLE_NAME']  # 从env获取TABLE_NAME，这个环境变量由template.yaml传入

def respond(err, res=None):
    return {
        'statusCode': '400' if err else '200',
        'body': err.message if err else json.dumps(res),
        'headers': {
            'Content-Type': 'application/json',
        },
    }


def lambda_handler(event, context):
    print("Received event: " + json.dumps(event, indent=2))
    scan_result = dynamo.scan(TableName=table_name)  # 扫描DynamoDB全表数据
    return respond(None, res=scan_result)  # 将上面的查询结果返回

更新template.yaml内容如下：

# SAM FILE
AWSTemplateFormatVersion: '2010-09-09'
Transform: 'AWS::Serverless-2016-10-31'
Description: A starter AWS Lambda function.
Resources:
  helloworldpython3:
    Type: 'AWS::Serverless::Function'
    Properties:
      Handler: app.lambda_handler
      Runtime: python3.7   # 版本和当前环境Python保持一致
      CodeUri: hello_world/   # 代码目录在hello_world下面
      Description: A starter AWS Lambda function.
      MemorySize: 128
      Timeout: 3
      Environment:  # 传入环境变量
        Variables:
          TABLE_NAME: !Ref Table   
          REGION_NAME: !Ref AWS::Region
      Events:  # API Gateway设置
        HelloWorldSAMAPI:
          Type: Api
          Properties:
            Path: /hello
            Method: GET
      Policies:  # Lambda需要有访问DynamoDB CRUD权限
        - DynamoDBCrudPolicy:
            TableName: !Ref Table  

  Table:  # 创建DynamoDB表
    Type: AWS::Serverless::SimpleTable
    Properties:
      PrimaryKey: # Primary设置
        Name: greeting
        Type: String
      ProvisionedThroughput:  # DynamoDB RCU/WCU配置
        ReadCapacityUnits: 2
        WriteCapacityUnits: 2