checks/check_sql

   1 #!/usr/bin/python
   2 # -*- encoding: utf-8; py-indent-offset: 4 -*-
   3 # +------------------------------------------------------------------+
   4 # |             ____ _               _        __  __ _  __           |
   5 # |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
   6 # |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
   7 # |           | |___| | | |  __/ (__|   <    | |  | | . \            |
   8 # |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
   9 # |                                                                  |
  10 # | Copyright Mathias Kettner 2014             mk@mathias-kettner.de |
  11 # +------------------------------------------------------------------+
  12 #
  13 # This file is part of Check_MK.
  14 # The official homepage is at http://mathias-kettner.de/check_mk.
  15 #
  16 # check_mk is free software;  you can redistribute it and/or modify it
  17 # under the  terms of the  GNU General Public License  as published by
  18 # the Free Software Foundation in version 2.  check_mk is  distributed
  19 # in the hope that it will be useful, but WITHOUT ANY WARRANTY;  with-
  20 # out even the implied warranty of  MERCHANTABILITY  or  FITNESS FOR A
  21 # PARTICULAR PURPOSE. See the  GNU General Public License for more de-
  22 # tails. You should have  received  a copy of the  GNU  General Public
  23 # License along with GNU Make; see the file  COPYING.  If  not,  write
  24 # to the Free Software Foundation, Inc., 51 Franklin St,  Fifth Floor,
  25 # Boston, MA 02110-1301 USA.
  26
  27 #active_checks['sql'] = [
  28 #  ( {'dbms': 'postgres', 'description': u'SQL Test', 'user': 'golla',
  29 #             'sql': 'testerer', 'password': 'toor',
  30 #             'procedure': {'useprocs': True, 'input': '1223,456'},
  31 #             'name': 'messpc'}, [], ALL_HOSTS ),
  32 #  ] + active_checks['sql']
  33
  34
  35 def check_sql_arguments(params):
  36     args = []
  37
  38     if "host" in params:
  39         args += ["--hostname=%s" % params["host"]]
  40     else:
  41         args += ["--hostname=$HOSTADDRESS$"]
  42
  43     args += ["--dbms=%s" % params["dbms"]]
  44     args += ["--name=%s" % params["name"]]
  45     args += ["--user=%s" % params["user"]]
  46     args += [passwordstore_get_cmdline("--password=%s", params["password"])]
  47
  48     if "port" in params:
  49         args += ["--port=%s" % params["port"]]
  50
  51     if "procedure" in params:
  52         if "procedure" in params and "useprocs" in params["procedure"]:
  53             args += ["--procedure"]
  54             if "input" in params["procedure"]:
  55                 args += ["--inputvars=%s" % params["procedure"]["input"]]
  56
  57     if "levels" in params:
  58         upper = params["levels"]
  59     else:
  60         upper = "", ""
  61
  62     if "levels_low" in params:
  63         lower = params["levels_low"]
  64     else:
  65         lower = "", ""
  66
  67     if "perfdata" in params:
  68         args += ["--metrics"]
  69
  70     if "levels" in params or "levels_low" in params:
  71         args += ["-w%s:%s" % (lower[0], upper[0])]
  72         args += ["-c%s:%s" % (lower[1], upper[1])]
  73
  74     if isinstance(params["sql"], tuple):
  75         sql_tmp = params["sql"][-1]
  76     else:
  77         sql_tmp = params["sql"]
  78
  79     args += ["%s" % sql_tmp.replace("\n", r"\n").replace(";", r"\;")]
  80
  81     return args
  82
  83
  84 active_check_info['sql'] = {
  85     "command_line": '$USER1$/check_sql $ARG1$',
  86     "argument_function": check_sql_arguments,
  87     "service_description": lambda args: args["description"],
  88     "has_perfdata": True,
  89 }