1 files changed, 16 insertions, 9 deletions

diff --git a/src/feature/jisoku.rs b/src/feature/jisoku.rs
index 70bc886..8f88e81 100644
--- a/src/feature/jisoku.rs
+++ b/src/feature/jisoku.rs
@@ -1,16 +1,15 @@
 extern crate telegram_bot;
 use telegram_bot::{Api, Message, MessageType};
-use std::sync::{Arc, Mutex};
 use std::thread;
 use std::time::Duration;
-
-use std::ascii::AsciiExt;
+use std::sync::Arc;
+use std::sync::atomic::{AtomicBool, Ordering};
 
 use feature::FeatureResult;
 use feature::Feature;
 
 pub struct Jisoku {
-    running: bool,
+    running: Arc<AtomicBool>,
 }
 
 const MESSAGES: [&'static str; 5] = ["JISOKU METER FURIKIRI UBAU GAME",
@@ -33,15 +32,24 @@ impl Feature for Jisoku {
             return Ok(FeatureResult::Skip);
         }
 
-        if self.running {
+        if self.running.compare_and_swap(false, true, Ordering::SeqCst) {
             return Ok(FeatureResult::Skip);
         }
-        self.running = true;
+        let running = self.running.clone();
         thread::spawn(move || {
             for s in MESSAGES.into_iter() {
-                a.send_message(m.chat.id(), (*s).to_owned(), None, None, None, None);
+                if let Err(e) = a.send_message(m.chat.id(),
+                                               (*s).to_owned(),
+                                               None,
+                                               None,
+                                               None,
+                                               None) {
+                    println!("{}", e);
+                    break;
+                };
                 thread::sleep(Duration::new(1, 0));
             }
+            running.store(false, Ordering::SeqCst);
         });
         Ok(FeatureResult::Handled)
     }
@@ -49,7 +57,6 @@ impl Feature for Jisoku {
 
 impl Jisoku {
     pub fn new() -> Jisoku {
-        let mut j = Jisoku { running: false };
-        j
+        Jisoku { running: Arc::new(AtomicBool::new(false)) }
     }
 }