solve threads...

This commit is contained in:
itiB
2020-10-31 20:25:22 +09:00
parent 352ad7bb09
commit 96e379cc52
4 changed files with 19 additions and 22 deletions

View File

@@ -83,14 +83,3 @@ fn read_csv(filename: &str) -> Vec<Vec<String>> {
return ret; return ret;
} }
/// Argsから言語情報を読み取り Lang を返す
pub fn get_lang() -> Lang {
let lang: String = singleton().args.value_of("lang").unwrap_or("").to_string();
match &*lang {
"Ja" | "ja" => Lang::Ja,
"En" | "en" => Lang::En,
_ => Lang::En,
}
}

View File

@@ -27,7 +27,7 @@ impl PowerShell {
} }
let message = MESSAGES.lock().unwrap(); let message = MESSAGES.lock().unwrap();
println!("{}", message.return_message("4103")); println!("{}", message.get("4103"));
let default = String::from(""); let default = String::from("");
let commandline = event_data.get("ContextInfo").unwrap_or(&default); let commandline = event_data.get("ContextInfo").unwrap_or(&default);
@@ -53,7 +53,7 @@ impl PowerShell {
return; return;
} }
let message = MESSAGES.lock().unwrap(); let message = MESSAGES.lock().unwrap();
println!("{}", message.return_message("4104")); println!("{}", message.get("4104"));
let default = String::from(""); let default = String::from("");
let path = event_data.get("Path").unwrap().to_string(); let path = event_data.get("Path").unwrap().to_string();

View File

@@ -1,5 +1,5 @@
extern crate lazy_static; extern crate lazy_static;
use crate::detections::configs::{get_lang, Lang}; use crate::detections::configs::{singleton, Lang};
use crate::models::rule::MessageText; use crate::models::rule::MessageText;
use lazy_static::lazy_static; use lazy_static::lazy_static;
use std::collections::HashMap; use std::collections::HashMap;
@@ -29,12 +29,12 @@ impl Message {
} }
/// メッセージを設定 /// メッセージを設定
pub fn insert_message(&mut self, error_code: String, message: MessageText) { pub fn insert(&mut self, error_code: String, message: MessageText) {
self.map.insert(error_code, message); self.map.insert(error_code, message);
} }
/// メッセージを返す /// メッセージを返す
pub fn return_message(&self, message_num: &str) -> &MessageText { pub fn get(&self, message_num: &str) -> &MessageText {
self.map self.map
.get(message_num) .get(message_num)
.unwrap_or(self.map.get("undefined").unwrap()) .unwrap_or(self.map.get("undefined").unwrap())
@@ -52,11 +52,22 @@ impl fmt::Display for MessageText {
} }
} }
/// Argsから言語情報を読み取り Lang を返す
pub fn get_lang() -> Lang {
let lang: String = singleton().args.value_of("lang").unwrap_or("").to_string();
match &*lang {
"Ja" | "ja" => Lang::Ja,
"En" | "en" => Lang::En,
_ => Lang::En,
}
}
#[test] #[test]
fn test_create_and_read_message() { fn test_create_and_read_message() {
let mut error_message = Message::new(); let mut error_message = Message::new();
error_message.insert_message( error_message.insert(
"4103".to_string(), "4103".to_string(),
MessageText { MessageText {
ja: "パイプライン実行をしています".to_string(), ja: "パイプライン実行をしています".to_string(),
@@ -64,10 +75,7 @@ fn test_create_and_read_message() {
}, },
); );
let display = format!( let display = format!("{}", format_args!("{}", error_message.get("4103")));
"{}",
format_args!("{}", error_message.return_message("4103"))
);
assert_eq!(display, "Execute pipeline") assert_eq!(display, "Execute pipeline")
} }

View File

@@ -20,7 +20,7 @@ fn main() -> Result<(), DeError> {
let mut message = MESSAGES.lock().unwrap(); let mut message = MESSAGES.lock().unwrap();
if let Some(messages) = _rule.rule.messages { if let Some(messages) = _rule.rule.messages {
for (key, texts) in messages { for (key, texts) in messages {
message.insert_message(key, texts); message.insert(key, texts);
} }
} }
} }