emit stats

kong/dns/client.lua

@@ -652,11 +652,7 @@ end
 
 
 function _M.stats()
-  local stats = {}
-  for k, v in pairs(dns_client.stats) do
-    stats[format_key(k)] = v
-  end
-  return stats
+  return dns_client.stats:emit(format_key)
 end
 
 

kong/dns/stats.lua

@@ -1,3 +1,7 @@
+local tb_new = require(table.new)
+
+
+local pairs = pairs
 local setmetatable = setmetatable
 
 
@@ -14,25 +18,39 @@ function _M.new()
 end
 
 
-function _M:incr(name, key)
+function _M:_get_stats(name)
   local stats = self.stats
 
   if not stats[name] then
-    stats[name] = {}
+    stats[name] = tb_new(0, 6)
   end
 
-  stats[name][key] = (stats[name][key] or 0) + 1
+  return stats[name]
+end
+
+
+function _M:incr(name, key)
+  local stats = self:_get_stats(name)
+
+  stats[key] = (stats[key] or 0) + 1
 end
 
 
 function _M:set(name, key, value)
-  local stats = self.stats
+  local stats = self:_get_stats(name)
 
-  if not stats[name] then
-    stats[name] = {}
+  stats[key] = value
+end
+
+
+function _M:emit(fmt)
+  local output = {}
+
+  for k, v in pairs(self.stats) do
+    output[fmt(k)] = v
   end
 
-  stats[name][key] = value
+  return output
 end