feat: add support for JSONB operators in List hydrate functions (#584)

turbot · Jul 5, 2023 · f3ceed5 · f3ceed5
1 parent c81a226
commit f3ceed5
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/plugin/key_column.go b/plugin/key_column.go
@@ -37,6 +37,14 @@ func GetValidOperators() []string {
 		quals.QualOperatorNotIRegex,
 		quals.QualOperatorIsNull,
 		quals.QualOperatorIsNotNull,
+
+		quals.QualOperatorJsonbContainsLeftRight,
+		quals.QualOperatorJsonbContainsRightLeft,
+		quals.QualOperatorJsonbExistsOne,
+		quals.QualOperatorJsonbExistsAny,
+		quals.QualOperatorJsonbExistsAll,
+		quals.QualOperatorJsonbPathExists,
+		quals.QualOperatorJsonbPathPredicate,
 	}
 }
 
@@ -46,7 +54,7 @@ func GetValidOperators() []string {
 //
 // # Operators
 //
-// This property specifies the accepted operators (from a possible set: "=", "<>", "<", "<=", ">", ">=", "~~", "!~~", "~~*", "!~~*", "~", "~*", "!~", "!~*")
+// This property specifies the accepted operators (from a possible set: "=", "<>", "<", "<=", ">", ">=", "~~", "!~~", "~~*", "!~~*", "~", "~*", "!~", "!~*", "@>", "<@", "?", "?|", "?&", "@?", "@@")
 //
 // # Require
 //

diff --git a/plugin/quals/qual.go b/plugin/quals/qual.go
@@ -22,6 +22,14 @@ const (
 	QualOperatorNotIRegex      = "!~*"
 	QualOperatorIsNull         = "is null"
 	QualOperatorIsNotNull      = "is not null"
+
+	QualOperatorJsonbContainsLeftRight = "@>"
+	QualOperatorJsonbContainsRightLeft = "<@"
+	QualOperatorJsonbExistsOne         = "?"
+	QualOperatorJsonbExistsAny         = "?|"
+	QualOperatorJsonbExistsAll         = "?&"
+	QualOperatorJsonbPathExists        = "@?"
+	QualOperatorJsonbPathPredicate     = "@@" // FIXME: I'm a little confused about this one
 )
 
 // Qual is a struct which represents a database qual in a more easily digestible form that proto.Qual