risingwave_expr_impl/aggregate/
bool_or.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
15use risingwave_expr::aggregate;
16
17/// Returns true if any non-null input value is true, otherwise false.
18///
19/// # Example
20///
21/// ```slt
22/// statement ok
23/// create table t (b1 boolean, b2 boolean, b3 boolean, b4 boolean);
24///
25/// query T
26/// select bool_or(b1) from t;
27/// ----
28/// NULL
29///
30/// statement ok
31/// insert into t values
32///     (true,  null, false, null),
33///     (false, true, null,  null),
34///     (null,  true, false, null);
35///
36/// query TTTTTT
37/// select
38///     bool_or(b1),
39///     bool_or(b2),
40///     bool_or(b3),
41///     bool_or(b4),
42///     bool_or(NOT b2),
43///     bool_or(NOT b3)
44/// FROM t;
45/// ----
46/// t t f NULL f t
47///
48/// statement ok
49/// drop table t;
50/// ```
51#[aggregate("bool_or(boolean) -> boolean")]
52fn bool_or_append_only(state: bool, input: bool) -> bool {
53    state || input
54}
55
56/// Returns true if any non-null input value is true, otherwise false.
57///
58/// # Example
59///
60/// ```slt
61/// statement ok
62/// create table t (b boolean);
63///
64/// statement ok
65/// create materialized view mv as select bool_or(b) from t;
66///
67/// query T
68/// select * from mv;
69/// ----
70/// NULL
71///
72/// statement ok
73/// insert into t values (true), (false), (null);
74///
75/// query T
76/// select * from mv;
77/// ----
78/// t
79///
80/// statement ok
81/// delete from t where b is true;
82///
83/// query T
84/// select * from mv;
85/// ----
86/// f
87///
88/// statement ok
89/// drop materialized view mv;
90///
91/// statement ok
92/// drop table t;
93/// ```
94#[derive(Debug, Default, Clone)]
95struct BoolOrUpdatable;
96
97#[aggregate("bool_or(boolean) -> boolean", state = "int8")]
98impl BoolOrUpdatable {
99    // state is the number of true values
100
101    fn accumulate(&self, state: i64, input: bool) -> i64 {
102        if input { state + 1 } else { state }
103    }
104
105    fn retract(&self, state: i64, input: bool) -> i64 {
106        if input { state - 1 } else { state }
107    }
108
109    #[allow(dead_code)] // TODO: support merge
110    fn merge(&self, state1: i64, state2: i64) -> i64 {
111        state1 + state2
112    }
113
114    fn finalize(&self, state: i64) -> bool {
115        state != 0
116    }
117}
risingwave_expr_impl/aggregate/bool_or.rs

risingwave_expr_impl/aggregate/
bool_or.rs
