fix: compiler example

prisma · Jan 15, 2025 · 3413289 · 3413289
1 parent 1d35fa4
commit 3413289
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/query-engine/query-engine/Cargo.toml b/query-engine/query-engine/Cargo.toml
@@ -39,6 +39,7 @@ serial_test = "*"
 quaint.workspace = true
 indoc.workspace = true
 indexmap.workspace = true
+sql-query-builder = { path = "../query-builders/sql-query-builder" }
 
 [build-dependencies]
 build-utils.path = "../../libs/build-utils"
diff --git a/query-engine/query-engine/examples/compiler.rs b/query-engine/query-engine/examples/compiler.rs
@@ -1,8 +1,13 @@
 use std::sync::Arc;
 
+use quaint::{
+    prelude::{ConnectionInfo, ExternalConnectionInfo, SqlFamily},
+    visitor::Postgres,
+};
 use query_core::{query_graph_builder::QueryGraphBuilder, QueryDocument};
 use request_handlers::{JsonBody, JsonSingleQuery, RequestBody};
 use serde_json::json;
+use sql_query_builder::{Context, SqlQueryBuilder};
 
 pub fn main() -> anyhow::Result<()> {
     let schema_string = include_str!("./schema.prisma");
@@ -66,7 +71,16 @@ pub fn main() -> anyhow::Result<()> {
 
     println!("{graph}");
 
-    let expr = query_core::compiler::translate(graph)?;
+    let connection_info = ConnectionInfo::External(ExternalConnectionInfo::new(
+        SqlFamily::Postgres,
+        "public".to_owned(),
+        None,
+    ));
+
+    let ctx = Context::new(&connection_info, None);
+    let builder = SqlQueryBuilder::<Postgres<'_>>::new(ctx);
+
+    let expr = query_core::compiler::translate(graph, &builder)?;
 
     println!("{}", expr.pretty_print(true, 80)?);