Initial work on latest version check code
* Add code to pull the latest supported versions of PostgreSQL from the official RSS feed. * TODO: Split rss parsing code into separate function for unit testing. * TODO: Test/debug * TODO: Add metrics to return how far behind the latest version the current cluster is.
This commit is contained in:
parent
15097dcba4
commit
487386a7cc
111
src/pgmon.py
111
src/pgmon.py
@ -24,6 +24,9 @@ from http.server import BaseHTTPRequestHandler, HTTPServer
|
|||||||
from http.server import ThreadingHTTPServer
|
from http.server import ThreadingHTTPServer
|
||||||
from urllib.parse import urlparse, parse_qs
|
from urllib.parse import urlparse, parse_qs
|
||||||
|
|
||||||
|
import requests
|
||||||
|
import re
|
||||||
|
|
||||||
VERSION = "1.0.2"
|
VERSION = "1.0.2"
|
||||||
|
|
||||||
# Configuration
|
# Configuration
|
||||||
@ -43,6 +46,12 @@ cluster_version = None
|
|||||||
cluster_version_next_check = None
|
cluster_version_next_check = None
|
||||||
cluster_version_lock = Lock()
|
cluster_version_lock = Lock()
|
||||||
|
|
||||||
|
# PostgreSQL latest version information
|
||||||
|
latest_version = None
|
||||||
|
latest_version_next_check = None
|
||||||
|
latest_version_lock = Lock()
|
||||||
|
release_supported = None
|
||||||
|
|
||||||
# Running state (used to gracefully shut down)
|
# Running state (used to gracefully shut down)
|
||||||
running = True
|
running = True
|
||||||
|
|
||||||
@ -83,6 +92,10 @@ class MetricVersionError(Exception):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class LatestVersionCheckError(Exception):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
# Default config settings
|
# Default config settings
|
||||||
default_config = {
|
default_config = {
|
||||||
# The address the agent binds to
|
# The address the agent binds to
|
||||||
@ -116,6 +129,8 @@ default_config = {
|
|||||||
"reconnect_cooldown": 30,
|
"reconnect_cooldown": 30,
|
||||||
# How often to check the version of PostgreSQL (seconds)
|
# How often to check the version of PostgreSQL (seconds)
|
||||||
"version_check_period": 300,
|
"version_check_period": 300,
|
||||||
|
# How often to check the latest supported version of PostgreSQL (seconds)
|
||||||
|
"latest_version_check_period": 86400,
|
||||||
# Metrics
|
# Metrics
|
||||||
"metrics": {},
|
"metrics": {},
|
||||||
}
|
}
|
||||||
@ -482,6 +497,102 @@ def get_cluster_version():
|
|||||||
return cluster_version
|
return cluster_version
|
||||||
|
|
||||||
|
|
||||||
|
def get_latest_version():
|
||||||
|
"""
|
||||||
|
Get the latest supported version of the major PostgreSQL release running on the server being monitored.
|
||||||
|
"""
|
||||||
|
global latest_version
|
||||||
|
global latest_version_next_check
|
||||||
|
global release_supported
|
||||||
|
|
||||||
|
# If we don't know the latest version or it's past the recheck time, get the
|
||||||
|
# version from the PostgreSQL RSS feed. Only one thread needs to do this, so
|
||||||
|
# they all try to grab the lock, and then make sure nobody else beat them to it.
|
||||||
|
if (
|
||||||
|
latest_version is None
|
||||||
|
or latest_version_next_check is None
|
||||||
|
or latest_version_next_check < datetime.now()
|
||||||
|
):
|
||||||
|
with latest_version_lock:
|
||||||
|
# Only check if nobody already got the version before us
|
||||||
|
if (
|
||||||
|
latest_version is None
|
||||||
|
or latest_version_next_check is None
|
||||||
|
or latest_version_next_check < datetime.now()
|
||||||
|
):
|
||||||
|
log.info("Checking latest PostgreSQL version")
|
||||||
|
cluster_version = get_cluster_version()
|
||||||
|
latest_version_next_check = datetime.now() + timedelta(
|
||||||
|
seconds=int(config["latest_version_check_period"])
|
||||||
|
)
|
||||||
|
|
||||||
|
# Extract the release
|
||||||
|
# 90603 => 9.6
|
||||||
|
# 130010 => 13
|
||||||
|
if cluster_version // 10000 < 10:
|
||||||
|
release = cluster_version // 10000 + (
|
||||||
|
cluster_version % 10000 // 100 / 10
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
release = cluster_version // 10000
|
||||||
|
|
||||||
|
# Grab the RSS feed
|
||||||
|
raw_rss = requests.get("https://www.postgresql.org/versions.rss")
|
||||||
|
if raw_rss.status != 200:
|
||||||
|
raise LatestVersionCheckError("code={}".format(r.status))
|
||||||
|
|
||||||
|
# Regular expressions for parsing the RSS document
|
||||||
|
version_line = re.compile(
|
||||||
|
r"([0-9][0-9.]+) is the latest release in the ([0-9][0-9.]+) series"
|
||||||
|
)
|
||||||
|
unsupported_line = re.compile(r"^This version is unsupported")
|
||||||
|
|
||||||
|
# Loop through the RSS until we find the current release
|
||||||
|
release_found = False
|
||||||
|
for line in raw_rss.text.lines():
|
||||||
|
m = version_line.match(line)
|
||||||
|
if m:
|
||||||
|
if release == int(m.group(2)):
|
||||||
|
release_found = True
|
||||||
|
version = float(m.group(1))
|
||||||
|
if version < 10:
|
||||||
|
parts = list(map(int, version.split(".")))
|
||||||
|
latest_version = int(
|
||||||
|
"{}{:02}{:02}".format(parts[0], parts[1], parts[2])
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
parts = list(map(int, version.split(".")))
|
||||||
|
latest_version = int(
|
||||||
|
"{}00{:02}".format(parts[0], parts[1])
|
||||||
|
)
|
||||||
|
release_found = True
|
||||||
|
elif release_found:
|
||||||
|
if unsupported.match(line):
|
||||||
|
release_supported = False
|
||||||
|
else:
|
||||||
|
release_supported = True
|
||||||
|
break
|
||||||
|
|
||||||
|
# Make sure we actually found it
|
||||||
|
if not release_found:
|
||||||
|
raise LatestVersionCheckError(
|
||||||
|
"Current release ({}) not found".format(release)
|
||||||
|
)
|
||||||
|
|
||||||
|
log.info(
|
||||||
|
"Got latest PostgreSQL version: {} supported={}".format(
|
||||||
|
latest_version, release_supported
|
||||||
|
)
|
||||||
|
)
|
||||||
|
log.debug(
|
||||||
|
"Next latest PostgreSQL version check will be after: {}".format(
|
||||||
|
latest_version_next_check
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
return latest_version
|
||||||
|
|
||||||
|
|
||||||
def sample_metric(dbname, metric_name, args, retry=True):
|
def sample_metric(dbname, metric_name, args, retry=True):
|
||||||
"""
|
"""
|
||||||
Run the appropriate query for the named metric against the specified database
|
Run the appropriate query for the named metric against the specified database
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user