src/libstd/sync/poison.rs

   1 // Copyright 2014 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 use option::Option::None;
  12 use rustrt::task::Task;
  13 use rustrt::local::Local;
  14
  15 pub struct Flag { pub failed: bool }
  16
  17 impl Flag {
  18     pub fn borrow(&mut self) -> Guard {
  19         Guard { flag: &mut self.failed, failing: failing() }
  20     }
  21 }
  22
  23 pub struct Guard<'a> {
  24     flag: &'a mut bool,
  25     failing: bool,
  26 }
  27
  28 impl<'a> Guard<'a> {
  29     pub fn check(&self, name: &str) {
  30         if *self.flag {
  31             panic!("poisoned {} - another task failed inside", name);
  32         }
  33     }
  34
  35     pub fn done(&mut self) {
  36         if !self.failing && failing() {
  37             *self.flag = true;
  38         }
  39     }
  40 }
  41
  42 fn failing() -> bool {
  43     if Local::exists(None::<Task>) {
  44         Local::borrow(None::<Task>).unwinder.unwinding()
  45     } else {
  46         false
  47     }
  48 }