allow setting of heartbeat_key like logger, redis and tz

nickelser · Oct 8, 2016 · 8e64073 · 8e64073
1 parent e2ff979
commit 8e64073
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/lib/zhong.rb b/lib/zhong.rb
@@ -18,7 +18,7 @@
 module Zhong
   class << self
     extend Forwardable
-    attr_writer :logger, :redis
+    attr_writer :logger, :redis, :heartbeat_key
     attr_accessor :tz
 
     def_delegators :scheduler, :start, :stop, :clear, :jobs, :redis_time
@@ -29,7 +29,7 @@ def self.schedule(&block)
   end
 
   def self.scheduler
-    @scheduler ||= Scheduler.new(logger: logger, redis: redis, tz: tz)
+    @scheduler ||= Scheduler.new(logger: logger, redis: redis, tz: tz, heartbeat_key: heartbeat_key)
   end
 
   def self.any_running?(grace = 60.seconds)
@@ -68,4 +68,8 @@ def self.logger
   def self.redis
     @redis ||= Redis.new(url: ENV["REDIS_URL"])
   end
+
+  def self.heartbeat_key
+    @heartbeat_key ||= "zhong:heartbeat"
+  end
 end
diff --git a/lib/zhong/scheduler.rb b/lib/zhong/scheduler.rb
@@ -6,8 +6,7 @@ class Scheduler
       timeout: 0.5,
       grace: 15.minutes,
       long_running_timeout: 5.minutes,
-      tz: nil,
-      heartbeat_key: "zhong:heartbeat",
+      tz: nil
     }.freeze
 
     def initialize(config = {})