Merge pull request #33 from reddevilmidzy/api

링크 확인 작업 취소 API 추가 및 링크 확인 작업 비동기 처리

Author: reddevilmidzy

Cargo.lock

@@ -13,6 +13,8 @@ chrono = "0.4"
 regex = "1"
 reqwest = { version = "0.12", features = ["json"] }
 tokio = { version = "1", features = ["full"] }
+tokio-util = "0.7"
 git2 = "0.20"
 serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
+once_cell = "1.18"

Cargo.toml

@@ -1,8 +1,8 @@
-use queensac::check_repository_links;
+use queensac::{cancel_repository_checker, check_repository_links};
 
 use axum::{
     Json, Router,
-    routing::{get, post},
+    routing::{delete, get, post},
     serve,
 };
 use serde::Deserialize;
@@ -32,6 +32,16 @@ async fn check_handler(Json(payload): Json<CheckRequest>) -> &'static str {
     "Repository checker started"
 }
 
+#[derive(Deserialize)]
+struct CancelRequest {
+    repo_url: String,
+}
+
+async fn cancel_handler(Json(payload): Json<CancelRequest>) -> &'static str {
+    cancel_repository_checker(&payload.repo_url).await;
+    "Repository checker cancelled"
+}
+
 #[tokio::main]
 async fn main() {
     let app = app();
@@ -45,4 +55,5 @@ fn app() -> Router {
         .route("/", get(|| async { "Sacrifice the Queen!!" }))
         .route("/health", get(health_check))
         .route("/check", post(check_handler))
+        .route("/cancel", delete(cancel_handler))
 }

src/main.rs

@@ -4,37 +4,69 @@
 use crate::git;
 use crate::link::{LinkCheckResult, check_link};
 use chrono::Local;
+use once_cell::sync::Lazy;
+use std::collections::HashMap;
+use std::sync::Mutex;
 use std::time::Duration;
+use tokio_util::sync::CancellationToken;
+
+static REPO_TASKS: Lazy<Mutex<HashMap<String, CancellationToken>>> =
+    Lazy::new(|| Mutex::new(HashMap::new()));
 
 pub async fn check_repository_links(repo_url: &str, interval_duration: Duration) {
+    let token = CancellationToken::new();
+    {
+        let mut map = REPO_TASKS.lock().unwrap();
+        map.insert(repo_url.to_string(), token.clone());
+    }
     let mut interval = tokio::time::interval(interval_duration);
-
     loop {
-        interval.tick().await;
-        let current_time = Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
-        println!("[{}] 링크 확인 중", current_time);
-
-        match git::extract_links_from_repo_url(repo_url) {
-            Ok(links) => {
-                let mut handles = Vec::new();
-                for link in links {
-                    let handle = tokio::spawn(async move {
-                        let result = check_link(&link).await;
-                        (link, result)
-                    });
-                    handles.push(handle);
-                }
-
-                for handle in handles {
-                    if let Ok((link, LinkCheckResult::Invalid(message))) = handle.await {
-                        println!("유효하지 않은 링크: '{}', 실패 원인: {}", link, message);
+        tokio::select! {
+            _ = interval.tick() => {
+                let current_time = Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
+                println!("[{}] 링크 확인 중", current_time);
+                match git::extract_links_from_repo_url(repo_url) {
+                    Ok(links) => {
+                        let mut handles = Vec::new();
+                        for link in links {
+                            let handle = tokio::spawn(async move {
+                                let result = check_link(&link).await;
+                                (link, result)
+                            });
+                            handles.push(handle);
+                        }
+                        for handle in handles {
+                            if let Ok((link, LinkCheckResult::Invalid(message))) = handle.await {
+                                println!("유효하지 않은 링크: '{}', 실패 원인: {}", link, message);
+                            }
+                        }
                     }
+                    Err(e) => println!("Repository 처리 중 오류 발생: {}", e),
                 }
+                println!("링크 확인 완료. 다음 간격 대기...");
+            },
+            // This branch listens for the `CancellationToken` to be triggered,
+            // allowing the periodic check loop to terminate cleanly upon cancellation.
+            _ = token.cancelled() => {
+                println!("[{}] 체크 작업이 취소되었습니다.", repo_url);
+                break;
             }
-            Err(e) => println!("Repository 처리 중 오류 발생: {}", e),
         }
+    }
+    let mut map = REPO_TASKS.lock().unwrap();
+    map.remove(repo_url);
+}
 
-        println!("링크 확인 완료. 다음 간격 대기...");
+pub async fn cancel_repository_checker(repo_url: &str) {
+    let token = {
+        let mut map = REPO_TASKS.lock().unwrap();
+        map.remove(repo_url)
+    };
+    if let Some(token) = token {
+        token.cancel();
+        println!("[{}] 체크 작업 취소 요청됨.", repo_url);
+    } else {
+        println!("[{}] 체크 작업이 존재하지 않습니다.", repo_url);
     }
 }