From 04fc48e406eedae26cd806064d0f3bf3b266caef Mon Sep 17 00:00:00 2001 From: FuckYou Date: Thu, 1 Dec 2016 17:01:19 +0400 Subject: [PATCH] Minor fix - Fixed runnable execution to properly set up Async environment for thread --- src/com/tofvesson/async/Async.java | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/src/com/tofvesson/async/Async.java b/src/com/tofvesson/async/Async.java index c39f8ff..f5273be 100644 --- a/src/com/tofvesson/async/Async.java +++ b/src/com/tofvesson/async/Async.java @@ -54,7 +54,19 @@ public class Async { * @param r Runnable to execute as new task. */ public Async(Runnable r){ - task = new Thread(r); // Execute thread with runnable + task = new Thread(()->{ + try { + new ThreadLocal() + .set(Async.this); // Store ThreadLocal reference to Async object + r.run(); // Execute runnable + complete = true; // Notify all threads who are checking + } catch (Throwable t1) { // Prepare for failure + if(!failed) { // Checks if task was canceled + failed = true; // Notifies all threads that task failed + t=t1; // Makes error accessible to be thrown + } + } + }); // Execute thread with runnable task.setDaemon(true); // Ensure that process dies with program task.start(); // Start task }