refactor: extract database creation query to MySqlQueriesBuilders for…

… improved code organization
Kremilly · Jan 7, 2025 · 9e00a19 · 9e00a19
1 parent 5959f19
commit 9e00a19
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/handlers/import_handlers.rs b/src/handlers/import_handlers.rs
@@ -1,6 +1,9 @@
 use regex::Regex;
 
-use crate::constants::regexp::RegExp;
+use crate::{
+    constants::regexp::RegExp,
+    handlers::queries_builders::MySqlQueriesBuilders,
+};
 
 pub struct ImportHandlers {
     dbname: String,
@@ -27,7 +30,7 @@ impl ImportHandlers {
                 caps[2].to_string()
             };
 
-            format!("CREATE DATABASE IF NOT EXISTS `{}`;", db_name)
+            MySqlQueriesBuilders.create_database_not_exists(&db_name)
         });
 
         let dump_content = use_db_regex.replace_all(&content, |caps: &regex::Captures| {

diff --git a/src/handlers/queries_builders.rs b/src/handlers/queries_builders.rs
@@ -14,6 +14,10 @@ impl MySqlQueriesBuilders {
         format!("DROP TABLE IF EXISTS `{}`;", table)
     }
 
+    pub fn create_database_not_exists(&self, dbname: &str) -> String {
+        format!("CREATE DATABASE IF NOT EXISTS `{}`;", dbname)
+    }
+
     pub fn create_database(&self, dbname: &str) -> Result<(String, String), String> {
         let create_db = format!("CREATE DATABASE IF NOT EXISTS `{}`;\n", dbname);
         let use_db = format!("USE `{}`;", dbname);