Skip to content

create_publisher

faststream.kafka.publisher.factory.create_publisher #

create_publisher(
    *,
    autoflush,
    batch,
    key,
    topic,
    partition,
    headers,
    reply_to,
    config,
    middlewares,
    schema_,
    title_,
    description_,
    include_in_schema,
)
Source code in faststream/kafka/publisher/factory.py 
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
def create_publisher(
    *,
    autoflush: bool,
    batch: bool,
    key: bytes | None,
    topic: str,
    partition: int | None,
    headers: dict[str, str] | None,
    reply_to: str,
    # Publisher args
    config: "KafkaBrokerConfig",
    middlewares: Sequence["PublisherMiddleware"],
    # Specification args
    schema_: Any | None,
    title_: str | None,
    description_: str | None,
    include_in_schema: bool,
) -> BatchPublisher | DefaultPublisher:
    publisher_config = KafkaPublisherConfig(
        key=key,
        topic=topic,
        partition=partition,
        headers=headers,
        reply_to=reply_to,
        middlewares=middlewares,
        _outer_config=config,
    )

    specification = KafkaPublisherSpecification(
        _outer_config=config,
        specification_config=KafkaPublisherSpecificationConfig(
            topic=topic,
            schema_=schema_,
            title_=title_,
            description_=description_,
            include_in_schema=include_in_schema,
        ),
    )

    if batch:
        if key:
            msg = "You can't setup `key` with batch publisher"
            raise SetupError(msg)

        publisher: BatchPublisher | DefaultPublisher = BatchPublisher(
            publisher_config,
            specification,
        )
        publish_method = "_basic_publish_batch"

    else:
        publisher = DefaultPublisher(publisher_config, specification)
        publish_method = "_basic_publish"

    if autoflush:
        default_publish: Callable[..., Awaitable[Any | None]] = getattr(
            publisher,
            publish_method,
        )

        @wraps(default_publish)
        async def autoflush_wrapper(*args: Any, **kwargs: Any) -> Any | None:
            result = await default_publish(*args, **kwargs)
            await publisher.flush()
            return result

        setattr(publisher, publish_method, autoflush_wrapper)

    return publisher