main.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 #
   4 # Server monitoring system
   5 #
   6 # Copyright © 2011 Rodrigo Eduardo Lazo Paz
   7 #
   8 # This program is free software: you can redistribute it and/or modify
   9 # it under the terms of the GNU General Public License as published by
  10 # the Free Software Foundation, either version 3 of the License, or
  11 # (at your option) any later version.
  12 #
  13 # This program is distributed in the hope that it will be useful,
  14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 # GNU General Public License for more details.
  17 #
  18 # You should have received a copy of the GNU General Public License
  19 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 #
  21
  22
  23 """Launching script."""
  24
  25
  26 __author__ = "rlazo.paz@gmail.com (Rodrigo Lazo)"
  27 __version__ = 0.1
  28
  29
  30 import argparse
  31 import time
  32
  33 from multiprocessing import Process, managers
  34 from monitor.collector import HttpCollector
  35 from monitor import http_display
  36 from monitor import datastore
  37
  38
  39 MANAGER_PORT = 53170
  40 MANAGER_AUTHKEY = 'xlz98'
  41
  42
  43 class MonitorManager(managers.SyncManager):
  44     """Custom multiprocessing manager."""
  45     pass
  46
  47
  48 def timed_collector(sources, interval):
  49     """Runs collector in intervals.
  50
  51     This method executes and infinite loop collecting data each
  52     `interval` seconds. The `DataStore` object used to store the
  53     retrieved data is a remote object obtained from a `MonitorManager`.
  54
  55
  56     Arguments:
  57     - `sources`: Iterable, pairs of (hostname, port) to use for collection.
  58     - `interval`: Integer, seconds between collections.
  59
  60     """
  61     manager = MonitorManager(address=('localhost', MANAGER_PORT),
  62                              authkey=MANAGER_AUTHKEY)
  63     manager.connect()
  64     store = manager.get_store()
  65     collector = HttpCollector(sources)
  66     while (True):
  67         collected_data = collector.collect()
  68         store.insert_dict(collected_data)
  69         print collected_data
  70         time.sleep(interval)
  71
  72
  73 def displayer(port):
  74     """Runs the http exporter interface.
  75
  76     This method executes an HTTP server
  77     (`monitor.displayer.http_display`). The `DataStore` object used to
  78     call this method is a remote object obtained from a
  79     `MonitorManager`.
  80
  81     """
  82     manager = MonitorManager(address=('localhost', MANAGER_PORT),
  83                              authkey=MANAGER_AUTHKEY)
  84     manager.connect()
  85     store = manager.get_store()
  86     http_display.main(store, port,
  87                       "/home/rodrigo/workspace/monitor/monitor/www_templates/",
  88                       "/home/rodrigo/workspace/monitor/monitor/static/")
  89
  90
  91 def main():
  92     """Main function, launches process for each part of the system."""
  93     parser = argparse.ArgumentParser(description="Server monitoring system.")
  94     parser.add_argument('--http-port', metavar="PORT", type=int, default=9012,
  95                         help="Port for the HTTP server (displayer).")
  96     parser.add_argument('--collection-interval', metavar="SECONDS",
  97                         type=int, default=30,
  98                         help="Interval, in seconds, between data collections.")
  99     parser.add_argument('--remote-servers', nargs="+",
 100                         help=("List, separated by spaces, of remote servers "
 101                               "(HOST:PORT) to monitor."))
 102     args = parser.parse_args()
 103
 104     #sources = (("localhost", 8080),)
 105     sources = [x.split(':') for x in args.remote_servers]
 106
 107     collector_p = Process(target=timed_collector,
 108                           args=(sources, args.collection_interval))
 109     displayer_p = Process(target=displayer, args=(args.http_port,))
 110     collector_p.start()
 111     displayer_p.start()
 112     collector_p.join()
 113     displayer_p.join()
 114
 115
 116 if __name__ == '__main__':
 117     store = datastore.DataStore()
 118     MonitorManager.register('get_store', callable=lambda: store)
 119     manager = MonitorManager(address=('', MANAGER_PORT),
 120                              authkey=MANAGER_AUTHKEY)
 121     manager.start()
 122     main()