From b13da6f3327caf114a9941fd89d79b61a06b6bc9 Mon Sep 17 00:00:00 2001 From: Jim Myhrberg Date: Mon, 18 Apr 2011 12:37:24 +0100 Subject: [PATCH] create a flush buffer #at_exit callback to ensure any buffered messages are flushed to Redis on process exit --- lib/redistat.rb | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/lib/redistat.rb b/lib/redistat.rb index 23ca423..4ff2389 100644 --- a/lib/redistat.rb +++ b/lib/redistat.rb @@ -20,6 +20,7 @@ require 'redistat/mixins/options' require 'redistat/mixins/synchronize' require 'redistat/mixins/database' require 'redistat/mixins/date_helper' + require 'redistat/connection' require 'redistat/buffer' require 'redistat/collection' @@ -36,6 +37,7 @@ require 'redistat/version' require 'redistat/core_ext' + module Redistat KEY_NEXT_ID = ".next_id" @@ -91,3 +93,9 @@ module Redistat end end + + +# ensure buffer is flushed on program exit +Kernel.at_exit do + Redistat.buffer.flush(true) +end