diff options
Diffstat (limited to 'lib/gitlab/database/connection_timer.rb')
-rw-r--r-- | lib/gitlab/database/connection_timer.rb | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/lib/gitlab/database/connection_timer.rb b/lib/gitlab/database/connection_timer.rb new file mode 100644 index 00000000000..ef8d52ba71c --- /dev/null +++ b/lib/gitlab/database/connection_timer.rb @@ -0,0 +1,50 @@ +# frozen_string_literal: true + +module Gitlab + module Database + class ConnectionTimer + DEFAULT_INTERVAL = 3600 + RANDOMIZATION_INTERVAL = 600 + + class << self + def configure + yield self + end + + def starting_now + # add a small amount of randomization to the interval, so reconnects don't all occur at once + new(interval_with_randomization, current_clock_value) + end + + attr_writer :interval + + def interval + @interval ||= DEFAULT_INTERVAL + end + + def interval_with_randomization + interval + rand(RANDOMIZATION_INTERVAL) if interval.positive? + end + + def current_clock_value + Concurrent.monotonic_time + end + end + + attr_reader :interval, :starting_clock_value + + def initialize(interval, starting_clock_value) + @interval = interval + @starting_clock_value = starting_clock_value + end + + def expired? + interval&.positive? && self.class.current_clock_value > (starting_clock_value + interval) + end + + def reset! + @starting_clock_value = self.class.current_clock_value + end + end + end +end |