deb-sahara/sahara/utils/procutils.py
Nam Nguyen Hoai 36c7b3de74 TrivialFix: Remove logging import usused
This patch removes logging import unused.

Change-Id: Ia609f2a569828089a24811d55af80eda56bffa7e
2016-08-26 14:03:06 +07:00

97 lines
2.9 KiB
Python

# Copyright (c) 2013 Mirantis Inc.
#
# 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 os
import pickle # nosec
import sys
from eventlet.green import subprocess
from eventlet import timeout as e_timeout
from sahara import context
from sahara import exceptions
def _get_sub_executable():
return '%s/_sahara-subprocess' % os.path.dirname(sys.argv[0])
def start_subprocess():
return subprocess.Popen((sys.executable, _get_sub_executable()),
close_fds=True,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
def run_in_subprocess(proc, func, args=None, kwargs=None, interactive=False):
args = args or ()
kwargs = kwargs or {}
try:
# TODO(elmiko) these pickle usages should be reinvestigated to
# determine a more secure manner to deploy remote commands.
pickle.dump(func, proc.stdin) # nosec
pickle.dump(args, proc.stdin) # nosec
pickle.dump(kwargs, proc.stdin) # nosec
proc.stdin.flush()
if not interactive:
result = pickle.load(proc.stdout) # nosec
if 'exception' in result:
raise exceptions.SubprocessException(result['exception'])
return result['output']
finally:
# NOTE(dmitryme): in oslo.concurrency's file processutils.py it
# is suggested to sleep a little between calls to multiprocessing.
# That should allow it make some necessary cleanup
context.sleep(0)
def _finish(cleanup_func):
cleanup_func()
sys.stdin.close()
sys.stdout.close()
sys.stderr.close()
sys.exit(0)
def shutdown_subprocess(proc, cleanup_func):
try:
with e_timeout.Timeout(5):
# timeout would mean that our single-threaded subprocess
# is hung on previous task which blocks _finish to complete
run_in_subprocess(proc, _finish, (cleanup_func,))
except BaseException:
# exception could be caused by either timeout, or
# successful shutdown, ignoring anyway
pass
finally:
kill_subprocess(proc)
def kill_subprocess(proc):
proc.stdin.close()
proc.stdout.close()
proc.stderr.close()
try:
proc.kill()
proc.wait()
except OSError:
# could be caused by process already dead, so ignoring
pass