Skip to content

Commit 3a36512

Browse files
committed
add headers parametr to constructor
1 parent 917396d commit 3a36512

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

kafka_bridge_client/consumer.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ def __init__(
5656
consumer_name: str,
5757
sleep_interval_seconds: int = 2,
5858
client_timeout_seconds: int = 15,
59+
headers: t.Dict[str, t.Any] = None,
5960
) -> None:
6061
self._group_id = group_id
6162
self._consumer_name = consumer_name
@@ -72,6 +73,7 @@ def __init__(
7273
self._sleep_interval_seconds = sleep_interval_seconds
7374
self._bootstrap_server = bootstrap_server
7475
self._client_timeout_seconds = client_timeout_seconds
76+
self._headers = headers or {}
7577

7678
async def _request(
7779
self,
@@ -84,12 +86,13 @@ async def _request(
8486
) -> Response:
8587
params = params or {}
8688
data = data or {}
87-
headers = headers or {}
89+
_headers = headers or {}
90+
_headers.update(self._headers)
8891
url = urljoin(self._bootstrap_server, path)
8992

9093
async with aiohttp.ClientSession(
9194
timeout=aiohttp.ClientTimeout(self._client_timeout_seconds),
92-
headers=headers,
95+
headers=_headers,
9396
) as session:
9497
try:
9598
resp = await session.request(

0 commit comments

Comments
 (0)