doryan
/
AIT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: rewrite function logic with use custom Error type

This commit is contained in:
doryan 2024-08-10 01:10:43 +04:00
parent cad9823c83
commit 5a8a4b00ce
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/controller/model_utils/hamming_code_seven_four.rs
View File

@ -18,27 +18,25 @@ static SYNDROMES: LazyLock<HashMap<usize, (bool, bool, bool)>> = LazyLock::new(|
]) ])
}); });
pub fn hamming(raw_input: String, mode: HammingMode) -> Result<String, String> { pub fn hamming(raw_input: String, mode: HammingMode) -> Result<String> {
let length_of_code: usize = mode.clone() as usize; let length_of_code: usize = mode.clone() as usize;
let prepared_input: String = processing_input(&raw_input); let prepared_input: String = processing_input(&raw_input);
let (fc, sc): (bool, bool) = let (first_condition, second_condition): (bool, bool) =
check_correct_binary_code(&raw_input, &prepared_input, length_of_code); check_correct_binary_code(&raw_input, &prepared_input, length_of_code);
if !fc || !sc { if raw_input.is_empty() {
Err("Ошибка. Проверьте корректность ввода.".to_string()) Err("Введите код.".into())
} else if !first_condition || !second_condition {
Err("Проверьте корректность ввода.".into())
} else { } else {
let mut data: String = String::new();
let prepared_data: Vec<u8> = from_string_to_vec_bits(prepared_input); let prepared_data: Vec<u8> = from_string_to_vec_bits(prepared_input);
match mode { match mode {
HammingMode::Encrypt => hamming_encrypt_data(&prepared_data, &mut data, length_of_code), HammingMode::Encrypt => Ok(hamming_encrypt_data(&prepared_data, length_of_code)),
HammingMode::Decrypt => hamming_decrypt_data(&prepared_data, &mut data, length_of_code), HammingMode::Decrypt => Ok(hamming_decrypt_data(&prepared_data, length_of_code)),
} }
Ok(data)
} }
} }