prometheus-node-exporter: fix ubus stations

[feed/packages.git] / utils / prometheus-node-exporter-lua / files / usr / lib / lua / prometheus-collectors / hostapd_ubus_stations.lua

diff --git a/utils/prometheus-node-exporter-lua/files/usr/lib/lua/prometheus-collectors/hostapd_ubus_stations.lua b/utils/prometheus-node-exporter-lua/files/usr/lib/lua/prometheus-collectors/hostapd_ubus_stations.lua
index 88a993c54fdf0768fd6221e3c3986dc2797cb00b..3ae441474c3af4cdb1f4274a38318a35d3d89af5 100644 (file)
--- a/utils/prometheus-node-exporter-lua/files/usr/lib/lua/prometheus-collectors/hostapd_ubus_stations.lua
+++ b/utils/prometheus-node-exporter-lua/files/usr/lib/lua/prometheus-collectors/hostapd_ubus_stations.lua
@@ -1,14 +1,13 @@
 local ubus = require "ubus"
 local bit = require "bit"
 
-local function get_wifi_interfaces() -- based on hostapd_stations.lua
-  local u = ubus.connect()
-  local status = u:call("network.wireless", "status", {})
+local function get_wifi_hostapd_interfaces(u)
+  local ubuslist = u:objects()
   local interfaces = {}
 
-  for _, dev_table in pairs(status) do
-    for _, intf in ipairs(dev_table['interfaces']) do
-      table.insert(interfaces, intf['ifname'])
+  for _,net in ipairs(ubuslist) do
+    if net.find(net,"hostapd.") then
+      table.insert(interfaces, net)
     end
   end
 
@@ -16,6 +15,7 @@ local function get_wifi_interfaces() -- based on hostapd_stations.lua
 end
 
 local function scrape()
+  local u = ubus.connect()
   local metric_hostapd_ubus_station_rrm_caps_link_measurement =
     metric("hostapd_ubus_station_rrm_caps_link_measurement", "gauge")
   local metric_hostapd_ubus_station_rrm_caps_neighbor_report =
@@ -55,14 +55,15 @@ local function scrape()
     metric_hostapd_ubus_station_rrm_caps_ftm_range_report(label_station, rrm_caps_ftm_range_report)
   end
 
-  for _, ifname in ipairs(get_wifi_interfaces()) do
-    local u = ubus.connect()
-    local clients_call = u:call("hostapd." .. ifname, "get_clients", {})
+  for _, hostapd_int in ipairs(get_wifi_hostapd_interfaces(u)) do
+    local clients_call = u:call(hostapd_int, "get_clients", {})
+    local ifname = hostapd_int:gsub("hostapd.", "")
 
     for client, client_table in pairs(clients_call['clients']) do
       evaluate_metrics(ifname,  clients_call['freq'], client, client_table)
     end
   end
+  u:close()
 end
 
 return { scrape = scrape }