1 files changed, 42 insertions, 0 deletions
diff --git a/lua/clever_f/repeat_resolver.lua b/lua/clever_f/repeat_resolver.lua
index 5e1dc41..c62a86f 100644
--- a/lua/clever_f/repeat_resolver.lua
+++ b/lua/clever_f/repeat_resolver.lua
@@ -34,6 +34,22 @@ local function require_policy(service)
   return service
 end
 
+local function require_clock(clock)
+  if clock ~= nil and (type(clock) ~= "table"
+      or type(clock.read_time_ms) ~= "function")
+  then
+    fail("RepeatResolver clock must provide read_time_ms", 3)
+  end
+  return clock
+end
+
+local function finite_time(value)
+  return type(value) == "number"
+    and value == value
+    and value > -math.huge
+    and value < math.huge
+end
+
 local resolver_metatable = {
   __index = RepeatResolver,
   __newindex = function()
@@ -64,6 +80,7 @@ function RepeatResolver.new(options)
 
   local resolver = setmetatable({}, resolver_metatable)
   resolver_records[resolver] = {
+    clock = require_clock(options.clock or options.time_provider or options.host),
     policy = require_policy(options.policy or options.policy_service),
     state = state,
   }
@@ -109,6 +126,27 @@ end
 
 RepeatResolver.sample_repeat_timeout = RepeatResolver.sample_repeat_timeout_ms
 
+function RepeatResolver:evaluate_timeout()
+  local timeout = self:sample_repeat_timeout_ms()
+  if timeout == 0 then
+    return M.Decision.REPEAT, nil
+  end
+
+  local clock = resolver_records[self].clock
+  if clock == nil then
+    fail("RepeatResolver requires a clock for positive repeat timeout", 2)
+  end
+  local current_time = clock:read_time_ms()
+  if not finite_time(current_time) then
+    fail("repeat clock must return finite milliseconds", 2)
+  end
+  local elapsed_ms = current_time - resolver_records[self].state.repeat_timestamp_ms
+  return nil, elapsed_ms
+end
+
+RepeatResolver.check_timeout = RepeatResolver.evaluate_timeout
+RepeatResolver.resolve_timeout = RepeatResolver.evaluate_timeout
+
 function M.new(options)
   return RepeatResolver.new(options)
 end
@@ -131,6 +169,10 @@ function M.sample_repeat_timeout_ms(options)
   return RepeatResolver.new(options):sample_repeat_timeout_ms()
 end
 
+function M.evaluate_timeout(options)
+  return RepeatResolver.new(options):evaluate_timeout()
+end
+
 setmetatable(M, {
   __call = function(_, options)
     return RepeatResolver.new(options)