charm-hacluster/hooks/pcmk.py
Felipe Reyes 02d83b2e4e Support update parameters of a resource
This patch implements support to update parameters of an already
existing resource using "crm configure load update FILE"

Change-Id: I22730091d674145db4a1187b0904d9f88d9d8c6d
Partial-Bug: #1753432
2018-06-07 08:59:04 -04:00

259 lines
7.2 KiB
Python

# Copyright 2016 Canonical Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import commands
import re
import subprocess
import socket
import tempfile
import time
import xml.etree.ElementTree as etree
from distutils.version import StrictVersion
from StringIO import StringIO
from charmhelpers.core.hookenv import (
log,
ERROR,
INFO,
DEBUG,
WARNING,
)
class ServicesNotUp(Exception):
pass
class PropertyNotFound(Exception):
pass
def wait_for_pcmk(retries=12, sleep=10):
crm_up = None
hostname = socket.gethostname()
for i in range(retries):
if crm_up:
return True
output = commands.getstatusoutput("crm node list")[1]
crm_up = hostname in output
time.sleep(sleep)
if not crm_up:
raise ServicesNotUp("Pacemaker or Corosync are still down after "
"waiting for {} retries. Last output: {}"
"".format(retries, output))
def commit(cmd):
return subprocess.call(cmd.split())
def is_resource_present(resource):
status = commands.getstatusoutput("crm resource status %s" % resource)[0]
if status != 0:
return False
return True
def standby(node=None):
if node is None:
cmd = "crm -F node standby"
else:
cmd = "crm -F node standby %s" % node
commit(cmd)
def online(node=None):
if node is None:
cmd = "crm -F node online"
else:
cmd = "crm -F node online %s" % node
commit(cmd)
def crm_opt_exists(opt_name):
output = commands.getstatusoutput("crm configure show")[1]
if opt_name in output:
return True
return False
def crm_res_running(opt_name):
(_, output) = commands.getstatusoutput("crm resource status %s" % opt_name)
if output.startswith("resource %s is running" % opt_name):
return True
return False
def list_nodes():
cmd = ['crm', 'node', 'list']
out = subprocess.check_output(cmd)
nodes = []
for line in str(out).split('\n'):
if line != '':
nodes.append(line.split(':')[0])
return nodes
def _maas_ipmi_stonith_resource(node, power_params):
rsc_name = 'res_stonith_%s' % node
rsc = ('primitive %s stonith:external/ipmi params hostname=%s ipaddr=%s '
'userid=%s passwd=%s interface=lan' %
(rsc_name, node, power_params['power_address'],
power_params['power_user'], power_params['power_pass']))
# ensure ipmi stonith agents are not running on the nodes that
# they manage.
constraint = ('location const_loc_stonith_avoid_%s %s -inf: %s' %
(node, rsc_name, node))
return rsc, constraint
def maas_stonith_primitive(maas_nodes, crm_node):
power_type = power_params = None
for node in maas_nodes:
if node['hostname'].startswith(crm_node):
power_type = node['power_type']
power_params = node['power_parameters']
if not power_type or not power_params:
return False, False
rsc = constraint = None
# we can extend to support other power flavors in the future?
if power_type == 'ipmi':
rsc, constraint = _maas_ipmi_stonith_resource(crm_node, power_params)
else:
log('Unsupported STONITH power_type: %s' % power_type, ERROR)
return False, False
if not rsc or not constraint:
return False, False
return rsc, constraint
def get_property_from_xml(name, output):
"""Read a configuration property from the XML generated by 'crm configure show
xml'
:param name: property's name
:param output: string with the output of `crm configure show xml`
:returns: value of the property
:rtype: str
:raises: pcmk.PropertyNotFound
"""
tree = etree.parse(StringIO(output))
root = tree.getroot()
crm_config = root.find('configuration').find('crm_config')
props = crm_config.find('cluster_property_set')
for element in props:
if element.attrib['name'] == name:
# property found!
return element.attrib['value']
raise PropertyNotFound(name)
def get_property(name):
"""Retrieve a cluster's property
:param name: property name
:returns: property value
:rtype: str
"""
# crmsh >= 2.3 renamed show-property to get-property, 2.3.x is
# available since zesty
if crm_version() >= StrictVersion('2.3.0'):
output = subprocess.check_output(['crm', 'configure',
'get-property', name],
universal_newlines=True)
elif crm_version() < StrictVersion('2.2.0'):
# before 2.2.0 there is no method to get a property
output = subprocess.check_output(['crm', 'configure', 'show', 'xml'],
universal_newlines=True)
return get_property_from_xml(name, output)
else:
output = subprocess.check_output(['crm', 'configure',
'show-property', name],
universal_newlines=True)
return output
def set_property(name, value):
"""Set a cluster's property
:param name: property name
:param value: new value
"""
subprocess.check_output(['crm', 'configure',
'property', '%s=%s' % (name, value)],
universal_newlines=True)
def crm_version():
"""Parses the output of `crm --version` and returns a
distutils.version.StrictVersion instance
"""
ver = subprocess.check_output(['crm', '--version'],
universal_newlines=True)
r = re.compile(r'.*(\d\.\d\.\d).*')
matched = r.match(ver)
if not matched:
raise ValueError('error parsin crm version: %s' % ver)
else:
return StrictVersion(matched.group(1))
def crm_update_resource(res_name, res_type, res_params=None):
"""Update a resource using `crm configure load update`
:param res_name: resource name
:param res_type: resource type (e.g. IPaddr2)
:param res_params: resource's parameters (e.g. "params ip=10.5.250.250")
"""
with tempfile.NamedTemporaryFile() as f:
f.write('primitive {} {}'.format(res_name, res_type))
if res_params:
f.write(' \\\n\t{}'.format(res_params))
else:
f.write('\n')
f.flush()
f.seek(0)
log('Updating resource {}'.format(res_name), level=INFO)
log('File content:\n{}'.format(f.read()), level=DEBUG)
cmd = "crm configure load update {}".format(f.name)
log('Update command: {}'.format(cmd))
retcode = commit(cmd)
if retcode == 0:
level = DEBUG
else:
level = WARNING
log('crm command exit code: {}'.format(retcode), level=level)
return retcode