chore: import upstream snapshot with attribution

server/functions/aggregate/bool_aggregates.go

@@ -0,0 +1,115 @@
+// Copyright 2025 Dolthub, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package aggregate
+
+import (
+	"github.com/dolthub/go-mysql-server/sql"
+
+	"github.com/dolthub/doltgresql/server/functions/framework"
+	pgtypes "github.com/dolthub/doltgresql/server/types"
+)
+
+// initBoolAggs registers the functions to the catalog.
+func initBoolAggs() {
+	framework.RegisterAggregateFunction(boolAnd)
+	framework.RegisterAggregateFunction(boolOr)
+}
+
+// boolAnd represents the PostgreSQL bool_and function.
+var boolAnd = framework.Func1Aggregate{
+	Function1: framework.Function1{
+		Name:   "bool_and",
+		Return: pgtypes.Bool,
+		Parameters: [1]*pgtypes.DoltgresType{
+			pgtypes.Bool,
+		},
+		Callable: func(ctx *sql.Context, paramsAndReturn [2]*pgtypes.DoltgresType, val1 any) (any, error) {
+			return nil, nil
+		},
+	},
+	NewAggBuffer: newBoolAndBuffer,
+}
+
+// boolOr represents the PostgreSQL bool_or function.
+var boolOr = framework.Func1Aggregate{
+	Function1: framework.Function1{
+		Name:   "bool_or",
+		Return: pgtypes.Bool,
+		Parameters: [1]*pgtypes.DoltgresType{
+			pgtypes.Bool,
+		},
+		Callable: func(ctx *sql.Context, paramsAndReturn [2]*pgtypes.DoltgresType, val1 any) (any, error) {
+			return nil, nil
+		},
+	},
+	NewAggBuffer: newBoolOrBuffer,
+}
+
+// boolAggBuffer is an aggregation buffer for boolean logic aggregates
+type boolAggBuffer struct {
+	expr   sql.Expression
+	b      bool
+	sawOne bool
+	isAnd  bool
+}
+
+var _ sql.AggregationBuffer = (*boolAggBuffer)(nil)
+
+// newBoolAndBuffer creates a new aggregation buffer for the bool_and aggregate function.
+func newBoolAndBuffer(exprs []sql.Expression) (sql.AggregationBuffer, error) {
+	return &boolAggBuffer{
+		expr:  exprs[0],
+		b:     true,
+		isAnd: true,
+	}, nil
+}
+
+// newBoolOrBuffer creates a new aggregation buffer for the bool_or aggregate function.
+func newBoolOrBuffer(exprs []sql.Expression) (sql.AggregationBuffer, error) {
+	return &boolAggBuffer{
+		expr: exprs[0],
+	}, nil
+}
+
+// Dispose implements the sql.AggregationBuffer interface.
+func (a *boolAggBuffer) Dispose(ctx *sql.Context) {}
+
+// Eval implements the sql.AggregationBuffer interface.
+func (a *boolAggBuffer) Eval(context *sql.Context) (interface{}, error) {
+	if !a.sawOne {
+		return nil, nil
+	}
+	return a.b, nil
+}
+
+// Update implements the sql.AggregationBuffer interface.
+func (a *boolAggBuffer) Update(ctx *sql.Context, row sql.Row) error {
+	eval, err := a.expr.Eval(ctx, row)
+	if err != nil {
+		return err
+	}
+
+	if eval == nil {
+		return nil
+	}
+
+	a.sawOne = true
+	if a.isAnd {
+		a.b = a.b && eval.(bool)
+	} else {
+		a.b = a.b || eval.(bool)
+	}
+	return nil
+}

server/functions/aggregate/init.go

@@ -0,0 +1,19 @@
+// Copyright 2025 Dolthub, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package aggregate
+
+func Init() {
+	initBoolAggs()
+}