275 lines
7.8 KiB
Python
275 lines
7.8 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
|
|
|
|
|
|
class PcmkError(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):
|
|
(code, output) = commands.getstatusoutput("crm configure show xml")
|
|
if code != 0:
|
|
raise PcmkError(code, output)
|
|
|
|
tree = etree.parse(StringIO(output))
|
|
root = tree.getroot()
|
|
resources = root.find('configuration').find('resources')
|
|
result = resources.findall(".//*[@id='%s']" % opt_name)
|
|
if len(result) == 0:
|
|
return False
|
|
elif len(result) == 1:
|
|
return True
|
|
else:
|
|
log('crm configure show xml:\n%s' % output, level=DEBUG)
|
|
raise PcmkError(len(result),
|
|
"Found more %d elements with id '%s'" % (len(result),
|
|
opt_name))
|
|
|
|
|
|
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
|