mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-26 00:46:22 +02:00
34eb083836
* Plugin architecture for messaging fabric * Schemas use a technology neutral expression * Schemas strictness has uncovered some incorrect schema use which is fixed
30 lines
821 B
Python
30 lines
821 B
Python
|
|
from . metrics import SubscriberMetrics
|
|
from . subscriber import Subscriber
|
|
from . spec import Spec
|
|
|
|
class SubscriberSpec(Spec):
|
|
|
|
def __init__(self, name, schema):
|
|
self.name = name
|
|
self.schema = schema
|
|
|
|
def add(self, flow, processor, definition):
|
|
|
|
subscriber_metrics = SubscriberMetrics(
|
|
processor = flow.id, flow = flow.name, name = self.name
|
|
)
|
|
|
|
subscriber = Subscriber(
|
|
backend = processor.pubsub,
|
|
topic = definition[self.name],
|
|
subscription = flow.id,
|
|
consumer_name = flow.id,
|
|
schema = self.schema,
|
|
metrics = subscriber_metrics,
|
|
)
|
|
|
|
# Put it in the consumer map, does that work?
|
|
# It means it gets start/stop call.
|
|
flow.consumer[self.name] = subscriber
|
|
|