Some refactoring of the results data handling.

1 files changed, 41 insertions, 36 deletions

diff --git a/apt-panopticommon.lua b/apt-panopticommon.lua
index 5c0dfe3..aba2a2a 100644
--- a/apt-panopticommon.lua
+++ b/apt-panopticommon.lua
@@ -395,7 +395,7 @@ APT.plurals = function(e, w)
     return result
 end
 
-APT.fill = function(results)
+APT.padResults = function(results)
     for k, v in pairs{"ftp", "http", "https", "rsync"} do
         tests = results[v]
         if nil == tests then tests = {errors = 0; warnings = 0} end
@@ -410,14 +410,12 @@ APT.fill = function(results)
     return results
 end
 
-APT.collate = function(host, ip, results)
-    results = APT.fill(results)
-    local f = "results/" .. host .. "_" .. ip .. ".lua"
-    local rfile, e = io.open(f, "r")
-    if nil == rfile then I("opening " .. f .. " file - " .. e) else
-        rfile:close()
+APT.collate = function(l, host, ip, results)
+    results = APT.padResults(results)
+    local f = l .. "/" .. host .. "_" .. ip .. ".lua"
+    if APT.checkFile(f) then
         local rs = loadfile(f)()
-        rs = APT.fill(rs)
+        rs = APT.padResults(rs)
         for k, v in pairs(rs) do
             if "table" == type(v) then
                 if ("speed" == k) and (nil ~= results.speed) then
@@ -427,7 +425,11 @@ APT.collate = function(host, ip, results)
                     for i, u in pairs(v) do
                         if "table" == type(u) then
                             for h, t in pairs(u) do
-                                local a = results[k][i][h]
+                                local a = results[k]
+                                if nil == a then results[k]    = {i = {}}; a = results[k]    end
+                                a = a[i]
+                                if nil == a then results[k][i] = {h = {}}; a = results[k][i] end
+                                a = a[h]
                                 if nil == a then a = 0 end
                                 results[k][i][h] = a + t
                             end
@@ -448,8 +450,31 @@ APT.collate = function(host, ip, results)
     end
     return results
 end
-
-
+APT.collateAll = function(l, host, func)
+    results = {}
+    local f = l .. "/" .. host .. ".lua"
+    if APT.checkFile(f) then
+        results = loadfile(f)()
+        results = APT.padResults(results)
+        if nil ~= func then func(results) end
+        local v = APT.mirrors[host]
+        if nil ~= v then
+            local IPs = v.IPs
+            for i, u in pairs(IPs) do
+                if "table" == type(u) then
+                    for h, t in pairs(u) do
+                        results = APT.collate(l, host, h, results)
+                        if nil ~= func then func(results, h) end
+                    end
+                else
+                    results = APT.collate(l, host, i, results)
+                    if nil ~= func then func(results, i) end
+                end
+            end
+        end
+    end
+    return results
+end
 
 
 APT.now = 0
@@ -471,7 +496,6 @@ local DSSe = 'DS:URLSanityErrors:GAUGE:' .. hb .. ':0:U'
 local DSSw = 'DS:URLSanityWarnings:GAUGE:' .. hb .. ':0:U'
 local DSx = 'DS:max:GAUGE:' .. hb .. ':0:U'
 local DSn = 'DS:min:GAUGE:' .. hb .. ':0:U'
-
 -- What Collectd uses.
 local RRAc0  = 'RRA:AVERAGE:0.9:1:1200'
 local RRAc1  = 'RRA:MIN:0.9:1:1200'
@@ -488,7 +512,6 @@ local RRAc11 = 'RRA:MAX:0.9:223:1202'
 local RRAc12 = 'RRA:AVERAGE:0.9:2635:1201'
 local RRAc13 = 'RRA:MIN:0.9:2635:1201'
 local RRAc14 = 'RRA:MAX:0.9:2635:1201'
-
 -- Try LAST.
 local RRAl0  = 'RRA:LAST:0.9:1:1200'
 local RRAl1  = 'RRA:LAST:0.9:7:1235'
@@ -534,30 +557,12 @@ APT.updateRRD = function(results, host, ip)
     end
 end
 
-
 APT.doRRD = function(l, k, v)
-    if APT.checkFile(l .. "/" .. k .. ".lua") then
-        local results = loadfile(l .. "/" .. k .. ".lua")()
-        results = APT.fill(results)
-        APT.createRRD(k)
-        APT.updateRRD(results, k)
-        if nil ~= v then
-            local IPs = v.IPs
-            for i, u in pairs(IPs) do
-                if "table" == type(u) then
-                    for h, t in pairs(u) do
-                        APT.createRRD(k, h)
-                        results = APT.collate(k, h, results)
-                        APT.updateRRD(results, k, h)
-                    end
-                else
-                    APT.createRRD(k, i)
-                    results = APT.collate(k, i, results)
-                    APT.updateRRD(results, k, i)
-                end
-            end
-        end
-    end
+    APT.collateAll(l, k,
+        function(results, ip)
+            APT.createRRD(k, ip)
+            APT.updateRRD(results, k, ip)
+        end)
 end