Improve Sqlite support for sub-queries and CTE's

sqlx-core/src/logger.rs

@@ -1,4 +1,7 @@
 use crate::connection::LogSettings;
+use std::collections::HashSet;
+use std::fmt::Debug;
+use std::hash::Hash;
 use std::time::Instant;
 
 pub(crate) struct QueryLogger<'q> {
@@ -78,6 +81,83 @@ impl<'q> Drop for QueryLogger<'q> {
     }
 }
 
+pub(crate) struct QueryPlanLogger<'q, O: Debug + Hash + Eq, R: Debug + Hash + Eq, P: Debug> {
+    sql: &'q str,
+    unknown_operations: HashSet<O>,
+    results: HashSet<R>,
+    program: Vec<P>,
+    settings: LogSettings,
+}
+
+impl<'q, O: Debug + Hash + Eq, R: Debug + Hash + Eq, P: Debug> QueryPlanLogger<'q, O, R, P> {
+    pub(crate) fn new(sql: &'q str, settings: LogSettings) -> Self {
+        Self {
+            sql,
+            unknown_operations: HashSet::new(),
+            results: HashSet::new(),
+            program: Vec::new(),
+            settings,
+        }
+    }
+
+    pub(crate) fn add_result(&mut self, result: R) {
+        self.results.insert(result);
+    }
+
+    pub(crate) fn add_program(&mut self, program: Vec<P>) {
+        self.program = program;
+    }
+
+    pub(crate) fn add_unknown_operation(&mut self, operation: O) {
+        self.unknown_operations.insert(operation);
+    }
+
+    pub(crate) fn finish(&self) {
+        let lvl = self.settings.statements_level;
+
+        if let Some(lvl) = lvl
+            .to_level()
+            .filter(|lvl| log::log_enabled!(target: "sqlx::explain", *lvl))
+        {
+            let mut summary = parse_query_summary(&self.sql);
+
+            let sql = if summary != self.sql {
+                summary.push_str(" …");
+                format!(
+                    "\n\n{}\n",
+                    sqlformat::format(
+                        &self.sql,
+                        &sqlformat::QueryParams::None,
+                        sqlformat::FormatOptions::default()
+                    )
+                )
+            } else {
+                String::new()
+            };
+
+            log::logger().log(
+                &log::Record::builder()
+                    .args(format_args!(
+                        "{}; program:{:?}, unknown_operations:{:?}, results: {:?}{}",
+                        summary, self.program, self.unknown_operations, self.results, sql
+                    ))
+                    .level(lvl)
+                    .module_path_static(Some("sqlx::explain"))
+                    .target("sqlx::explain")
+                    .build(),
+            );
+        }
+    }
+}
+
+impl<'q, O: Debug + Hash + Eq, R: Debug + Hash + Eq, P: Debug> Drop
+    for QueryPlanLogger<'q, O, R, P>
+{
+    fn drop(&mut self) {
+        self.finish();
+    }
+}
+
 fn parse_query_summary(sql: &str) -> String {
     // For now, just take the first 4 words
     sql.split_whitespace()