Prewarmer (#435)

* Prewarmer

* hmm

* Tests

* default

* fix test

* Correct configuration

* Added minimal config example

* remove connect_timeout

src/plugins/query_logger.rs

@@ -5,44 +5,33 @@ use crate::{
     plugins::{Plugin, PluginOutput},
     query_router::QueryRouter,
 };
-use arc_swap::ArcSwap;
 use async_trait::async_trait;
 use log::info;
-use once_cell::sync::Lazy;
 use sqlparser::ast::Statement;
-use std::sync::Arc;
 
-static ENABLED: Lazy<ArcSwap<bool>> = Lazy::new(|| ArcSwap::from_pointee(false));
-
-pub struct QueryLogger;
-
-pub fn setup() {
-    ENABLED.store(Arc::new(true));
-
-    info!("Logging queries to stdout");
-}
-
-pub fn disable() {
-    ENABLED.store(Arc::new(false));
-}
-
-pub fn enabled() -> bool {
-    **ENABLED.load()
+pub struct QueryLogger<'a> {
+    pub enabled: bool,
+    pub user: &'a str,
+    pub db: &'a str,
 }
 
 #[async_trait]
-impl Plugin for QueryLogger {
+impl<'a> Plugin for QueryLogger<'a> {
     async fn run(
         &mut self,
         _query_router: &QueryRouter,
         ast: &Vec<Statement>,
     ) -> Result<PluginOutput, Error> {
+        if !self.enabled {
+            return Ok(PluginOutput::Allow);
+        }
+
         let query = ast
             .iter()
             .map(|q| q.to_string())
             .collect::<Vec<String>>()
             .join("; ");
-        info!("{}", query);
+        info!("[pool: {}][user: {}] {}", self.user, self.db, query);
 
         Ok(PluginOutput::Allow)
     }