risingwave_expr_impl/table_function/
jsonb.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15//! JSONB table functions.
16
17use anyhow::anyhow;
18use risingwave_common::types::JsonbRef;
19use risingwave_expr::{Result, function};
20
21/// Expands the top-level JSON array into a set of JSON values.
22#[function("jsonb_array_elements(jsonb) -> setof jsonb")]
23fn jsonb_array_elements(json: JsonbRef<'_>) -> Result<impl Iterator<Item = JsonbRef<'_>>> {
24    json.array_elements().map_err(|e| anyhow!(e).into())
25}
26
27/// Expands the top-level JSON array into a set of text values.
28#[function("jsonb_array_elements_text(jsonb) -> setof varchar")]
29fn jsonb_array_elements_text(json: JsonbRef<'_>) -> Result<impl Iterator<Item = Box<str>> + '_> {
30    let elems = jsonb_array_elements(json)?;
31    Ok(elems.map(|elem| elem.force_string().into()))
32}
33
34/// Returns the set of keys in the top-level JSON object.
35#[function("jsonb_object_keys(jsonb) -> setof varchar")]
36fn jsonb_object_keys(json: JsonbRef<'_>) -> Result<impl Iterator<Item = &str>> {
37    json.object_keys().map_err(|e| anyhow!(e).into())
38}
39
40/// Expands the top-level JSON object into a set of key/value pairs.
41#[function("jsonb_each(jsonb) -> setof struct<key varchar, value jsonb>")]
42fn jsonb_each(json: JsonbRef<'_>) -> Result<impl Iterator<Item = (&str, JsonbRef<'_>)> + '_> {
43    json.object_key_values().map_err(|e| anyhow!(e).into())
44}
45
46/// Expands the top-level JSON object into a set of key/value pairs.
47#[function("jsonb_each_text(jsonb) -> setof struct<key varchar, value varchar>")]
48fn jsonb_each_text(json: JsonbRef<'_>) -> Result<impl Iterator<Item = (&str, Box<str>)> + '_> {
49    let elems = jsonb_each(json)?;
50    Ok(elems.map(|(k, v)| (k, v.force_string().into())))
51}
risingwave_expr_impl/table_function/jsonb.rs

risingwave_expr_impl/table_function/
jsonb.rs
