voussoir/cmd
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add threadpool to internetcheck.

Browse source
This commit is contained in:
voussoir 2021-06-12 12:43:35 -07:00
parent 6a920f32ee
commit ce80bb32a7
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
internetcheck.py
View file

@ -4,6 +4,7 @@ import sys
import time import time
from voussoirkit import hms from voussoirkit import hms
from voussoirkit import threadpool
from voussoirkit import vlogging from voussoirkit import vlogging
log = vlogging.getLogger(__name__, 'internetcheck') log = vlogging.getLogger(__name__, 'internetcheck')
@ -28,6 +29,7 @@ cur.executescript(DB_INIT)
down = False down = False
outage_started = None outage_started = None
thread_pool = threadpool.ThreadPool(10)
def percentage(items): def percentage(items):
trues = sum(bool(i) for i in items) trues = sum(bool(i) for i in items)
@ -63,7 +65,10 @@ def check_dns():
'spectrum.com', 'spectrum.com',
'youtube.com', 'youtube.com',
] ]
checks = [check(name) for name in names] thread_pool.pause()
jobs = [{'function': check, 'args': [name]} for name in names]
thread_pool.add_many(jobs)
checks = [job.value for job in thread_pool.result_generator()]
return percentage(checks) return percentage(checks)
def check_ip(): def check_ip():
@ -87,7 +92,10 @@ def check_ip():
'35.165.194.49', '35.165.194.49',
'69.252.80.75', '69.252.80.75',
] ]
checks = [check(ip) for ip in ips] thread_pool.pause()
jobs = [{'function': check, 'args': [ip]} for ip in ips]
thread_pool.add_many(jobs)
checks = [job.value for job in thread_pool.result_generator()]
return percentage(checks) return percentage(checks)
def set_down(lan_ok, dns_ok, ip_ok): def set_down(lan_ok, dns_ok, ip_ok):