def restart_zookeeper_if_config_changed():
"""Restart Zookeeper if zoo.cfg has changed.
As peers come and go, zoo.cfg will be updated. When that file changes,
restart the Zookeeper service and set an appropriate status message.
"""
# Possibly update bind address
network_interface = hookenv.config().get('network_interface')
if data_changed("zookeeper.bind_address", network_interface):
zk = Zookeeper()
zk.update_bind_address()
zoo_cfg = DistConfig().path('zookeeper_conf') / 'zoo.cfg'
if any_file_changed([zoo_cfg]):
hookenv.status_set('maintenance',
'Server config changed: restarting Zookeeper')
zk = Zookeeper()
zk.stop()
zk.start()
zk_count = int(zk.get_zk_count())
extra_status = ""
if zk_count < 3:
extra_status = ": less than 3 is suboptimal"
elif (zk_count % 2 == 0):
extra_status = ": even number is suboptimal"
hookenv.status_set('active',
'Ready (%d zk units%s)' % (zk_count, extra_status))
else:
# Make sure zookeeper is running in any case
zk = Zookeeper()
zk.start()
zk.open_ports()
def restart_zookeeper_if_config_changed():
"""Restart Zookeeper if zoo.cfg has changed.
As peers come and go, zoo.cfg will be updated. When that file changes,
restart the Zookeeper service and set an appropriate status message.
"""
# Possibly update bind address
network_interface = hookenv.config().get('network_interface')
if data_changed("zookeeper.bind_address", network_interface):
zk = Zookeeper()
zk.update_bind_address()
zoo_cfg = DistConfig().path('zookeeper_conf') / 'zoo.cfg'
if any_file_changed([zoo_cfg]):
hookenv.status_set('maintenance', 'Server config changed: restarting Zookeeper')
zk = Zookeeper()
zk.stop()
zk.start()
zk_count = int(zk.get_zk_count())
extra_status = ""
if zk_count < 3:
extra_status = ": less than 3 is suboptimal"
elif (zk_count % 2 == 0):
extra_status = ": even number is suboptimal"
hookenv.status_set('active', 'Ready (%d zk units%s)' % (zk_count, extra_status))
def storage_detaching():
unitdata.kv().unset('zookeeper.storage.data_dir')
zookeeper = Zookeeper()
zookeeper.close_ports()
zookeeper.stop()
clear_flag('zookeeper.configured')
hookenv.status_set('waiting', 'reconfiguring to use temporary storage')
clear_flag('zookeeper.storage.data.attached')
def storage_attach():
storageids = hookenv.storage_list('data')
if not storageids:
hookenv.status_set('blocked', 'cannot locate attached storage')
return
storageid = storageids[0]
mount = hookenv.storage_get('location', storageid)
if not mount:
hookenv.status_set('blocked', 'cannot locate attached storage mount')
return
data_dir = os.path.join(mount, "data")
unitdata.kv().set('zookeeper.storage.data_dir', data_dir)
hookenv.log('Zookeeper data storage attached at {}'.format(data_dir))
# Stop Zookeeper; removing zookeeper.configured state will trigger
# a reconfigure if/when it's ready
zookeeper = Zookeeper()
zookeeper.close_ports()
zookeeper.stop()
clear_flag('zookeeper.configured')
hookenv.status_set('waiting', 'reconfiguring to use attached storage')
set_flag('zookeeper.storage.data.attached')
def restart_zookeeper():
hookenv.status_set('maintenance', 'Server config changed: restarting Zookeeper')
zk = Zookeeper()
zk.stop()
zk.start()
hookenv.status_set('active', 'Ready')
