Compare commits
No commits in common. "441ce5c6304141e22a79cf3996f78746d50fb12e" and "5a5b5fb4a351e47f527ca1f8543b8f948a0dda4d" have entirely different histories.
441ce5c630
...
5a5b5fb4a3
File diff suppressed because it is too large
Load Diff
|
@ -15,5 +15,4 @@ log = "0.4.17"
|
|||
mysql = "23.0.1"
|
||||
sprintf = "0.1.3"
|
||||
termsize = "0.1.6"
|
||||
thiserror = "2.0.12"
|
||||
time = "0.3.20"
|
||||
|
|
|
@ -9,23 +9,15 @@ use crate::{
|
|||
|
||||
use super::command::{Cli, Commands};
|
||||
|
||||
use thiserror::Error;
|
||||
|
||||
#[derive(Debug, Error)]
|
||||
enum CommandError {
|
||||
#[error("Empty search provided.")]
|
||||
EmptySearch(),
|
||||
}
|
||||
|
||||
pub fn process_args(args: Cli, db: Db) -> Result<(), Box<dyn std::error::Error>> {
|
||||
pub fn process_args(args: Cli, db: Db) {
|
||||
match args.mode {
|
||||
Commands::Merch { search } => {
|
||||
let mut merchant_service: MerchantService = container::get_merchant_service(db);
|
||||
merchant_service.get_merchant(&search.ok_or(CommandError::EmptySearch())?);
|
||||
merchant_service.get_merchant(&search.unwrap());
|
||||
}
|
||||
Commands::Schema { search } => {
|
||||
let mut schema_service: SchemaService = container::get_schema_service(db);
|
||||
schema_service.get_columns(&search.ok_or(CommandError::EmptySearch())?)
|
||||
schema_service.get_columns(&search.unwrap())
|
||||
}
|
||||
Commands::Filter {
|
||||
filter_id,
|
||||
|
@ -37,7 +29,7 @@ pub fn process_args(args: Cli, db: Db) -> Result<(), Box<dyn std::error::Error>>
|
|||
if config {
|
||||
filter_service.get_filter_configs(&filter_id)
|
||||
} else if log {
|
||||
filter_service.get_filter_log(&filter_id, all)
|
||||
filter_service.get_filter_log(&filter_id)
|
||||
} else {
|
||||
filter_service.get_filter(&filter_id, all)
|
||||
}
|
||||
|
@ -55,6 +47,4 @@ pub fn process_args(args: Cli, db: Db) -> Result<(), Box<dyn std::error::Error>>
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
|
|
@ -1,50 +1,35 @@
|
|||
use core::fmt;
|
||||
use mysql::{Opts, Pool, PooledConn};
|
||||
use std::{
|
||||
env,
|
||||
error::Error,
|
||||
path::PathBuf,
|
||||
sync::{Arc, Mutex},
|
||||
};
|
||||
|
||||
use directories::BaseDirs;
|
||||
|
||||
#[derive(Debug)]
|
||||
struct InitError(String);
|
||||
|
||||
impl fmt::Display for InitError {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
write!(f, "{}", self.0)
|
||||
}
|
||||
}
|
||||
|
||||
impl Error for InitError {}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct Db {
|
||||
pool: Arc<Mutex<Pool>>,
|
||||
}
|
||||
|
||||
impl Db {
|
||||
pub fn initialize(env: String) -> Result<Self, Box<dyn std::error::Error>> {
|
||||
let base_dir: BaseDirs = BaseDirs::new()
|
||||
.ok_or_else(|| Box::new(InitError("No config folder found.".to_string())))?;
|
||||
pub fn initialize(env: String) -> Result<Self, mysql::Error> {
|
||||
let base_dir: BaseDirs = match BaseDirs::new() {
|
||||
Some(dirs) => dirs,
|
||||
None => panic!("No config folder found."),
|
||||
};
|
||||
|
||||
let config_file_path: PathBuf = base_dir.config_dir().join("rcc/").join(env);
|
||||
|
||||
dotenv::from_path(config_file_path.as_path()).map_err(|_| {
|
||||
Box::new(InitError(format!(
|
||||
"Failed to load dotenv file '{}'",
|
||||
config_file_path.display()
|
||||
)))
|
||||
})?;
|
||||
match dotenv::from_path(config_file_path.as_path()) {
|
||||
Ok(env) => env,
|
||||
Err(_) => panic!("Could not load .env file {:?}", config_file_path.as_path()),
|
||||
};
|
||||
|
||||
let location: String = env::var("LOCATION").unwrap_or_else(|_| "localhost".to_string());
|
||||
let user: String = env::var("USERNAME")
|
||||
.map_err(|_| Box::new(InitError("Missing USERNAME in dotenv file.".to_string())))?;
|
||||
let user: String = env::var("USERNAME").unwrap();
|
||||
let port: String = env::var("PORT").unwrap_or_else(|_| "3306".to_string());
|
||||
let password: String = env::var("PASSWORD")
|
||||
.map_err(|_| Box::new(InitError("Missing PASSWORD in dotenv file.".to_string())))?;
|
||||
let password: String = env::var("PASSWORD").unwrap();
|
||||
let db_name: String = env::var("DB").unwrap_or_else(|_| "efulfilment".to_string());
|
||||
let url: String = format!(
|
||||
"mysql://{}:{}@{}:{}/{}",
|
||||
|
@ -60,7 +45,7 @@ impl Db {
|
|||
}
|
||||
|
||||
pub fn get_connection(&self) -> Result<PooledConn, mysql::Error> {
|
||||
let pool = self.pool.lock()?;
|
||||
let pool = self.pool.lock().unwrap();
|
||||
pool.get_conn()
|
||||
}
|
||||
}
|
||||
|
|
|
@ -15,6 +15,7 @@ fn main() -> std::result::Result<(), Box<dyn std::error::Error>> {
|
|||
let env: &String = &Cli::parse().env.unwrap_or("local".to_string());
|
||||
let db: Db = Db::initialize(env.to_string())?;
|
||||
|
||||
controller::process_args(args, db)?;
|
||||
controller::process_args(args, db);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
|
|
@ -15,7 +15,8 @@ impl FileRepo {
|
|||
pub fn find_page(&mut self, file_id: &usize) -> Result<Vec<Page>, mysql::Error> {
|
||||
let mut connection: mysql::PooledConn = self.db_pool.get_connection()?;
|
||||
let stat = connection
|
||||
.prep("SELECT m_id, description, file FROM global_data.effi_file_mapping WHERE file_id = :file_id")?;
|
||||
.prep("SELECT m_id, description, file FROM global_data.effi_file_mapping WHERE file_id = :file_id")
|
||||
.unwrap();
|
||||
|
||||
connection.exec_map(
|
||||
stat,
|
||||
|
|
|
@ -22,7 +22,8 @@ impl FilterRepo {
|
|||
FROM filter f
|
||||
JOIN global_data.filter_modules fm ON fm.filter_module_id = f.filter_module_id
|
||||
WHERE f.filter_id = :filter_id
|
||||
")?;
|
||||
")
|
||||
.unwrap();
|
||||
|
||||
let params = params! {"filter_id" => filter_id};
|
||||
connection.exec_map(
|
||||
|
@ -51,7 +52,8 @@ impl FilterRepo {
|
|||
JOIN global_data.effi_user t2 on t1.upd_user = t2.user_id
|
||||
WHERE t1.filter_id = :filter_id
|
||||
",
|
||||
)?;
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
connection.exec_map(
|
||||
stat,
|
||||
|
@ -68,13 +70,15 @@ impl FilterRepo {
|
|||
|
||||
pub fn find_filter_log(&mut self, filter_id: &usize) -> Result<Vec<FilterLog>, mysql::Error> {
|
||||
let mut connection: PooledConn = self.db.get_connection()?;
|
||||
let stat = connection.prep(
|
||||
let stat = connection
|
||||
.prep(
|
||||
"
|
||||
SELECT t1.run_ts, t1.error_code, t1.error_msg, t1.mysql_error
|
||||
FROM filter_run_error_log t1
|
||||
WHERE t1.filter_id = :filter_id
|
||||
",
|
||||
)?;
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
connection.exec_map(
|
||||
stat,
|
||||
|
|
|
@ -17,7 +17,8 @@ impl JobRepo {
|
|||
|
||||
pub fn find_by_id(&mut self, job_id: &usize) -> Result<Vec<Job>, mysql::Error> {
|
||||
let mut connection: mysql::PooledConn = self.db.get_connection()?;
|
||||
let stat = connection.prep(
|
||||
let stat = connection
|
||||
.prep(
|
||||
"SELECT
|
||||
jm.file_name, jm.description
|
||||
FROM jobs j
|
||||
|
@ -25,7 +26,8 @@ impl JobRepo {
|
|||
WHERE
|
||||
j.job_id = :job_id
|
||||
",
|
||||
)?;
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
connection.exec_map(
|
||||
stat,
|
||||
|
@ -39,7 +41,8 @@ impl JobRepo {
|
|||
|
||||
pub fn find_log(&mut self, job_id: &usize) -> Result<Vec<JobLog>, mysql::Error> {
|
||||
let mut connection: mysql::PooledConn = self.db.get_connection()?;
|
||||
let stat = connection.prep(
|
||||
let stat = connection
|
||||
.prep(
|
||||
"
|
||||
SELECT jea.content, jea.upd_ts
|
||||
FROM jobs j
|
||||
|
@ -48,7 +51,8 @@ impl JobRepo {
|
|||
WHERE
|
||||
j.job_id = :job_id
|
||||
",
|
||||
)?;
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
connection.exec_map(stat, params! {"job_id" => job_id}, |(content, upd_ts)| {
|
||||
JobLog { content, upd_ts }
|
||||
|
|
|
@ -14,7 +14,8 @@ impl MerchantRepo {
|
|||
pub fn find_by_name_or_id(&mut self, search: &str) -> Result<Vec<Merchant>, mysql::Error> {
|
||||
let mut connection = self.db_pool.get_connection()?;
|
||||
let stat = connection
|
||||
.prep("SELECT m_name, m_id FROM global_data.merchant WHERE m_id = :search OR m_name LIKE :search2;")?;
|
||||
.prep("SELECT m_name, m_id FROM global_data.merchant WHERE m_id = :search OR m_name LIKE :search2;")
|
||||
.unwrap();
|
||||
|
||||
let search_like = format!("%{}%", search);
|
||||
|
||||
|
|
|
@ -14,7 +14,8 @@ impl SchemaRepo {
|
|||
pub fn find_by_column(&mut self, column_name: &str) -> Result<Vec<Schema>, mysql::Error> {
|
||||
let mut connection: mysql::PooledConn = self.db.get_connection()?;
|
||||
let stat = connection
|
||||
.prep("SELECT TABLE_NAME, COLUMN_NAME, COLUMN_TYPE FROM INFORMATION_SCHEMA.COLUMNS WHERE column_name LIKE :c_name")?;
|
||||
.prep("SELECT TABLE_NAME, COLUMN_NAME, COLUMN_TYPE FROM INFORMATION_SCHEMA.COLUMNS WHERE column_name LIKE :c_name")
|
||||
.unwrap();
|
||||
|
||||
connection.exec_map(
|
||||
stat,
|
||||
|
|
|
@ -73,7 +73,7 @@ impl FilterService {
|
|||
};
|
||||
}
|
||||
|
||||
pub fn get_filter_log(&mut self, filter_id: &usize, all: bool) {
|
||||
pub fn get_filter_log(&mut self, filter_id: &usize) {
|
||||
let result = self.repo.find_filter_log(filter_id);
|
||||
match result {
|
||||
Ok(filter_log) => {
|
||||
|
@ -89,21 +89,12 @@ impl FilterService {
|
|||
match filter_log.error_code.as_str() {
|
||||
"WARNING" => {
|
||||
println!("{}", filter_log.error_msg.yellow());
|
||||
if all {
|
||||
println!("{}", filter_log.mysql_error.yellow());
|
||||
}
|
||||
}
|
||||
"ERROR" => {
|
||||
println!("{}", filter_log.error_msg.red());
|
||||
if all {
|
||||
println!("{}", filter_log.mysql_error.red());
|
||||
}
|
||||
}
|
||||
"DEBUG" => {
|
||||
println!("{}", filter_log.error_msg.green());
|
||||
if all {
|
||||
println!("{}", filter_log.error_msg.green())
|
||||
}
|
||||
}
|
||||
_ => {
|
||||
println!("{}", filter_log.error_msg);
|
||||
|
|
Loading…
Reference in New Issue