Minor refactor

This commit is contained in:
Douglas Barone 2022-04-18 18:29:45 -04:00
parent 1a39236ccb
commit 05c3fc3e0e

View File

@ -1,6 +1,6 @@
import { subMinutes } from 'date-fns' import { subMinutes } from 'date-fns'
import { getOnlineWifiDevices as getOnlineUnifiDevices } from './unifiController' import * as unifiController from './unifiController'
import { getOnlineWifiDevices as getOnlineCiscoDevices } from './ciscoController' import * as ciscoController from './ciscoController'
import prisma from '../prisma' import prisma from '../prisma'
@ -15,16 +15,13 @@ const TIMEOUT_IN_MILLISECONDS = process.env.TASK_TIMEOUT || 120000
let working = false let working = false
const wifiControllers = [unifiController, ciscoController]
async function getOnlineDevices() { async function getOnlineDevices() {
const onlineUnifiDevicesPromise = getOnlineUnifiDevices()
const onlineCiscoDevicesPromise = getOnlineCiscoDevices()
const [onlineUnifiDevices, onlineCiscoDevices] = await Promise.all([ const onlineDevicesPromises = wifiControllers.map(wifiController => wifiController.getOnlineWifiDevices())
onlineUnifiDevicesPromise,
onlineCiscoDevicesPromise
])
const onlineDevices = [...onlineUnifiDevices, ...onlineCiscoDevices] const onlineDevices = (await Promise.all(onlineDevicesPromises)).flat()
return onlineDevices return onlineDevices
} }