# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2018 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# https://cloudlinux.com/docs/LICENCE.TXT
#
import logging
import pathlib
from clconfigure import run, task
from clconfigure.packages import install_package
from clconfigure.services import (
STATE_FAILED,
STATE_MASKED,
STATE_RUNNING,
STATE_STOPPED,
STATE_UNMASKED,
STATE_ENABLED,
STATE_DISABLED,
get_service_state,
set_service_state
)
def initialize_lvestats():
# NOTE(vlebedev): We need to trigger 'lve-stats' scriptlets to properly initialized it's database stuff.
install_package('lve-stats', reinstall=True)
enable_lve_services()
@task("Changing default limits state to '{desired_state}'")
def set_default_limits_state(desired_state):
"""
Brings default limits to given state (unlimited | default).
May be executed more that once, does't crash on future calls
"""
if desired_state == 'unlimited':
run(['lvectl', 'set', 'default', '--unlimited'])
else:
raise RuntimeError('Another states temporary unavailable')
def disable_lve_services():
"""
Turn off all lve-related services.
"""
for state in [STATE_STOPPED, STATE_DISABLED, STATE_MASKED]:
for service in ['lve', 'lve_namespaces', 'lvestats', 'lvectl']:
set_service_state(state, service)
def enable_lve_services():
"""
Turn on all lve-related services.
"""
for state in [STATE_UNMASKED, STATE_RUNNING, STATE_ENABLED]:
for service in ['lve', 'lve_namespaces', 'lvestats', 'lvectl']:
set_service_state(state, service)
@task("Unloading lve module")
def unload_lve_module():
"""
Restart services that still use lve device and unload lve module then
"""
if not pathlib.Path('/dev/lve').exists():
# already unloaded
return
# Check if lve device is held by any process
res = run(['lsof', '/dev/lve'])
res_stdout_lines = res.stdout.split() if res.stdout is not None else []
services = [
service for service in ['mysqld', 'mysql', 'mariadb', 'httpd', 'apache2']
if any(line.startswith(service) for line in res_stdout_lines)
]
if services:
run(['systemctl', 'restart', *services])
res = run(['modprobe', '-rf', 'kmodlve'])
if pathlib.Path('/sys/module/kmodlve').exists():
logging.warning('Failed to unload the lve module. Please reboot the server.')
def apply_workaround_lve_failed():
"""
Apply workaround for the case
When `systemctl stop lve` makes it transition to "failed" state instead of "inactive"
"""
if get_service_state('lve') != STATE_FAILED:
return
set_service_state(STATE_UNMASKED, 'lve')
set_service_state(STATE_RUNNING, 'lve')
set_service_state(STATE_STOPPED, 'lve')
set_service_state(STATE_MASKED, 'lve')
|