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

Cleanup the not finished tasks

Browse Source
This commit is contained in:
William Desportes
2024-06-28 19:07:13 +02:00
parent d5297c775a
commit cf05597a30
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -545,6 +545,11 @@ fn main() -> Result<()> {
thread::spawn(move || { thread::spawn(move || {
let conn = get_connection(db_file.as_str()); let conn = get_connection(db_file.as_str());
// Reset scan tasks
let _ = conn.execute("UPDATE scan_tasks SET updated_at = :updated_at, still_processing_at = NULL, started_at = NULL WHERE (still_processing_at IS NOT NULL OR started_at IS NOT NULL) AND ended_at IS NULL",
named_params! {
":updated_at": Utc::now().naive_utc().to_string(),
}).unwrap();
loop { 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();