Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

refactor

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-06-21 11:35:01 +02:00
parent 0ac5621099
commit 49de36c5a0
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pyinfra/server/server.py
View File

@ -11,7 +11,7 @@ from pyinfra.server.stream.rest import LazyRestProcessor
@singledispatch
def set_up_processing_server(arg: Union[dict, Callable], buffer_size=1):
def set_up_processing_server(arg: Union[dict, Callable], buffer_size):
"""Produces a processing server given a streamable function or a mapping from operations to streamable functions.
Streamable functions are constructed by calling pyinfra.server.utils.make_streamable_and_wrap_in_packing_logic on a
function taking a tuple of data and metadata and also returning a tuple or yielding tuples of data and metadata.
@ -32,12 +32,12 @@ def set_up_processing_server(arg: Union[dict, Callable], buffer_size=1):
@set_up_processing_server.register
def _(operation2stream_fn: dict, buffer_size):
def _(operation2stream_fn: dict, buffer_size=1):
return __stream_fn_to_processing_server(operation2stream_fn, buffer_size)
@set_up_processing_server.register
def _(stream_fn: object, buffer_size):
def _(stream_fn: object, buffer_size=1):
operation2stream_fn = {None: stream_fn}
return __stream_fn_to_processing_server(operation2stream_fn, buffer_size)