feat: add In and Like comparison operators to gql

Larkooo · May 10, 2024 · 016ef91 · 016ef91
1 parent 6dca944
commit 016ef91
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/crates/torii/graphql/src/query/data.rs b/crates/torii/graphql/src/query/data.rs
@@ -227,6 +227,18 @@ fn build_conditions(keys: &Option<Vec<String>>, filters: &Option<Vec<Filter>>) -
         conditions.extend(filters.iter().map(|filter| match &filter.value {
             FilterValue::Int(i) => format!("{} {} {}", filter.field, filter.comparator, i),
             FilterValue::String(s) => format!("{} {} '{}'", filter.field, filter.comparator, s),
+            FilterValue::List(list) => {
+                let values = list
+                    .iter()
+                    .map(|value| match value {
+                        FilterValue::Int(i) => i.to_string(),
+                        FilterValue::String(s) => format!("'{}'", s),
+                        FilterValue::List(_) => unreachable!(),
+                    })
+                    .collect::<Vec<_>>()
+                    .join(", ");
+                format!("{} {} ({})", filter.field, filter.comparator, values)
+            }
         }));
     }
 

diff --git a/crates/torii/graphql/src/query/filter.rs b/crates/torii/graphql/src/query/filter.rs
@@ -13,6 +13,9 @@ pub enum Comparator {
     Lte,
     Neq,
     Eq,
+    In,
+    NotIn,
+    Like,
 }
 
 impl fmt::Display for Comparator {
@@ -24,6 +27,9 @@ impl fmt::Display for Comparator {
             Comparator::Lte => write!(f, "<="),
             Comparator::Neq => write!(f, "!="),
             Comparator::Eq => write!(f, "="),
+            Comparator::In => write!(f, "IN"),
+            Comparator::NotIn => write!(f, "NOT IN"),
+            Comparator::Like => write!(f, "LIKE"),
         }
     }
 }
@@ -32,6 +38,7 @@ impl fmt::Display for Comparator {
 pub enum FilterValue {
     Int(i64),
     String(String),
+    List(Vec<FilterValue>),
 }
 
 #[derive(Debug)]