DefStream 0.1.19__79-py3-none-any.whl

DefStream/Broker/Configuration.py

@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+
+from abc import (
+	ABCMeta,
+	abstractmethod,
+)
+
+__all__ = (
+	'Configuration',
+)
+
+
+class Configuration(metaclass=ABCMeta):
+	"""Configuration Interface for Event Broker"""
+	@abstractmethod
+	def __init__(self, *args, **kwargs):
+		raise NotImplementedError('{} must be implemented __init__'.format(self.__class__.__name__))

DefStream/Broker/Connection.py

@@ -0,0 +1,141 @@
+# -*- coding: utf-8 -*-
+
+from abc import abstractmethod
+from typing import (
+	Any,
+	Optional,
+	Protocol,
+	runtime_checkable,
+)
+
+from .Consumer import Consumer
+from .Event import EventHandler
+from .Protocol import Sendable
+
+__all__ = (
+	'Connection',
+	'BrokerFactory',
+	'GetExchange',
+	'GetQueue',
+	'GetStream',
+	'GetConsumer',
+	'Declarable',
+	'Emittable',
+	'Requestable',
+	'PrioritySupport',
+)
+
+
+class Connection(Protocol):
+	@abstractmethod
+	def connect(self) -> None: ...
+
+	@abstractmethod
+	def close(self) -> None: ...
+
+
+class BrokerFactory(Protocol):
+	"""호출마다 독립적인 새 브로커 연결을 반환하는 팩터리 계약."""
+
+	@abstractmethod
+	def __call__(self) -> Connection: ...
+
+
+@runtime_checkable
+class GetExchange(Protocol):
+	@abstractmethod
+	def exchange(
+		self,
+		name: str,
+		/,
+	) -> Sendable: ...
+
+
+@runtime_checkable
+class GetQueue(Protocol):
+	@abstractmethod
+	def queue(
+		self,
+		name: str,
+		/,
+	) -> Sendable: ...
+
+
+@runtime_checkable
+class GetStream(Protocol):
+	@abstractmethod
+	def stream(
+		self,
+		name: str,
+		/,
+	) -> Sendable: ...
+
+
+@runtime_checkable
+class GetConsumer(Protocol):
+	@abstractmethod
+	def consumer(
+		self,
+		handler: Optional[EventHandler] = None,
+	) -> Consumer: ...
+
+
+@runtime_checkable
+class Declarable(Protocol):
+	@abstractmethod
+	def declare(
+		self,
+		parameters: Any,
+	) -> None: ...
+
+
+@runtime_checkable
+class Emittable(Protocol):
+	@abstractmethod
+	def emit(
+		self,
+		body: Any,
+		parameters: Any,
+	) -> Any: ...
+
+
+@runtime_checkable
+class Requestable(Protocol):
+	@abstractmethod
+	def request(
+		self,
+		body: Any,
+		parameters: Any,
+		timeout: float = 5,
+	) -> Any: ...
+
+
+@runtime_checkable
+class PrioritySupport(
+	Emittable,
+	Requestable,
+	Protocol,
+):
+	"""작업 구독 설정의 우선순위 처리와 우선순위 발행을 지원하는 계약."""
+
+	def priority_parameters(
+		self,
+		parameters: Any,
+	) -> Any: ...
+
+	def emit(
+		self,
+		body: Any,
+		parameters: Any,
+		*,
+		priority: Optional[int] = None,
+	) -> Any: ...
+
+	def request(
+		self,
+		body: Any,
+		parameters: Any,
+		timeout: float = 5,
+		*,
+		priority: Optional[int] = None,
+	) -> Any: ...

DefStream/Broker/Consumer.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+
+from abc import (
+	ABCMeta,
+	abstractmethod,
+)
+from typing import (
+	Any,
+	Callable,
+	Optional,
+)
+
+from .Event import (
+	Event,
+	EventHandler,
+	EventErrorPolicy,
+)
+
+__all__ = (
+	'Consumer',
+)
+
+
+class Consumer(metaclass=ABCMeta):
+	"""Consumer Interface of Event Broker"""
+	def complete(
+		self,
+		event: Event,
+		result: Any,
+	) -> None:
+		event.ack()
+
+	def call(self, callback: Callable[[], None]) -> None:
+		"""Schedule a callback on the connection owner thread."""
+		raise NotImplementedError('Consumer must implement thread-safe callback scheduling')
+
+	def fail(
+		self,
+		event: Event,
+		policy: Optional[EventErrorPolicy] = None,
+	) -> None:
+		raise NotImplementedError('Consumer must implement failure policies')
+
+	def close(self) -> None:
+		raise NotImplementedError('Consumer must implement close')
+
+	@abstractmethod
+	def subs(
+		self,
+		queue: Any,
+		handler: Optional[EventHandler] = None,
+	) -> None:
+		raise NotImplementedError('{} must implement subs'.format(self.__class__.__name__))
+	@abstractmethod
+	def run(self):
+		raise NotImplementedError('{} must be implemented run'.format(self.__class__.__name__))
+	@abstractmethod
+	def stop(self):
+		raise NotImplementedError('{} must be implemented stop'.format(self.__class__.__name__))

DefStream/Broker/Errors.py

@@ -0,0 +1,14 @@
+__all__ = (
+	'RemoteError',
+)
+
+
+class RemoteError(RuntimeError):
+	def __init__(
+		self,
+		code: str,
+		message: str,
+	):
+		self.code = code
+		self.message = message
+		super().__init__(f'{code}: {message}')

DefStream/Broker/Event.py

@@ -0,0 +1,35 @@
+from abc import abstractmethod
+from typing import (
+	Any,
+	Protocol,
+)
+
+__all__ = (
+	'Event',
+	'EventHandler',
+	'EventErrorPolicy',
+)
+
+
+class Event(Protocol):
+	@property
+	def body(self) -> Any: ...
+
+	def ack(self) -> None: ...
+	def nack(self) -> None: ...
+	def reject(self) -> None: ...
+
+
+class EventHandler(Protocol):
+	@abstractmethod
+	def __call__(
+		self,
+		event: Event,
+	) -> None: ...
+
+
+class EventErrorPolicy(Protocol):
+	def __call__(
+		self,
+		event: Event,
+	) -> None: ...

DefStream/Broker/Implements/RabbitMQ/Configuration.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+
+from DefStream.Broker import Configuration as BaseConfiguration
+
+from .Serializer import (
+	Encoder,
+	Decoder,
+	JavaScriptObjectNotationEncoder,
+	JavaScriptObjectNotationDecoder,
+)
+
+
+from pika import SSLOptions
+from ssl import create_default_context
+from typing import Optional
+
+__all__ = (
+	'Configuration',
+)
+
+
+class Configuration(BaseConfiguration):
+	"""
+	Configuration of Event Broker for RabbitMQ
+	"""
+	def __init__(
+		self,
+		host: str,
+		port: int,
+		username: str,
+		password: str,
+		ssl: bool = False,
+		vhost: str = '/',
+		timeout: Optional[int] = None,
+		heartbeat: Optional[int] = None,
+		encode: Encoder = JavaScriptObjectNotationEncoder(),
+		decode: Decoder = JavaScriptObjectNotationDecoder(),
+	):
+		self.host = host
+		self.port = port
+		self.username = username
+		self.password = password
+		self.ssl = SSLOptions(create_default_context(), host) if ssl else None
+		self.vhost = vhost
+		self.timeout = timeout/1000 if timeout else None
+		self.heartbeat = int(heartbeat*2/1000) if heartbeat else None
+		self.encode = encode
+		self.decode = decode
+		return