Move duration into ensure block since it wasn't getting set on failures

jclusso · Feb 17, 2024 · 6c9a00c · 6c9a00c
1 parent 4726232
commit 6c9a00c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/app/lib/dns_lookup.rb b/app/lib/dns_lookup.rb
@@ -13,7 +13,6 @@ def run(domain, type)
       domain = "#{domain.split('.').reverse.join('.')}.in-addr.arpa"
     end
     response = resolver.query(domain, type)
-    @duration = (Time.monotonic_now - start_time) * 1000
 
     {
       json: {
@@ -33,6 +32,8 @@ def run(domain, type)
     { json: { status: 'NOTIMP', from: @server }, zone: e.response.to_s }
   rescue Dnsruby::OtherResolvError
     { json: { status: 'OTHER ERROR', from: @server } }
+  ensure
+    @duration = (Time.monotonic_now - start_time) * 1000
   end
 
   private