feat(spanner/spannertest): implement ANY_VALUE aggregation function (#…

…3428) Fixes #3375.
googleapis · Dec 9, 2020 · e16c3e9 · e16c3e9
1 parent 48aa1ce
commit e16c3e9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/spanner/spannertest/funcs.go b/spanner/spannertest/funcs.go
@@ -37,6 +37,19 @@ type aggregateFunc struct {
 
 // TODO: more aggregate funcs.
 var aggregateFuncs = map[string]aggregateFunc{
+	"ANY_VALUE": {
+		// https://cloud.google.com/spanner/docs/aggregate_functions#any_value
+		Eval: func(values []interface{}, typ spansql.Type) (interface{}, spansql.Type, error) {
+			// Return the first non-NULL value.
+			for _, v := range values {
+				if v != nil {
+					return v, typ, nil
+				}
+			}
+			// Either all values are NULL, or there are no values.
+			return nil, typ, nil
+		},
+	},
 	"ARRAY_AGG": {
 		// https://cloud.google.com/spanner/docs/aggregate_functions#array_agg
 		Eval: func(values []interface{}, typ spansql.Type) (interface{}, spansql.Type, error) {

diff --git a/spanner/spansql/keywords.go b/spanner/spansql/keywords.go
@@ -129,6 +129,7 @@ var keywords = map[string]bool{
 // https://cloud.google.com/spanner/docs/functions-and-operators
 var funcs = map[string]bool{
 	// Aggregate functions.
+	"ANY_VALUE": true,
 	"ARRAY_AGG": true,
 	"AVG":       true,
 	"BIT_XOR":   true,