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

refactoring

Browse Source
This commit is contained in:
Matthias Bisping 2022-05-06 23:13:48 +02:00
parent 866df2dee3
commit 1daaf2b904
1 changed files with 7 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
pyinfra/server/server.py
View File

@ -24,7 +24,7 @@ def make_processor_fn(operation, buffer_size, batched):
return operation
class OnDemandProcessor(abc.ABC):
class OnDemandProcessor:
def __init__(self, processor_fn):
self.execution_queue = chain([])
self.processor_fn = processor_fn
@ -41,22 +41,12 @@ class OnDemandProcessor(abc.ABC):
return Nothing
class OnDemandProcessorAdapter(abc.ABC):
@abc.abstractmethod
def submit(self, *args, **kwargs) -> None:
raise NotImplementedError
class RestOndDemandProcessorAdapter(OnDemandProcessor):
def __init__(self, processor_fn):
super(RestOndDemandProcessorAdapter, self).__init__(processor_fn=processor_fn)
def compute_next(self) -> Union[Nothing, Any]:
result = self.processor.compute_next()
return result
class RestProcessorAdapter(OnDemandProcessorAdapter):
def __init__(self, processor: OnDemandProcessor):
self.processor = processor
def submit(self, request) -> None:
self.processor.submit(request.json, final=request.method == "POST")
def submit(self, request, **kwargs) -> None:
super(RestOndDemandProcessorAdapter, self).submit(request.json, final=request.method == "POST")
class RestStreamer:
@ -93,7 +83,7 @@ def valid(result):
def set_up_processing_server(process_fn):
app = Flask(__name__)
streamer = RestStreamer(RestProcessorAdapter(OnDemandProcessor(process_fn)))
streamer = RestStreamer(RestOndDemandProcessorAdapter(process_fn))
@app.route("/ready", methods=["GET"])
def ready():