diff options
author | shmel1k <shmel1k@ydb.tech> | 2023-11-26 18:16:14 +0300 |
---|---|---|
committer | shmel1k <shmel1k@ydb.tech> | 2023-11-26 18:43:30 +0300 |
commit | b8cf9e88f4c5c64d9406af533d8948deb050d695 (patch) | |
tree | 218eb61fb3c3b96ec08b4d8cdfef383104a87d63 /contrib/python/Twisted/py2/twisted/internet/threads.py | |
parent | 523f645a83a0ec97a0332dbc3863bb354c92a328 (diff) | |
download | ydb-b8cf9e88f4c5c64d9406af533d8948deb050d695.tar.gz |
add kikimr_configure
Diffstat (limited to 'contrib/python/Twisted/py2/twisted/internet/threads.py')
-rw-r--r-- | contrib/python/Twisted/py2/twisted/internet/threads.py | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/contrib/python/Twisted/py2/twisted/internet/threads.py b/contrib/python/Twisted/py2/twisted/internet/threads.py new file mode 100644 index 0000000000..8852d00265 --- /dev/null +++ b/contrib/python/Twisted/py2/twisted/internet/threads.py @@ -0,0 +1,127 @@ +# Copyright (c) Twisted Matrix Laboratories. +# See LICENSE for details. + +""" +Extended thread dispatching support. + +For basic support see reactor threading API docs. +""" + +from __future__ import division, absolute_import + +from twisted.python.compat import _PY3 +if not _PY3: + import Queue +else: + import queue as Queue + +from twisted.python import failure +from twisted.internet import defer + + +def deferToThreadPool(reactor, threadpool, f, *args, **kwargs): + """ + Call the function C{f} using a thread from the given threadpool and return + the result as a Deferred. + + This function is only used by client code which is maintaining its own + threadpool. To run a function in the reactor's threadpool, use + C{deferToThread}. + + @param reactor: The reactor in whose main thread the Deferred will be + invoked. + + @param threadpool: An object which supports the C{callInThreadWithCallback} + method of C{twisted.python.threadpool.ThreadPool}. + + @param f: The function to call. + @param *args: positional arguments to pass to f. + @param **kwargs: keyword arguments to pass to f. + + @return: A Deferred which fires a callback with the result of f, or an + errback with a L{twisted.python.failure.Failure} if f throws an + exception. + """ + d = defer.Deferred() + + def onResult(success, result): + if success: + reactor.callFromThread(d.callback, result) + else: + reactor.callFromThread(d.errback, result) + + threadpool.callInThreadWithCallback(onResult, f, *args, **kwargs) + + return d + + +def deferToThread(f, *args, **kwargs): + """ + Run a function in a thread and return the result as a Deferred. + + @param f: The function to call. + @param *args: positional arguments to pass to f. + @param **kwargs: keyword arguments to pass to f. + + @return: A Deferred which fires a callback with the result of f, + or an errback with a L{twisted.python.failure.Failure} if f throws + an exception. + """ + from twisted.internet import reactor + return deferToThreadPool(reactor, reactor.getThreadPool(), + f, *args, **kwargs) + + +def _runMultiple(tupleList): + """ + Run a list of functions. + """ + for f, args, kwargs in tupleList: + f(*args, **kwargs) + + +def callMultipleInThread(tupleList): + """ + Run a list of functions in the same thread. + + tupleList should be a list of (function, argsList, kwargsDict) tuples. + """ + from twisted.internet import reactor + reactor.callInThread(_runMultiple, tupleList) + + +def blockingCallFromThread(reactor, f, *a, **kw): + """ + Run a function in the reactor from a thread, and wait for the result + synchronously. If the function returns a L{Deferred}, wait for its + result and return that. + + @param reactor: The L{IReactorThreads} provider which will be used to + schedule the function call. + @param f: the callable to run in the reactor thread + @type f: any callable. + @param a: the arguments to pass to C{f}. + @param kw: the keyword arguments to pass to C{f}. + + @return: the result of the L{Deferred} returned by C{f}, or the result + of C{f} if it returns anything other than a L{Deferred}. + + @raise: If C{f} raises a synchronous exception, + C{blockingCallFromThread} will raise that exception. If C{f} + returns a L{Deferred} which fires with a L{Failure}, + C{blockingCallFromThread} will raise that failure's exception (see + L{Failure.raiseException}). + """ + queue = Queue.Queue() + def _callFromThread(): + result = defer.maybeDeferred(f, *a, **kw) + result.addBoth(queue.put) + reactor.callFromThread(_callFromThread) + result = queue.get() + if isinstance(result, failure.Failure): + result.raiseException() + return result + + +__all__ = ["deferToThread", "deferToThreadPool", "callMultipleInThread", + "blockingCallFromThread"] |