System-Monitoring-Daemon/monitor.py

122 lines
2.6 KiB
Python
Executable File

#!/usr/bin/python
#
# Copyright 2012 Dan Ballard and Robert Hausch
#
## License:
#
# This file is part of System-Monitoring-Daemon.
#
# System-Monitoring-Daemon is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# System-Monitoring-Daemon is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with System-Monitoring-Daemon. If not, see <http://www.gnu.org/licenses/>.
##
import threading
from socket import *
import time
#from sensors.cpu import *
#from sensors.memory import *
import sensors
HOST = ''
PORT = 6000
ADDR = (HOST, PORT)
BUFSIZE = 4096
MAX_QUEUE = 5
class Stats():
def __init__(self):
self.read_lock = threading.Lock()
self.write_lock = threading.Lock()
self.sensors = sensors.sensors #[]
#self.sensors.append(cpu_monitor())
#self.sensors.append(mem_monitor())
self.stop = threading.Event()
t = threading.Thread(target=self.update_loop, args=())
t.start()
def update_loop(self):
while not self.stop.isSet():
self.acquire_write()
for s in self.sensors:
s.update()
self.release_write()
time.sleep(1)
def getStats(self):
data = []
self.acquire_read()
for s in self.sensors:
data.append(s.getFormatedData())
message = '{"Stats":['+','.join('%s'%x for x in data)+']}'
self.release_read()
return message
def acquire_read(self):
self.read_lock.acquire()
self.write_lock.acquire()
def release_read(self):
self.write_lock.release()
self.read_lock.release()
def acquire_write(self):
self.write_lock.acquire()
def release_write(self):
self.write_lock.release()
class ClientHandler:
def __init__(self, stats):
self.stats = stats
def handle(self, sock):
conn_sock.send( self.stats.getStats() )
conn_sock.close()
stats = Stats()
serv_sock = socket(AF_INET, SOCK_STREAM)
# Behave better after crash
serv_sock.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
serv_sock.bind((ADDR)) #tuple with one param
serv_sock.listen(MAX_QUEUE)
while 1:
try:
(conn_sock,addr) = serv_sock.accept()
except KeyboardInterrupt:
print "\nExiting..."
break;
except:
conn_sock = 0
if conn_sock:
#fork
#conn_sock.send('{ cpu_load: %d}' % (stats.cpu_load))
client = ClientHandler(stats)
client.handle(conn_sock)
stats.stop.set()
serv_sock.close()