Attach request variables to query executor context

src/dbs/dbs.rs

@@ -1,19 +1,19 @@
 use crate::dbs::executor::Executor;
 use crate::dbs::response::Responses;
 use crate::dbs::session::Session;
+use crate::dbs::variables::Attach;
+use crate::dbs::variables::Variables;
 use crate::err::Error;
 use crate::sql;
 use crate::sql::query::Query;
-use crate::sql::value::Value;
-use std::collections::HashMap;
-
-pub type Variables = Option<HashMap<String, Value>>;
 
 pub async fn execute(txt: &str, session: Session, vars: Variables) -> Result<Responses, Error> {
 	// Create a new query executor
 	let mut exe = Executor::new();
 	// Create a new execution context
 	let ctx = session.context();
+	// Attach the defined variables
+	let ctx = vars.attach(ctx);
 	// Parse the SQL query text
 	let ast = sql::parse(txt)?;
 	// Process all statements
@@ -27,6 +27,8 @@ pub async fn process(ast: Query, session: Session, vars: Variables) -> Result<Re
 	let mut exe = Executor::new();
 	// Store session info on context
 	let ctx = session.context();
+	// Attach the defined variables
+	let ctx = vars.attach(ctx);
 	// Process all statements
 	exe.ns = session.ns;
 	exe.db = session.db;

src/dbs/mod.rs

@@ -6,6 +6,7 @@ mod options;
 mod response;
 mod runtime;
 mod session;
+mod variables;
 
 pub use self::auth::*;
 pub use self::dbs::*;
@@ -15,6 +16,7 @@ pub use self::options::*;
 pub use self::response::*;
 pub use self::runtime::*;
 pub use self::session::*;
+pub use self::variables::*;
 
 #[cfg(test)]
 pub(crate) mod test;

src/dbs/variables.rs

@@ -0,0 +1,25 @@
+use crate::ctx::Context;
+use crate::sql::value::Value;
+use std::collections::HashMap;
+use std::sync::Arc;
+
+pub type Variables = Option<HashMap<String, Value>>;
+
+pub(crate) trait Attach {
+	fn attach(self, ctx: Arc<Context>) -> Arc<Context>;
+}
+
+impl Attach for Variables {
+	fn attach(self, ctx: Arc<Context>) -> Arc<Context> {
+		match self {
+			Some(m) => {
+				let mut ctx = Context::new(&ctx);
+				for (key, val) in m {
+					ctx.add_value(key, val);
+				}
+				ctx.freeze()
+			}
+			None => ctx,
+		}
+	}
+}