From c274d9177e4a495e7b793120dfd1ce12fa5632c7 Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Mon, 15 Apr 2013 19:40:00 +0200
Subject: o Using the returned references to wait for completion.

---
 src/main/java/io/trygvis/queue/AsyncService.java   | 13 ++++++++
 .../java/io/trygvis/queue/JpaAsyncService.java     | 37 ++++++++++++++++++++--
 2 files changed, 47 insertions(+), 3 deletions(-)

(limited to 'src/main/java/io/trygvis/queue')

diff --git a/src/main/java/io/trygvis/queue/AsyncService.java b/src/main/java/io/trygvis/queue/AsyncService.java
index f792f5e..b08db1f 100755
--- a/src/main/java/io/trygvis/queue/AsyncService.java
+++ b/src/main/java/io/trygvis/queue/AsyncService.java
@@ -2,6 +2,8 @@ package io.trygvis.queue;
 
 import org.quartz.*;
 
+import java.util.*;
+
 public interface AsyncService<QueueRef extends AsyncService.QueueRef, ExecutionRef extends AsyncService.ExecutionRef> {
 
     JpaAsyncService.JpaQueueRef registerQueue(String name, int interval, AsyncCallable callable) throws SchedulerException;
@@ -10,10 +12,21 @@ public interface AsyncService<QueueRef extends AsyncService.QueueRef, ExecutionR
 
     ExecutionRef schedule(QueueRef queue, String... args);
 
+    ExecutionRef update(ExecutionRef ref);
+
     interface QueueRef {
     }
 
     interface ExecutionRef {
+        List<String> getArguments();
+
+        Date getScheduled();
+
+        Date getLastRun();
+
+        Date getCompleted();
+
+        boolean isDone();
     }
 
     interface AsyncCallable {
diff --git a/src/main/java/io/trygvis/queue/JpaAsyncService.java b/src/main/java/io/trygvis/queue/JpaAsyncService.java
index 95d5ef3..e715ac7 100755
--- a/src/main/java/io/trygvis/queue/JpaAsyncService.java
+++ b/src/main/java/io/trygvis/queue/JpaAsyncService.java
@@ -101,6 +101,11 @@ public class JpaAsyncService implements AsyncService<JpaAsyncService.JpaQueueRef
         return new JpaExecutionRef(task);
     }
 
+    @Transactional(readOnly = true)
+    public JpaExecutionRef update(JpaExecutionRef ref) {
+        return new JpaExecutionRef(taskRepository.findOne(ref.task.getId()));
+    }
+
     public static class JpaQueueRef implements AsyncService.QueueRef {
         public final Queue queue;
 
@@ -122,6 +127,26 @@ public class JpaAsyncService implements AsyncService<JpaAsyncService.JpaQueueRef
             this.task = task;
         }
 
+        public List<String> getArguments() {
+            return Arrays.asList(task.getArguments());
+        }
+
+        public Date getScheduled() {
+            return task.getScheduled();
+        }
+
+        public Date getLastRun() {
+            return task.getLastRun();
+        }
+
+        public Date getCompleted() {
+            return task.getCompleted();
+        }
+
+        public boolean isDone() {
+            return task.isDone();
+        }
+
         public String toString() {
             return "JpaExecutionRef{" +
                     "task=" + task +
@@ -129,6 +154,12 @@ public class JpaAsyncService implements AsyncService<JpaAsyncService.JpaQueueRef
         }
     }
 
+    private static class TaskFailureException extends RuntimeException {
+        public TaskFailureException(Exception e) {
+            super(e);
+        }
+    }
+
     private class CheckTimerTask implements Runnable {
         private final AsyncCallable callable;
         private final JpaQueueRef queueRef;
@@ -149,12 +180,12 @@ public class JpaAsyncService implements AsyncService<JpaAsyncService.JpaQueueRef
                 for (final Task task : tasks) {
                     try {
                         executeTask(task);
-                    } catch (TransactionException e) {
+                    } catch (TransactionException | TaskFailureException e) {
                         log.warn("Task execution failed", e);
                     }
                 }
             } catch (Exception e) {
-                log.warn("Error while execution tasks.", e);
+                log.warn("Error while executing tasks.", e);
             }
         }
 
@@ -177,7 +208,7 @@ public class JpaAsyncService implements AsyncService<JpaAsyncService.JpaQueueRef
                         task.registerComplete(completed);
                         taskRepository.save(task);
                     } catch (Exception e) {
-                        throw new RuntimeException("Error while executing callback", e);
+                        throw new TaskFailureException(e);
                     }
                 }
             });
-- 
cgit v1.2.3