Implement IN and NOT IN (#139)

njord-rs · Oct 7, 2024 · 4869e4b · 4869e4b
2 parents a95c91d + 960115f
commit 4869e4b
Show file tree

Hide file tree

Showing 4 changed files with 59 additions and 0 deletions.
diff --git a/njord/db/insert.db b/njord/db/insert.db
diff --git a/njord/db/update.db b/njord/db/update.db
diff --git a/njord/src/condition.rs b/njord/src/condition.rs
@@ -46,6 +46,10 @@ pub enum Condition {
     And(Box<Condition>, Box<Condition>),
     /// Logical OR condition.
     Or(Box<Condition>, Box<Condition>),
+    /// In condition: column IN (value1, value2, ...).
+    In(String, Vec<String>),
+    /// Not in condition: column NOT IN (value1, value2, ...).
+    NotIn(String, Vec<String>),
 }
 
 impl Condition {
@@ -113,6 +117,22 @@ impl Condition {
             }
             Condition::And(left, right) => format!("({}) AND ({})", left.build(), right.build()),
             Condition::Or(left, right) => format!("({}) OR ({})", left.build(), right.build()),
+            Condition::In(column, values) => {
+                let values = values
+                    .iter()
+                    .map(|v| format!("'{}'", v))
+                    .collect::<Vec<String>>()
+                    .join(", ");
+                format!("{} IN ({})", column, values)
+            }
+            Condition::NotIn(column, values) => {
+                let values = values
+                    .iter()
+                    .map(|v| format!("'{}'", v))
+                    .collect::<Vec<String>>()
+                    .join(", ");
+                format!("{} NOT IN ({})", column, values)
+            }
         }
     }
 }
diff --git a/njord/tests/sqlite_test.rs b/njord/tests/sqlite_test.rs
@@ -539,3 +539,42 @@ fn update_with_sub_queries() {
         Err(e) => panic!("Failed to UPDATE: {:?}", e),
     };
 }
+
+fn select_in() {
+    let db_relative_path = "./db/select.db";
+    let db_path = Path::new(&db_relative_path);
+    let conn = sqlite::open(db_path);
+
+    let columns = vec![
+        Column::Text("id".to_string()),
+        Column::Text("username".to_string()),
+        Column::Text("email".to_string()),
+        Column::Text("address".to_string()),
+    ];
+
+    let condition = Condition::And(
+        Box::new(Condition::In(
+            "username".to_string(),
+            vec!["mjovanc".to_string(), "otheruser".to_string()],
+        )),
+        Box::new(Condition::NotIn(
+            "username".to_string(),
+            vec!["chasewillden".to_string()],
+        )),
+    );
+
+    match conn {
+        Ok(c) => {
+            let result = sqlite::select(&c, columns)
+                .from(User::default())
+                .where_clause(condition)
+                .build();
+
+            match result {
+                Ok(r) => assert_eq!(r.len(), 2),
+                Err(e) => panic!("Failed to SELECT: {:?}", e),
+            };
+        }
+        Err(e) => panic!("Failed to SELECT: {:?}", e),
+    };
+}