Skip to content

BatchParser

faststream.nats.parser.BatchParser #

BatchParser(*, pattern)

Bases: JsParser

A class to parse NATS batch messages.

Source code in faststream/nats/parser.py 
23
24
25
26
27
28
29
def __init__(
    self,
    *,
    pattern: str,
) -> None:
    path_re, _ = compile_nats_wildcard(pattern)
    self.__path_re = path_re

get_path #

get_path(subject)
Source code in faststream/nats/parser.py 
31
32
33
34
35
36
37
38
39
40
41
42
def get_path(
    self,
    subject: str,
) -> Optional["AnyDict"]:
    path: Optional[AnyDict] = None

    if (path_re := self.__path_re) is not None and (
        match := path_re.match(subject)
    ) is not None:
        path = match.groupdict()

    return path

decode_message async #

decode_message(msg)
Source code in faststream/nats/parser.py 
44
45
46
47
48
async def decode_message(
    self,
    msg: "StreamMessage[Any]",
) -> "DecodedMessage":
    return decode_message(msg)

parse_message async #

parse_message(message, *, path=None)
Source code in faststream/nats/parser.py 
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
async def parse_message(
    self,
    message: "Msg",
    *,
    path: Optional["AnyDict"] = None,
) -> "StreamMessage[Msg]":
    if path is None:
        path = self.get_path(message.subject)

    headers = message.header or {}

    return NatsMessage(
        raw_message=message,
        body=message.data,
        path=path or {},
        reply_to=headers.get("reply_to", ""),  # differ from core
        headers=headers,
        content_type=headers.get("content-type", ""),
        message_id=headers.get("message_id", gen_cor_id()),
        correlation_id=headers.get("correlation_id", gen_cor_id()),
    )

parse_batch async #

parse_batch(message)
Source code in faststream/nats/parser.py 
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
async def parse_batch(
    self,
    message: List["Msg"],
) -> "StreamMessage[List[Msg]]":
    body: List[bytes] = []
    batch_headers: List[Dict[str, str]] = []

    if message:
        path = self.get_path(message[0].subject)

        for m in message:
            batch_headers.append(m.headers or {})
            body.append(m.data)

    else:
        path = None

    headers = next(iter(batch_headers), {})

    return NatsBatchMessage(
        raw_message=message,
        body=body,
        path=path or {},
        headers=headers,
        batch_headers=batch_headers,
    )

decode_batch async #

decode_batch(msg)
Source code in faststream/nats/parser.py 
140
141
142
143
144
145
146
147
148
149
150
151
152
153
async def decode_batch(
    self,
    msg: "StreamMessage[List[Msg]]",
) -> List["DecodedMessage"]:
    data: List[DecodedMessage] = []

    path: Optional[AnyDict] = None
    for m in msg.raw_message:
        one_msg = await self.parse_message(m, path=path)
        path = one_msg.path

        data.append(decode_message(one_msg))

    return data