src/Pure/Concurrent/thread_attributes.ML

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/Pure/Concurrent/thread_attributes.ML	Sat Apr 09 14:00:23 2016 +0200
@@ -0,0 +1,63 @@
+(*  Title:      Pure/Concurrent/thread_attributes.ML
+    Author:     Makarius
+
+Thread attributes for interrupt handling.
+*)
+
+signature THREAD_ATTRIBUTES =
+sig
+  type attributes = Thread.Thread.threadAttribute list
+  val no_interrupts: attributes
+  val test_interrupts: attributes
+  val public_interrupts: attributes
+  val private_interrupts: attributes
+  val sync_interrupts: attributes -> attributes
+  val safe_interrupts: attributes -> attributes
+  val with_attributes: attributes -> (attributes -> 'a) -> 'a
+  val uninterruptible: ((('c -> 'd) -> 'c -> 'd) -> 'a -> 'b) -> 'a -> 'b
+end;
+
+structure Thread_Attributes: THREAD_ATTRIBUTES =
+struct
+
+type attributes = Thread.Thread.threadAttribute list;
+
+val no_interrupts =
+  [Thread.Thread.EnableBroadcastInterrupt false,
+    Thread.Thread.InterruptState Thread.Thread.InterruptDefer];
+
+val test_interrupts =
+  [Thread.Thread.EnableBroadcastInterrupt false,
+    Thread.Thread.InterruptState Thread.Thread.InterruptSynch];
+
+val public_interrupts =
+  [Thread.Thread.EnableBroadcastInterrupt true,
+    Thread.Thread.InterruptState Thread.Thread.InterruptAsynchOnce];
+
+val private_interrupts =
+  [Thread.Thread.EnableBroadcastInterrupt false,
+    Thread.Thread.InterruptState Thread.Thread.InterruptAsynchOnce];
+
+val sync_interrupts = map
+  (fn x as Thread.Thread.InterruptState Thread.Thread.InterruptDefer => x
+    | Thread.Thread.InterruptState _ => Thread.Thread.InterruptState Thread.Thread.InterruptSynch
+    | x => x);
+
+val safe_interrupts = map
+  (fn Thread.Thread.InterruptState Thread.Thread.InterruptAsynch =>
+      Thread.Thread.InterruptState Thread.Thread.InterruptAsynchOnce
+    | x => x);
+
+fun with_attributes new_atts e =
+  let
+    val orig_atts = safe_interrupts (Thread.Thread.getAttributes ());
+    val result = Exn.capture (fn () =>
+      (Thread.Thread.setAttributes (safe_interrupts new_atts); e orig_atts)) ();
+    val _ = Thread.Thread.setAttributes orig_atts;
+  in Exn.release result end;
+
+fun uninterruptible f x =
+  with_attributes no_interrupts (fn atts =>
+    f (fn g => fn y => with_attributes atts (fn _ => g y)) x);
+
+end;