【问题标题】:AWS Lambda S3 copy to Postgres succeeds but doesn't copy dataAWS Lambda S3 复制到 Postgres 成功但不复制数据
【发布时间】:2019-12-17 13:58:47
【问题描述】:

我有一个 lambda 函数,它使用 psycopg2 连接到我尝试配置的 Postgres RDS 实例,以便它使用 aws_s3.table_import_from_s3 方法将数据直接从 S3 复制到 Postgres。 lambda 函数说它执行成功,但是当我轮询数据库时,实际上没有插入任何数据。这是我的代码:

import psycopg2
import os

def lambda_handler(event, context):
    print(f'Received event {str(event)}')
    bucket = event['Records'][0]['s3']['bucket']['name']
    key = event['Records'][0]['s3']['object']['key']

    query_str = f"""
    select aws_s3.table_import_from_s3(
    'printer_mailings',
    '',
    '(format csv)',
    '{bucket}',
    '{key}',
    'us-east-1')
    ;"""

    try:
        connection = psycopg2.connect(user = os.environ["user"],
                                      password = os.environ["pwd"],
                                      host = os.environ["host"],
                                      port = os.environ["port"],
                                      database = os.environ["db"])

        cursor = connection.cursor()
        print('connected successfully')

        print(query_str)
        cursor.execute(query_str)
        print('Records added successfully')

    except (Exception, psycopg2.Error) as error :
        print ("Error while connecting to PostgreSQL", error)
    finally:
        #closing database connection.
            if(connection):
                cursor.close()
                connection.close()
                print("PostgreSQL connection is closed")

知道为什么这不起作用吗?如果我在 PSQL 中运行 query_str ,它可以工作,但即使我已成功连接到数据库,这里也不会。

【问题讨论】:

  • 你只需要提交。 cursor.commit()

标签: python postgresql amazon-web-services aws-lambda


【解决方案1】:
    cursor = connection.cursor()
    print('connected successfully')

    print(query_str)
    cursor.execute(query_str)
    connection.commit() # you missed this
    print('Records added successfully')

【讨论】:

  • 稍微更正,应该是 connection.commit() 但否则它工作
  • 我在使用 sqlalchemy 执行 table_import_from_s3 时遇到了提问者问题。切换到使用这里描述的 psycopg2 修复了它
猜你喜欢
  • 2017-08-26
  • 2019-05-22
  • 1970-01-01
  • 1970-01-01
  • 2018-12-08
  • 2020-07-07
  • 2020-02-04
  • 1970-01-01
  • 1970-01-01
相关资源
最近更新 更多