dboard 0.0.4 → 0.0.5

data/lib/collector.rb

@@ -11,12 +11,17 @@ module Dboard
       Collector.instance.register_source(key, instance)
     end
 
+    def self.register_after_update_callback(callback)
+      Collector.instance.register_after_update_callback(callback)
+    end
+
     def self.start
       instance.start
     end
 
     def initialize
       @sources = {}
+      @after_update_callback = lambda {}
     end
 
     def start
@@ -38,8 +43,12 @@ module Dboard
     end
 
     def update_source(source, instance)
-      data = instance.fetch
-      publish_data(source, data)
+      begin
+        data = instance.fetch
+        publish_data(source, data)
+      ensure
+        @after_update_callback.call
+      end
     rescue Exception => ex
       puts "Failed to update #{source}: #{ex.message}"
       puts ex.backtrace
@@ -49,6 +58,10 @@ module Dboard
       @sources.merge!({ key => instance })
     end
 
+    def register_after_update_callback(callback)
+      @after_update_callback = callback
+    end
+
     def publish_data(source, data)
       Publisher.publish(source, data)
     end

data/lib/version.rb

@@ -1,3 +1,3 @@
 module Dboard
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

data/spec/collector_spec.rb

@@ -13,6 +13,20 @@ describe Dboard::Collector, "register_source" do
     Dboard::Collector.instance.sources.should == { :new_relic => new_relic }
   end
 
+  it "can register an after update callback" do
+    new_relic = mock
+    new_relic.stub!(:fetch).and_return({ :db => "100%" })
+    callback = mock
+    Dboard::Collector.instance.register_after_update_callback callback
+
+    callback.should_receive(:call)
+    Dboard::Publisher.stub!(:publish)
+    Dboard::Collector.instance.update_source(:new_relic, new_relic)
+
+    # since it is a singleton, and this callbacks leaks into the other tests
+    Dboard::Collector.instance.register_after_update_callback(lambda {})
+  end
+
 end
 
 describe Dboard::Collector, "update_source" do

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: dboard
 version: !ruby/object:Gem::Version 
-  hash: 23
+  hash: 21
   prerelease: 
   segments: 
   - 0
   - 0
-  - 4
-  version: 0.0.4
+  - 5
+  version: 0.0.5
 platform: ruby
 authors: 
 - "Joakim Kolsj\xC3\xB6"
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-02-14 00:00:00 +01:00
+date: 2012-08-16 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency