add helpers, add 3d-party plugins to readme, add license file

Author: vadv

LICENSE

@@ -0,0 +1,29 @@
+BSD 3-Clause License
+
+Copyright (c) 2016, vadv
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the name of the copyright holder nor the names of its
+  contributors may be used to endorse or promote products derived from
+  this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

README.rst

@@ -149,6 +149,12 @@ See the `examples <https://github.com/postgrespro/mamonsu/tree/master/examples>`
 
 After add new plugin, you must to reexport template and import this file to zabbix.
 
+=========
+3rd-party
+=========
+
+* `repo mamonsu-plugins <https://github.com/tarabanton/mamonsu-plugins>`_ for skytools, nginx, rabbitmq, uwsgi, gdnsd.
+
 ====
 Run
 ====

mamonsu/plugins/pgsql/driver/pool.py

@@ -41,11 +41,13 @@ class Pool(ConnectionInfo):
     def __init__(self):
         super(Pool, self).__init__()
         self.all_connections = {}
-        self._server_version = {}
-        self._mamonsu_bootstrap = {}
-        self._mamonsu_bootstrap_cache = 0
-        self._in_recovery = {}
-        self._in_recovery_cache = 0
+        self._cache = {
+            'server_version': {'storage': {}},
+            'bootstrap': {'storage': {}, 'counter': 0, 'cache': 10},
+            'recovery': {'storage': {}, 'counter': 0, 'cache': 10},
+            'pgpro': {'storage': {}},
+            'pgproee': {'storage': {}}
+        }
 
     def connection_string(self, db=None):
         self._init_conn_(db)
@@ -58,46 +60,69 @@ def query(self, query, db=None):
         return self.all_connections[db].query(query)
 
     def server_version(self, db=None):
-        if db in self._server_version:
-            return self._server_version[db]
+        if db in self._cache['server_version']['storage']:
+            return self._cache['server_version']['storage'][db]
         if platform.PY2:
             result = self.query('show server_version', db)[0][0]
         elif platform.PY3:
             result = bytes(
                 self.query('show server_version', db)[0][0], 'utf-8')
-        self._server_version[db] = "{0}".format(result.decode('ascii'))
-        return self._server_version[db]
-
-    def in_recovery(self, db=None):
-        if (db in self._in_recovery) and (self._in_recovery_cache < 10):
-            self._in_recovery_cache += 1
-            return self._in_recovery[db]
-        self._in_recovery_cache = 0
-        self._in_recovery[db] = self.query(
-            "select pg_catalog.pg_is_in_recovery()")[0][0]
-        return self._in_recovery[db]
+        self._cache['server_version']['storage'][db] = '{0}'.format(
+            result.decode('ascii'))
+        return self.self._cache['server_version']['storage'][db]
 
     def server_version_greater(self, version, db=None):
         return self.server_version(db) >= LooseVersion(version)
 
     def server_version_less(self, version, db=None):
         return self.server_version(db) <= LooseVersion(version)
 
-    def mamonsu_bootstrap(self, db=None):
-        if (db in self._mamonsu_bootstrap) and (self._mamonsu_bootstrap_cache < 10):
-            self._mamonsu_bootstrap_cache += 1
-            return self._mamonsu_bootstrap[db]
-        self._mamonsu_bootstrap_cache = 0
+    def in_recovery(self, db=None):
+        if db in self._cache['recovery']['storage']:
+            if self._cache['recovery']['counter'] < self._cache['recovery']['cache']:
+                self._cache['recovery']['counter'] += 1
+                return self._cache['recovery']['storage'][db]
+        self._cache['recovery']['counter'] = 0
+        self._cache['recovery']['storage'][db] = self.query(
+            "select pg_catalog.pg_is_in_recovery()")[0][0]
+        return self._cache['recovery']['storage'][db]
+
+    def is_bootstraped(self, db=None):
+        if db in self._cache['bootstrap']['storage']:
+            if self.cache['bootstrap']['counter'] < self._cache['bootstrap']['cache']:
+                self.cache['bootstrap']['counter'] += 1
+                return self._cache['bootstrap']['storage'][db]
+        self._cache['bootstrap']['counter'] = 0
         sql = """select count(*) from pg_catalog.pg_class
             where relname = 'mamonsu_config'"""
         result = int(self.query(sql, db)[0][0])
-        self._mamonsu_bootstrap[db] = (result == 1)
-        if self._mamonsu_bootstrap[db]:
-            self.all_connections[db].log.info("Found mamonsu bootstrap")
+        self._cache['bootstrap']['storage'][db] = (result == 1)
+        if self._cache['bootstrap']['storage'][db]:
+            self.all_connections[db].log.info('Found mamonsu bootstrap')
         else:
-            self.all_connections[db].log.info("Can't found mamonsu bootstrap")
-            self.all_connections[db].log.info("hint: run `mamonsu bootstrap` if you want to run without superuser rights")
-        return self._mamonsu_bootstrap[db]
+            self.all_connections[db].log.info('Can\'t found mamonsu bootstrap')
+            self.all_connections[db].log.info('hint: run `mamonsu bootstrap` if you want to run without superuser rights')
+        return self._cache['bootstrap']['storage'][db]
+
+    def is_pgpro(self, db=None):
+        if db in self._cache['pgpro']:
+            return self._cache['pgpro'][db]
+        try:
+            self.query('select pgpro_version()')
+            self._cache['pgpro'][db] = True
+        except:
+            self._cache['pgpro'][db] = False
+        return self._cache['pgpro'][db]
+
+    def is_pgpro_ee(self, db=None):
+        if not self.is_pgpro(self, db):
+            return False
+        if db in self._cache['pgproee']:
+            return self._cache['pgproee'][db]
+        self._cache['pgproee'][db] = (
+            self.query('select pgpro_edition()')[0][0].lower() == 'enterprise'
+        )
+        return self._cache['pgproee'][db]
 
     def extension_installed(self, ext, db=None):
         result = self.query('select count(*) from pg_catalog.pg_extension\
@@ -116,7 +141,7 @@ def get_sql(self, typ, db=None):
         if typ not in self.SQL:
             raise LookupError("Unknown SQL type: '{0}'".format(typ))
         result = self.SQL[typ]
-        if self.mamonsu_bootstrap(db):
+        if self.is_bootstraped(db):
             return result[1]
         else:
             return result[0]

mamonsu/plugins/pgsql/plugin.py

@@ -40,3 +40,9 @@ def disable_and_exit_if_extension_is_not_installed(self, ext, db=None):
             raise PluginDisableException("""Disable plugin and exit, because '{0}' \
 extension is not installed. Enable it in PostgreSQL instance: '{1}', \
 if needed and restart.""".format(ext, Pooler.connection_string(db)))
+
+    def disable_and_exit_if_not_pgpro_ee(self, db=None):
+        if not Pooler.is_pgpro_ee(db):
+            raise PluginDisableException("""Disable plugin and exit, because \
+PostgresPro Enterprise Edition is not detected [instance: '{1}']
+""".format(Pooler.connection_string(db)))