如何在AWS Athena中执行由查询生成的Select语句

你已经成功生成了SELECT子句的内容，但需要通过动态SQL来执行这个生成的语句，不同数据库的实现方式略有差异，以下是主流数据库的解决方案：

PostgreSQL #

使用PL/pgSQL块实现动态执行：

DO $$
DECLARE
  dynamic_sql TEXT;
BEGIN
  -- 生成包含原列和对应NTILE计算的SELECT内容
  SELECT CONCAT(
    ARRAY_JOIN(ARRAY_AGG(column_name), ', '),
    ', ',
    ARRAY_JOIN(ARRAY_AGG(CONCAT('ntile(4) over (order by ', column_name, ' asc) AS ntile_', column_name)), ', ')
  )
  INTO dynamic_sql
  FROM information_schema.columns
  WHERE table_name = 'main_table';

  -- 拼接完整查询语句
  dynamic_sql := CONCAT('SELECT ', dynamic_sql, ' FROM main_table');

  -- 执行动态SQL
  EXECUTE dynamic_sql;
END $$;

MySQL #

通过PREPARE和EXECUTE语句执行：

-- 生成完整查询字符串
SET @dynamic_sql = (
  SELECT CONCAT(
    GROUP_CONCAT(column_name SEPARATOR ', '),
    ', ',
    GROUP_CONCAT(CONCAT('ntile(4) over (order by ', column_name, ' asc) AS ntile_', column_name) SEPARATOR ', ')
  )
  FROM information_schema.columns
  WHERE table_name = 'main_table'
);

SET @dynamic_sql = CONCAT('SELECT ', @dynamic_sql, ' FROM main_table');

-- 执行动态SQL
PREPARE stmt FROM @dynamic_sql;
EXECUTE stmt;
DEALLOCATE PREPARE stmt;

BigQuery #

使用EXECUTE IMMEDIATE命令：

DECLARE dynamic_sql STRING;

SET dynamic_sql = (
  SELECT CONCAT(
    ARRAY_TO_STRING(ARRAY_AGG(column_name), ', '),
    ', ',
    ARRAY_TO_STRING(ARRAY_AGG(CONCAT('ntile(4) over (order by ', column_name, ' asc) AS ntile_', column_name)), ', ')
  )
  FROM information_schema.columns
  WHERE table_name = 'main_table'
);

SET dynamic_sql = CONCAT('SELECT ', dynamic_sql, ' FROM main_table');

EXECUTE IMMEDIATE dynamic_sql;

注意事项 #

• 给每个NTILE计算结果添加别名（如ntile_col1），避免结果集中出现重复列名导致报错。
• 直接查询生成的字符串只会返回文本内容，必须通过数据库的动态SQL特性才能让数据库解析并执行这条SQL语句。

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