You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
2.1 KiB
69 lines
2.1 KiB
4 years ago
|
import uuid
|
||
|
|
import threading
|
||
|
|
from typing import Any
|
||
|
|
from .messages.invocation_message import InvocationClientStreamMessage
|
||
|
|
from .messages.stream_item_message import StreamItemMessage
|
||
|
|
from .messages.completion_message import CompletionClientStreamMessage
|
||
|
|
|
||
|
|
|
||
|
|
class Subject(object):
|
||
|
|
"""Client to server streaming
|
||
|
|
https://docs.microsoft.com/en-gb/aspnet/core/signalr/streaming?view=aspnetcore-5.0#client-to-server-streaming
|
||
|
|
items = list(range(0,10))
|
||
|
|
subject = Subject()
|
||
|
|
connection.send("UploadStream", subject)
|
||
|
|
while(len(self.items) > 0):
|
||
|
|
subject.next(str(self.items.pop()))
|
||
|
|
subject.complete()
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.connection = None
|
||
|
|
self.target = None
|
||
|
|
self.invocation_id = str(uuid.uuid4())
|
||
|
|
self.lock = threading.RLock()
|
||
|
|
|
||
|
|
def check(self):
|
||
|
|
"""Ensures that invocation streaming object is correct
|
||
|
|
|
||
|
|
Raises:
|
||
|
|
ValueError: if object is not valid, exception will be raised
|
||
|
|
"""
|
||
|
|
if self.connection is None\
|
||
|
|
or self.target is None\
|
||
|
|
or self.invocation_id is None:
|
||
|
|
raise ValueError(
|
||
|
|
"subject must be passed as an agument to a send function. "
|
||
|
|
+ "hub_connection.send([method],[subject]")
|
||
|
|
|
||
|
|
def next(self, item: Any):
|
||
|
|
"""Send next item to the server
|
||
|
|
|
||
|
|
Args:
|
||
|
|
item (any): Item that will be streamed
|
||
|
|
"""
|
||
|
|
self.check()
|
||
|
|
with self.lock:
|
||
|
|
self.connection.transport.send(StreamItemMessage(
|
||
|
|
self.invocation_id,
|
||
|
|
item))
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
"""Starts streaming
|
||
|
|
"""
|
||
|
|
self.check()
|
||
|
|
with self.lock:
|
||
|
|
self.connection.transport.send(
|
||
|
|
InvocationClientStreamMessage(
|
||
|
|
[self.invocation_id],
|
||
|
|
self.target,
|
||
|
|
[]))
|
||
|
|
|
||
|
|
def complete(self):
|
||
|
|
"""Finish streaming
|
||
|
|
"""
|
||
|
|
self.check()
|
||
|
|
with self.lock:
|
||
|
|
self.connection.transport.send(CompletionClientStreamMessage(
|
||
|
|
self.invocation_id))
|