Fix StaticType.flatten() on AnyOfType of AnyType

partiql · Mar 20, 2024 · 16a5220 · 16a5220
1 parent a18e8e0
commit 16a5220
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -35,6 +35,7 @@ Thank you to all who have contributed!
 ### Deprecated
 
 ### Fixed
+- `StaticType.flatten()` on an `AnyOfType` with `AnyType` will return `AnyType`
 
 ### Removed
 

diff --git a/partiql-types/src/main/kotlin/org/partiql/types/StaticType.kt b/partiql-types/src/main/kotlin/org/partiql/types/StaticType.kt
@@ -630,7 +630,7 @@ public data class AnyOfType(val types: Set<StaticType>, override val metas: Map<
         types = this.types.flatMap {
             when (it) {
                 is SingleType -> listOf(it)
-                is AnyType -> listOf(it)
+                is AnyType -> return@flatten it // if `AnyType`, return `AnyType`
                 is AnyOfType -> it.types
             }
         }.toSet()