时间戳

首页 > 解决方案 > 如何定义 Lambda 变量以通过 API Gateway 查询 Redshift

sql - 如何定义 Lambda 变量以通过 API Gateway 查询 Redshift

问题描述

我遇到了如何为 API Gateway 设置参数以使用 Lambda 函数查询 Amazon Redshift 的问题。

我的连接工作正常，但我一直得到全表响应。我需要定义变量，用户可以查询特定的参数、值和模式

有人可以建议一个如何设置它的例子吗

我的配置是：

    #!/usr/bin/env python

import psycopg2
import logging
import traceback
import json
from os import environ

query="SELECT * from public"

logger=logging.getLogger()
logger.setLevel(logging.INFO)

def make_connection():
    conn=psycopg2.connect(dbname= 'database', host='redshift-cluster.amazonaws.com',
    port= '5439', user= 'user', password= 'password')
    conn.autocommit=True
    return conn


    def log_err(errmsg):
        logger.error(errmsg)
        return {"body": errmsg , "headers": {}, "statusCode": 400,
        "isBase64Encoded":"false"}

        logger.info("Cold start complete.")

        print('Loading Function')

        def handler(event,context):

            try:
                cnx = make_connection()
                cursor=cnx.cursor()

                try:
                    cursor.execute(query)
                except:
                    return log_err ("ERROR: Cannot execute cursor.\n{}".format(
                    traceback.format_exc()) )

                    try:
                        results_list=[]
                        for result in cursor: results_list.append(result)
                        print(results_list)
                        cursor.close()

                    except:
                        return log_err ("ERROR: Cannot retrieve query data.\n{}".format(
                        traceback.format_exc()))


                        return {"body": str(results_list), "headers": {}, "statusCode": 200,
                        "isBase64Encoded":"false"}


                    except:
                        return log_err("ERROR: Cannot connect to database from handler.\n{}".format(
                        traceback.format_exc()))


                    finally:
                        try:
                            cnx.close()
                        except:
                            pass


                            if __name__== "__main__":
                                handler(None,None)

标签： sqlamazon-web-servicesaws-lambdaamazon-redshiftaws-api-gateway

解决方案

psycopg2我建议不要使用，而是使用 Amazon Redshift Data API。它提供了一种在 Amazon Redshift 上运行查询的更简单方法。

首先，使用 发送查询execute_statement()，然后使用 检索结果get_statement_result()。

推荐阅读