add demo and silence warnings

Author: connortsui20

optd-mvp/src/entities/prelude.rs

@@ -1,5 +1,7 @@
 //! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.0
 
+#![allow(unused_imports)]
+
 pub use super::cascades_group::Entity as CascadesGroup;
 pub use super::logical_children::Entity as LogicalChildren;
 pub use super::logical_expression::Entity as LogicalExpression;

optd-mvp/src/main.rs

@@ -1,3 +1,97 @@
-fn main() {
-    println!("Hello, world!");
+use sea_orm::*;
+use sea_orm_migration::prelude::*;
+use serde_json::json;
+
+mod entities;
+mod migrator;
+
+use entities::{prelude::*, *};
+use optd_mvp::DATABASE_URL;
+
+#[tokio::main]
+async fn main() {
+    basic_demo().await;
+}
+
+async fn basic_demo() {
+    let db = Database::connect(DATABASE_URL).await.unwrap();
+
+    // Create a new `CascadesGroup`.
+    let group = cascades_group::ActiveModel {
+        winner: ActiveValue::Set(None),
+        is_optimized: ActiveValue::Set(false),
+        ..Default::default()
+    }
+    .save(&db)
+    .await
+    .unwrap();
+
+    // Create a new logical expression.
+    let l_expr = logical_expression::ActiveModel {
+        group_id: group.id.clone(),
+        fingerprint: ActiveValue::Set(42), // Example fingerprint
+        kind: ActiveValue::Set(1),         // Example variant tag
+        data: ActiveValue::Set(json!({ // Example operator
+            "type": "Scan",
+            "table": "lineitem",
+            "predicate": "l_quantity < 10",
+        })),
+        ..Default::default()
+    }
+    .save(&db)
+    .await
+    .unwrap();
+
+    // Create a link between the group and the logical expression in the junction table.
+    let _link = logical_children::ActiveModel {
+        group_id: group.id.clone(),
+        logical_expression_id: l_expr.id.clone(),
+    }
+    .insert(&db)
+    .await
+    .unwrap();
+
+    // Basic lookup test on each table.
+    {
+        let groups: Vec<cascades_group::Model> = CascadesGroup::find().all(&db).await.unwrap();
+        assert_eq!(groups.len(), 1);
+
+        let l_expressions: Vec<logical_expression::Model> =
+            LogicalExpression::find().all(&db).await.unwrap();
+        assert_eq!(l_expressions.len(), 1);
+    }
+
+    // Retrieve all logical expressions that belong to this group with lazy loading.
+    {
+        let group = CascadesGroup::find_by_id(*group.id.try_as_ref().unwrap())
+            .one(&db)
+            .await
+            .unwrap()
+            .unwrap();
+
+        let group_expressions: Vec<logical_expression::Model> = group
+            .find_related(LogicalExpression)
+            .all(&db)
+            .await
+            .unwrap();
+        assert_eq!(group_expressions.len(), 1);
+    }
+
+    // Retrieve all logical expressions that belong to this group with eager loading.
+    {
+        let group_with_expressions: Vec<(cascades_group::Model, Vec<logical_expression::Model>)> =
+            CascadesGroup::find()
+                .find_with_related(LogicalExpression)
+                .all(&db)
+                .await
+                .unwrap();
+        assert_eq!(group_with_expressions.len(), 1);
+        assert_eq!(group_with_expressions[0].1.len(), 1);
+    }
+
+    // Clean up everything. Since everything is cascading, we only need to manually delete the group
+    // and then SeaORM will take care of the expression and the junction.
+    group.delete(&db).await.unwrap();
+
+    println!("Demo Finished!");
 }

optd-mvp/src/migrator/mod.rs

@@ -2,6 +2,8 @@ use sea_orm_migration::prelude::*;
 
 mod memo;
 
+/// This struct is just a helper to execute migrations easily.
+#[allow(dead_code)]
 pub struct Migrator;
 
 #[async_trait::async_trait]