Rooba
/
mitogen
mirror of https://github.com/mitogen-hq/mitogen.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

issue #155: core: implement Side._on_fork() 

Central mechanism for deleting all non-Latch file descriptors belonging
to the parent process during fork().
This commit is contained in:
David Wilson 2018-03-25 12:38:08 +05:45
parent 80642ed9ec
commit 872181bebd
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mitogen/core.py
View File

@ -45,6 +45,7 @@ import threading
import time import time
import traceback import traceback
import warnings import warnings
import weakref
import zlib import zlib
# TODO: usage of 'import' after setting __name__, but before fixing up # TODO: usage of 'import' after setting __name__, but before fixing up
@ -658,10 +659,13 @@ class LogHandler(logging.Handler):
class Side(object): class Side(object):
_fork_refs = weakref.WeakValueDictionary()
def __init__(self, stream, fd, cloexec=True, keep_alive=True): def __init__(self, stream, fd, cloexec=True, keep_alive=True):
self.stream = stream self.stream = stream
self.fd = fd self.fd = fd
self.keep_alive = keep_alive self.keep_alive = keep_alive
self._fork_refs[id(self)] = self
if cloexec: if cloexec:
set_cloexec(fd) set_cloexec(fd)
set_nonblock(fd) set_nonblock(fd)
@ -674,6 +678,11 @@ class Side(object):
raise StreamError('%r.fileno() called but no FD set', self) raise StreamError('%r.fileno() called but no FD set', self)
return self.fd return self.fd
@classmethod
def _on_fork(cls):
for side in list(cls._fork_refs.values()):
side.close()
def close(self): def close(self):
if self.fd is not None: if self.fd is not None:
_vv and IOLOG.debug('%r.close()', self) _vv and IOLOG.debug('%r.close()', self)