def start(self, env, upgrade_type=None):
import params
import status_params
install_nifi()
install_nifi_toolkit()
nifi_toolkit_util_common.copy_toolkit_scripts(
params.toolkit_files_dir,
params.toolkit_tmp_dir,
params.nifi_user,
params.nifi_group,
upgrade_type,
service=nifi_toolkit_util_common.NIFI)
self.configure(env)
ca_server_script = nifi_toolkit_util_common.get_toolkit_script(
'tls-toolkit.sh', params.toolkit_tmp_dir,
params.stack_version_buildnum)
run_ca_script = os.path.join(params.toolkit_tmp_dir, 'run_ca.sh')
Directory([params.nifi_config_dir],
owner=params.nifi_user,
group=params.nifi_group,
create_parents=True,
recursive_ownership=True)
File(ca_server_script, mode=0755)
File(run_ca_script, mode=0755)
Execute(
(run_ca_script, "start", params.jdk64_home, ca_server_script,
params.nifi_config_dir + '/nifi-certificate-authority.json',
params.nifi_ca_log_file_stdout, params.nifi_ca_log_file_stderr,
status_params.nifi_ca_pid_file, params.toolkit_ca_check_url),
user=params.nifi_user,
logoutput=True)
if not os.path.isfile(status_params.nifi_ca_pid_file):
raise Exception('Expected pid file to exist')
def stop(self, env, upgrade_type=None):
import params
import status_params
nifi_toolkit_util_common.copy_toolkit_scripts(
params.toolkit_files_dir,
params.toolkit_tmp_dir,
params.nifi_user,
params.nifi_group,
upgrade_type,
service=nifi_toolkit_util_common.NIFI)
run_ca_script = os.path.join(params.toolkit_tmp_dir, 'run_ca.sh')
ca_server_script = nifi_toolkit_util_common.get_toolkit_script(
'tls-toolkit.sh', params.toolkit_tmp_dir,
params.stack_version_buildnum)
File(ca_server_script, mode=0755)
File(run_ca_script, mode=0755)
if path_isfile(status_params.nifi_ca_pid_file):
Execute(
(run_ca_script, "stop", params.jdk64_home, ca_server_script,
params.nifi_config_dir + '/nifi-certificate-authority.json',
params.nifi_ca_log_file_stdout,
params.nifi_ca_log_file_stderr,
status_params.nifi_ca_pid_file, params.toolkit_ca_check_url),
user=params.nifi_user,
logoutput=True)
try:
self.status(env)
except ComponentIsNotRunning:
unlink(status_params.nifi_ca_pid_file)
def start(self, env, upgrade_type=None):
import params
import status_params
install_nifi()
install_nifi_toolkit()
nifi_toolkit_util_common.copy_toolkit_scripts(
params.toolkit_files_dir,
params.toolkit_tmp_dir,
params.nifi_user,
params.nifi_group,
upgrade_type,
service=nifi_toolkit_util_common.NIFI)
self.configure(env, is_starting=True)
setup_ranger_nifi(upgrade_type=None)
Execute('export JAVA_HOME=' + params.jdk64_home + ';' +
params.bin_dir + '/nifi.sh start >> ' +
params.nifi_node_log_file,
user=params.nifi_user)
# If nifi pid file not created yet, wait a bit
if not os.path.isfile(status_params.nifi_pid_dir + '/nifi.pid'):
Execute('sleep 5')
if params.nifi_registry_url and params.stack_support_nifi_auto_client_registration:
Logger.info(
"Trying to register NIFI Registry. This can take up to several minutes. Please, wait..."
)
try:
nifi_cli.create_or_update_reg_client(params.nifi_registry_host,
params.nifi_registry_url)
except:
Logger.warning(
"Unable to create or update registry client. Please review NiFi to setup the registry manually if necessary."
)
def install(self, env):
import params
import status_params
self.install_packages(env)
#Be sure ca script is in cache
nifi_toolkit_util_common.copy_toolkit_scripts(params.toolkit_files_dir, params.toolkit_tmp_dir, params.nifi_user, params.nifi_group, upgrade_type=None, service=nifi_toolkit_util_common.NIFI)
def install(self, env):
import params
self.install_packages(env)
Directory([params.nifi_node_dir, params.nifi_node_log_dir],
owner=params.nifi_user,
group=params.nifi_group,
create_parents=True,
recursive_ownership=True,
cd_access='a'
)
nifi_toolkit_util_common.copy_toolkit_scripts(params.toolkit_files_dir, params.toolkit_tmp_dir, params.nifi_user, params.nifi_group, upgrade_type=None, service=nifi_toolkit_util_common.NIFI)
Execute('touch ' + params.nifi_node_log_file, user=params.nifi_user)
def start(self, env, upgrade_type=None):
import params
import status_params
nifi_toolkit_util_common.copy_toolkit_scripts(
params.toolkit_files_dir,
params.toolkit_tmp_dir,
params.nifi_registry_user,
params.nifi_registry_group,
upgrade_type=None,
service=nifi_toolkit_util_common.NIFI_REGISTRY)
self.configure(env, is_starting=True)
setup_ranger_nifi_registry(upgrade_type=None)
Execute('export JAVA_HOME=' + params.jdk64_home + ';' +
params.bin_dir + '/nifi-registry.sh start >> ' +
params.nifi_registry_log_file,
user=params.nifi_registry_user)
#If nifi pid file not created yet, wait a bit
if not os.path.isfile(status_params.nifi_registry_pid_dir +
'/nifi-registry.pid'):
Execute('sleep 5')
