CDRIVER-2419 mongoc_bulk_operation_set_client_session

Assertions when setting client and client_session will ensure that the bulk never has a client_session belonging to a foreign client.

Author: jmikola

src/mongoc/mongoc-bulk-operation.c

@@ -841,6 +841,10 @@ mongoc_bulk_operation_set_client (mongoc_bulk_operation_t *bulk, void *client)
    BSON_ASSERT (bulk);
    BSON_ASSERT (client);
 
+   if (bulk->session) {
+      BSON_ASSERT (bulk->session->client == client);
+   }
+
    bulk->client = (mongoc_client_t *) client;
 
    /* if you call set_client, bulk was likely made by mongoc_bulk_operation_new,
@@ -852,6 +856,22 @@ mongoc_bulk_operation_set_client (mongoc_bulk_operation_t *bulk, void *client)
 }
 
 
+void
+mongoc_bulk_operation_set_client_session (
+   mongoc_bulk_operation_t *bulk,
+   struct _mongoc_client_session_t *client_session)
+{
+   BSON_ASSERT (bulk);
+   BSON_ASSERT (client_session);
+
+   if (bulk->client) {
+      BSON_ASSERT (bulk->client == client_session->client);
+   }
+
+   bulk->session = client_session;
+}
+
+
 uint32_t
 mongoc_bulk_operation_get_hint (const mongoc_bulk_operation_t *bulk)
 {

src/mongoc/mongoc-bulk-operation.h

@@ -31,6 +31,8 @@
 
 BSON_BEGIN_DECLS
 
+/* forward decl */
+struct _mongoc_client_session_t;
 
 typedef struct _mongoc_bulk_operation_t mongoc_bulk_operation_t;
 typedef struct _mongoc_bulk_write_flags_t mongoc_bulk_write_flags_t;
@@ -130,6 +132,10 @@ mongoc_bulk_operation_set_collection (mongoc_bulk_operation_t *bulk,
                                       const char *collection);
 MONGOC_EXPORT (void)
 mongoc_bulk_operation_set_client (mongoc_bulk_operation_t *bulk, void *client);
+MONGOC_EXPORT (void)
+mongoc_bulk_operation_set_client_session (
+   mongoc_bulk_operation_t *bulk,
+   struct _mongoc_client_session_t *client_session);
 /* These names include the term "hint" for backward compatibility, should be
  * mongoc_bulk_operation_get_server_id, mongoc_bulk_operation_set_server_id. */
 MONGOC_EXPORT (void)