Refactor: Extract ODBC connection options to a separate function

Co-authored-by: contact <[email protected]>

Author: cursoragent

src/webserver/database/connect.rs

@@ -10,6 +10,7 @@ use anyhow::Context;
 use futures_util::future::BoxFuture;
 use sqlx::{
     any::{Any, AnyConnectOptions, AnyKind},
+    odbc::OdbcConnectOptions,
     pool::PoolOptions,
     sqlite::{Function, SqliteConnectOptions, SqliteFunctionCtx},
     ConnectOptions, Connection, Executor,
@@ -208,10 +209,9 @@ fn set_custom_connect_options(options: &mut AnyConnectOptions, config: &AppConfi
     if let Some(sqlite_options) = options.as_sqlite_mut() {
         set_custom_connect_options_sqlite(sqlite_options, config);
     }
-	// Allow fetching very large text fields when using ODBC by removing the max column size limit
-	if let Some(odbc_options) = options.as_odbc_mut() {
-		*odbc_options = std::mem::take(odbc_options).max_column_size(None);
-	}
+    if let Some(odbc_options) = options.as_odbc_mut() {
+        set_custom_connect_options_odbc(odbc_options, config);
+    }
 }
 
 fn set_custom_connect_options_sqlite(
@@ -239,15 +239,20 @@ fn make_sqlite_fun(name: &str, f: fn(&str) -> String) -> Function {
     })
 }
 
+fn set_custom_connect_options_odbc(odbc_options: &mut OdbcConnectOptions, _config: &AppConfig) {
+    // Allow fetching very large text fields when using ODBC by removing the max column size limit
+    *odbc_options = std::mem::take(odbc_options).max_column_size(None);
+}
+
 fn set_database_password(options: &mut AnyConnectOptions, password: &str) {
     if let Some(opts) = options.as_postgres_mut() {
         *opts = take(opts).password(password);
     } else if let Some(opts) = options.as_mysql_mut() {
         *opts = take(opts).password(password);
     } else if let Some(opts) = options.as_mssql_mut() {
         *opts = take(opts).password(password);
-	} else if let Some(_opts) = options.as_odbc_mut() {
-		log::warn!(
+    } else if let Some(_opts) = options.as_odbc_mut() {
+        log::warn!(
 			"Setting a password for an ODBC connection is not supported via separate config; include credentials in the DSN or connection string"
 		);
     } else if let Some(_opts) = options.as_sqlite_mut() {