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 = 5
  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"],
  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                 return None
  64             for row in raw.split('\n'):
  65                 if row.startswith(("client", "server.all")):
  66                     tmp = row.split("=")
  67                     data[tmp[0].replace('.', '_').strip(' ')] = int(tmp[1])
  68         except (ValueError, AttributeError, TypeError):
  69             return None
  70
  71         if len(data) == 0:
  72             return None
  73         else:
  74             return data
  75
  76     def check(self):
  77         """
  78         Parse essential configuration, autodetect squid configuration (if needed), and check if data is available
  79         :return: boolean
  80         """
  81         self._parse_config()
  82         # format request
  83         req = self.request.decode()
  84         if not req.startswith("GET"):
  85             req = "GET " + req
  86         if not req.endswith(" HTTP/1.0\r\n\r\n"):
  87             req += " HTTP/1.0\r\n\r\n"
  88         self.request = req.encode()
  89         if self._get_data() is not None:
  90             return True
  91         else:
  92             self.error("No data returned")
  93             return False