Charmed-Kubernetes/etcd/lib/etcd_lib.py

95 lines
2.8 KiB
Python

from charmhelpers.contrib.templating.jinja import render
from charmhelpers.core.hookenv import (
network_get,
unit_private_ip,
)
import json
GRAFANA_DASHBOARD_FILE = "grafana_dashboard.json.j2"
def get_ingress_addresses(endpoint_name):
"""Returns all ingress-addresses belonging to the named endpoint, if
available. Falls back to private-address if necessary."""
try:
data = network_get(endpoint_name)
except NotImplementedError:
return [unit_private_ip()]
if "ingress-addresses" in data:
return data["ingress-addresses"]
else:
return [unit_private_ip()]
def get_ingress_address(endpoint_name):
"""Returns an ingress-address belonging to the named endpoint, if
available. Falls back to private-address if necessary."""
return get_ingress_addresses(endpoint_name)[0]
def get_bind_address(endpoint_name):
"""Returns the first bind-address found in network info
belonging to the named endpoint, if available.
Falls back to private-address if necessary.
@param endpoint_name the endpoint from where taking the
bind address
"""
try:
data = network_get(endpoint_name)
except NotImplementedError:
return unit_private_ip()
# Consider that network-get returns something like:
#
# bind-addresses:
# - macaddress: 02:d0:9e:31:d9:e0
# interfacename: ens5
# addresses:
# - hostname: ""
# address: 172.31.5.4
# cidr: 172.31.0.0/20
# - hostname: ""
# address: 172.31.5.4
# cidr: 172.31.0.0/20
# - macaddress: 8a:32:d7:8d:f6:9a
# interfacename: fan-252
# addresses:
# - hostname: ""
# address: 252.5.4.1
# cidr: 252.0.0.0/12
# egress-subnets:
# - 172.31.5.4/32
# ingress-addresses:
# - 172.31.5.4
# - 172.31.5.4
# - 252.5.4.1
if "bind-addresses" in data:
bind_addresses = data["bind-addresses"]
if len(bind_addresses) > 0:
if "addresses" in bind_addresses[0]:
if len(bind_addresses[0]["addresses"]) > 0:
return bind_addresses[0]["addresses"][0]["address"]
return unit_private_ip()
def render_grafana_dashboard(datasource):
"""Load grafana dashboard json model and insert prometheus datasource.
:param datasource: name of the 'prometheus' application that will be used
as datasource in grafana dashboard
:return: Grafana dashboard json model as a dict.
"""
datasource = "{} - Juju generated source".format(datasource)
jinja_args = {"variable_start_string": "<<", "variable_end_string": ">>"}
return json.loads(
render(
GRAFANA_DASHBOARD_FILE,
{"datasource": datasource},
jinja_env_args=jinja_args,
)
)