Intermediate commit from muting admin

1 files changed, 15 insertions, 9 deletions

diff --git a/src/scheduled_agent.clj b/src/scheduled_agent.clj
index 702b314..b42bb57 100644
--- a/src/scheduled_agent.clj
+++ b/src/scheduled_agent.clj
@@ -9,23 +9,29 @@
 (defn scheduled-agent
   [func period init]
   (let [pool  (Executors/newScheduledThreadPool 1)
-        r     (ref init)
+        data  (ref init)
         pfunc (runnable-proxy (fn []
                                 (try
                                  (dosync
-                                  (ref-set r (func (ensure r))))
+                                  (ref-set data (func (ensure data))))
                                  (catch Exception e
                                    (print-stack-trace e 5)))))
         future (.scheduleWithFixedDelay pool pfunc 0 period TimeUnit/SECONDS)]
-    {:pool pool
-     :data r
+    {:pool   pool
+     :data   data
      :future future
-     :func pfunc
+     :func   func
      :period period
-     :init init}))
-
-(defn cancel [{f :future}]
-  (.cancel f false))
+     :init   init}))
 
 (defn poll [{d :data}]
+  "Return current contents of agent."
   @d)
+
+(defn cancel! [{f :future}]
+  "Cancel automatic updating of agent data. Cannot be restarted."
+  (.cancel f false))
+
+(defn update! [{func :func data :data}]
+  "Synchronously update contents of agent."
+  (dosync (ref-set data (func (ensure data)))))
\ No newline at end of file