127 lines
3.3 KiB
Python
127 lines
3.3 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 subprocess
|
|
import socket
|
|
|
|
from charmhelpers.core.hookenv import (
|
|
log,
|
|
ERROR
|
|
)
|
|
|
|
|
|
def wait_for_pcmk():
|
|
crm_up = None
|
|
hostname = socket.gethostname()
|
|
while not crm_up:
|
|
output = commands.getstatusoutput("crm node list")[1]
|
|
crm_up = hostname in output
|
|
|
|
|
|
def commit(cmd):
|
|
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
|