Merge pull request #364 from dojoengine/fix/in-operator

fix: support in operator
dojoengine · Jan 1, 2025 · 58ccba5 · 58ccba5
2 parents 33821e5 + b9cb6be
commit 58ccba5
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/packages/sdk/src/convertQuerytoClause.ts b/packages/sdk/src/convertQuerytoClause.ts
@@ -313,6 +313,10 @@ function convertOperator(operator: string): torii.ComparisonOperator {
             return "Lt";
         case "$lte":
             return "Lte";
+        case "$in":
+            return "In";
+        case "$nin":
+            return "NotIn";
         default:
             throw new Error(`Unsupported operator: ${operator}`);
     }

diff --git a/packages/sdk/src/queryBuilder.ts b/packages/sdk/src/queryBuilder.ts
@@ -140,6 +140,20 @@ class QueryEntity<T extends SchemaType> {
         return this.addConstraint(field, value, Operator.lte);
     }
 
+    public in(
+        field: FirstLevelKeys<T[keyof T & string][keyof T[keyof T & string]]>,
+        value: any
+    ): QueryEntity<T> {
+        return this.addConstraint(field, value, Operator.in);
+    }
+
+    public notIn(
+        field: FirstLevelKeys<T[keyof T & string][keyof T[keyof T & string]]>,
+        value: any
+    ): QueryEntity<T> {
+        return this.addConstraint(field, value, Operator.nin);
+    }
+
     private addConstraint(
         field: FirstLevelKeys<T[keyof T & string][keyof T[keyof T & string]]>,
         value: any,
@@ -177,4 +191,6 @@ enum Operator {
     gte = "$gte",
     lt = "$lt",
     lte = "$lte",
+    in = "$in",
+    nin = "$nin",
 }