如何用presto中的json字符串中的列名替换参数值

Question

我有一个名为 sampleTable 的表，其值是这样的（请注意，列最多可以达到 key100，说明了一个小示例，以降低示例的复杂性）。

select * from sampleTable;

key1	key2
1111	2222

我正在尝试生成一个查询，该查询创建一个 JSON 字符串，该字符串可以将这些列值包含到每行一个 JSON 字符串，如下面提到的格式。

 '{"keys":{"key1":"1111", "key2":"2222"}}'

通过 presto 查询的正确方法是什么？ 我的尝试如下：

尝试 1

select 
    concat('{"keys":{"key1": ', a.key1, ',',
    '{"key2": ' ,a.key2, '}}') 
    AS keyJson 
from sampleTable ;

尝试 2

select 
    replace(
        replace('{"keys":{"key1": $key1, "key2": $key2}}','$key1',d.key1), 
        '$key2',d.key2 ) 
    AS keyJson 
from sampleTable ; 

考虑到密钥可以从 key1 变为 key100，有没有更简单的方法可以做到这一点。对于上下文，我从 java 程序动态生成查询。在存在多个键的情况下，在这两种情况下使用 StringBuilder 方法附加字符串变得复杂。

Answer 1

惯用的方法是构造一个ROW 值，然后将其强制转换为JSON：

• 首先，从列中构造一行：row(key1, key2, key3)。这将产生一个匿名行类型（即，具有未命名字段的行）
• 其次，将值转换为具有命名字段的行：CAST(... AS ROW(key1 VARCHAR, key2 VARCHAR, key3 VARCHAR))。当行转换为JSON时，字段名称将成为键。
• 最后，将结果值转换为JSON：CAST(... AS JSON)

这是一个完整的例子：

WITH data(key1, key2, key3) AS (
    VALUES
        ('1111', '2222', '3333')
)
SELECT
    CAST(
        CAST(row(key1, key2, key3) AS ROW(key1 VARCHAR, key2 VARCHAR, key3 VARCHAR)) AS JSON)
FROM data

产生：

                    _col0
---------------------------------------------
 {"key1":"1111","key2":"2222","key3":"3333"}
(1 row)

用Trino 359 (formerly known as PrestoSQL) 测试