python.d/squid.chart.py

   1 # -*- coding: utf-8 -*-
   2 # Description: squid netdata python.d module
   3 # Author: Pawel Krupa (paulfantom)
   4
   5 from base import SocketService
   6
   7 # default module values (can be overridden per job in `config`)
   8 # update_every = 2
   9 priority = 60000
  10 retries = 60
  11
  12 # charts order (can be overridden if you want less charts, or different order)
  13 ORDER = ['clients_net', 'clients_requests', 'servers_net', 'servers_requests']
  14
  15 CHARTS = {
  16     'clients_net': {
  17         'options': [None, "Squid Client Bandwidth", "kilobits/s", "clients", "squid.clients.net", "area"],
  18         'lines': [
  19             ["client_http_kbytes_in", "in", "incremental", 8, 1],
  20             ["client_http_kbytes_out", "out", "incremental", -8, 1],
  21             ["client_http_hit_kbytes_out", "hits", "incremental", -8, 1]
  22         ]},
  23     'clients_requests': {
  24         'options': [None, "Squid Client Requests", "requests/s", "clients", "squid.clients.requests", 'line'],
  25         'lines': [
  26             ["client_http_requests", "requests", "incremental"],
  27             ["client_http_hits", "hits", "incremental"],
  28             ["client_http_errors", "errors", "incremental", -1, 1]
  29         ]},
  30     'servers_net': {
  31         'options': [None, "Squid Server Bandwidth", "kilobits/s", "servers", "squid.servers.net", "area"],
  32         'lines': [
  33             ["server_all_kbytes_in", "in", "incremental", 8, 1],
  34             ["server_all_kbytes_out", "out", "incremental", -8, 1]
  35         ]},
  36     'servers_requests': {
  37         'options': [None, "Squid Server Requests", "requests/s", "servers", "squid.servers.requests", 'line'],
  38         'lines': [
  39             ["server_all_requests", "requests", "incremental"],
  40             ["server_all_errors", "errors", "incremental", -1, 1]
  41         ]}
  42 }
  43
  44
  45 class Service(SocketService):
  46     def __init__(self, configuration=None, name=None):
  47         SocketService.__init__(self, configuration=configuration, name=name)
  48         self.request = ""
  49         self.host = "localhost"
  50         self.port = 3128
  51         self.order = ORDER
  52         self.definitions = CHARTS
  53
  54     def _get_data(self):
  55         """
  56         Get data via http request
  57         :return: dict
  58         """
  59         data = {}
  60         try:
  61             raw = self._get_raw_data().split('\r\n')[-1]
  62             if raw.startswith('<'):
  63                 self.error("invalid data received")
  64                 return None
  65             for row in raw.split('\n'):
  66                 if row.startswith(("client", "server.all")):
  67                     tmp = row.split("=")
  68                     data[tmp[0].replace('.', '_').strip(' ')] = int(tmp[1])
  69         except (ValueError, AttributeError, TypeError):
  70             self.error("invalid data received")
  71             return None
  72
  73         if len(data) == 0:
  74             self.error("no data received")
  75             return None
  76         else:
  77             return data
  78
  79     def check(self):
  80         """
  81         Parse essential configuration, autodetect squid configuration (if needed), and check if data is available
  82         :return: boolean
  83         """
  84         self._parse_config()
  85         # format request
  86         req = self.request.decode()
  87         if not req.startswith("GET"):
  88             req = "GET " + req
  89         if not req.endswith(" HTTP/1.0\r\n\r\n"):
  90             req += " HTTP/1.0\r\n\r\n"
  91         self.request = req.encode()
  92         if self._get_data() is not None:
  93             return True
  94         else:
  95             return False