wdes/security
1
0
Fork 0
Code Actions Packages Releases Activity

Improve the loop

Browse Source
This commit is contained in:
William Desportes
2024-06-26 19:13:37 +02:00
parent 7209aed6e1
commit e28b8cb945
1 changed files with 49 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
snow-scanner/src/main.rs
View File

@ -460,8 +460,10 @@ fn main() -> Result<()> {
.execute("SELECT 0 WHERE 0;", named_params! {}) .execute("SELECT 0 WHERE 0;", named_params! {})
.expect("Failed to initialize database"); .expect("Failed to initialize database");
thread::spawn(move || loop { thread::spawn(move || {
let conn = get_connection(db_file.as_str()); let conn = get_connection(db_file.as_str());
loop {
let mut stmt = conn.prepare("SELECT task_group_id, cidr FROM scan_tasks WHERE started_at IS NULL ORDER BY created_at ASC").unwrap(); let mut stmt = conn.prepare("SELECT task_group_id, cidr FROM scan_tasks WHERE started_at IS NULL ORDER BY created_at ASC").unwrap();
let mut rows = stmt.query(named_params! {}).unwrap(); let mut rows = stmt.query(named_params! {}).unwrap();
println!("Waiting for jobs"); println!("Waiting for jobs");
@ -508,6 +510,7 @@ fn main() -> Result<()> {
let two_hundred_millis = Duration::from_millis(500); let two_hundred_millis = Duration::from_millis(500);
thread::sleep(two_hundred_millis); thread::sleep(two_hundred_millis);
}
}); });
rouille::start_server(server_address, move |request| { rouille::start_server(server_address, move |request| {