def terminate_rescan_process(self, silent=False):
from infi.storagemodel.base import gevent_wrapper
Process = gevent_wrapper.get_process_class()
gevent_wrapper.sleep(0) # give time for gipc time to join on the defunct rescan process
if isinstance(self.rescan_process, Process) and self.rescan_process.is_alive():
if not silent:
logger.debug("terminating previous rescan process")
else:
getLogger("gipc").addHandler(NullHandler())
try:
self.rescan_process.terminate()
gevent_wrapper.sleep(0) # give time for gipc time to join on the defunct rescan process
except Exception:
if not silent:
logger.exception("Failed to terminate rescan process")
self.rescan_process = None
def _initiate_rescan(self, wait_for_completion=True, raise_error=False):
from infi.storagemodel.base import gevent_wrapper
from os import getpid
def get_process_parent_pid(process_object):
# works for gipc._GProcess and multiprocessing.Process
return process_object._parent_pid
Process = gevent_wrapper.get_process_class()
gevent_wrapper.sleep(0) # give time for gipc time to join on the defunct rescan process
if self.rescan_process and getpid() != get_process_parent_pid(self.rescan_process):
self.rescan_process = None
if isinstance(self.rescan_process, Process) and self.rescan_process.is_alive():
if (datetime.now() - self.rescan_process_start_time).total_seconds() > self.rescan_subprocess_timeout:
logger.debug("rescan process timed out, killing it")
self.terminate_rescan_process()
self.rescan_process = None
self._initiate_rescan(wait_for_completion, raise_error)
else:
logger.debug("previous rescan process is still running")
if wait_for_completion:
logger.debug("waiting for rescan process completion")
self.rescan_process.join(self.rescan_subprocess_timeout)
if raise_error and self.rescan_process.exitcode not in (None, 0):
raise RescanError("rescan process exited with non-zero exit code {}".format(self.rescan_process.exitcode))
else:
if isinstance(self.rescan_process, Process):
self.rescan_process.join() # process no longer alive, this is safe
logger.debug("previous rescan process exit code: {}".format(self.rescan_process.exitcode))
if raise_error and self.rescan_process.exitcode not in (None, 0):
raise RescanError("rescan process exited with non-zero exit code {}".format(self.rescan_process.exitcode))
self.rescan_process_start_time = datetime.now()
self.rescan_process = gevent_wrapper.start_process(self.rescan_method)
logger.debug("rescan process started")
if wait_for_completion:
logger.debug("waiting for rescan process completion")
self.rescan_process.join(self.rescan_subprocess_timeout)
if raise_error and self.rescan_process.exitcode not in (None, 0):
raise RescanError("rescan process exited with non-zero exit code {}".format(self.rescan_process.exitcode))