Skip to content

Commit 554add1

Browse files
authored
Finish including Beam version details in MongoDB handshake (#36961)
* Revert #36949 * Update _MongoSink
1 parent 67d5477 commit 554add1

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

sdks/python/apache_beam/io/mongodbio.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -269,6 +269,12 @@ def __init__(
269269
**self.spec,
270270
driver=DriverInfo("Apache Beam", beam.__version__))
271271

272+
if "driver" not in self.spec:
273+
self.spec["driver"] = DriverInfo(
274+
name="Apache Beam",
275+
version=beam.__version__,
276+
)
277+
272278
def estimate_size(self):
273279
with MongoClient(self.uri, **self.spec) as client:
274280
return client[self.db].command("collstats", self.coll).get("size")
@@ -783,6 +789,12 @@ def __init__(self, uri=None, db=None, coll=None, extra_params=None):
783789
self.spec = extra_params
784790
self.client = None
785791

792+
if "driver" not in self.spec:
793+
self.spec["driver"] = DriverInfo(
794+
name="Apache Beam",
795+
version=beam.__version__,
796+
)
797+
786798
def write(self, documents):
787799
if self.client is None:
788800
self.client = MongoClient(host=self.uri, **self.spec)

0 commit comments

Comments
 (0)