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         time.sleep(interval)
  70
  71
  72 def displayer(port):
  73     """Runs the http exporter interface.
  74
  75     This method executes an HTTP server
  76     (`monitor.displayer.http_display`). The `DataStore` object used to
  77     call this method is a remote object obtained from a
  78     `MonitorManager`.
  79
  80     """
  81     manager = MonitorManager(address=('localhost', MANAGER_PORT),
  82                              authkey=MANAGER_AUTHKEY)
  83     manager.connect()
  84     store = manager.get_store()
  85     http_display.main(store, port,
  86                       "/home/rodrigo/workspace/monitor/monitor/www_templates/",
  87                       "/home/rodrigo/workspace/monitor/monitor/static/")
  88
  89
  90 def main():
  91     """Main function, launches process for each part of the system."""
  92     parser = argparse.ArgumentParser(description="Server monitoring system.")
  93     parser.add_argument('--http-port', metavar="PORT", type=int, default=9012,
  94                         help="Port for the HTTP server (displayer).")
  95     parser.add_argument('--collection-interval', metavar="SECONDS",
  96                         type=int, default=30,
  97                         help="Interval, in seconds, between data collections.")
  98     parser.add_argument('--remote-servers', nargs="+",
  99                         help=("List, separated by spaces, of remote servers "
 100                               "(HOST:PORT) to monitor."))
 101     args = parser.parse_args()
 102
 103     #sources = (("localhost", 8080),)
 104     sources = [x.split(':') for x in args.remote_servers]
 105
 106     collector_p = Process(target=timed_collector,
 107                           args=(sources, args.collection_interval))
 108     displayer_p = Process(target=displayer, args=(args.http_port,))
 109     collector_p.start()
 110     displayer_p.start()
 111     collector_p.join()
 112     displayer_p.join()
 113
 114
 115 if __name__ == '__main__':
 116     store = datastore.DataStore()
 117     MonitorManager.register('get_store', callable=lambda: store)
 118     manager = MonitorManager(address=('', MANAGER_PORT),
 119                              authkey=MANAGER_AUTHKEY)
 120     manager.start()
 121     main()