CDRIVER-837: Generate mongoc_collection_find futures

Author: bjori

build/generate-future-functions.py

@@ -129,6 +129,16 @@
                      param("const_bson_ptr", "options"),
                      param("const_mongoc_read_prefs_ptr", "read_prefs")]),
 
+    future_function("int64_t",
+                    "mongoc_collection_count",
+                    [param("mongoc_collection_ptr", "collection"),
+                     param("mongoc_query_flags_t", "flags"),
+                     param("const_bson_ptr", "query"),
+                     param("int64_t", "skip"),
+                     param("int64_t", "limit"),
+                     param("const_mongoc_read_prefs_ptr", "read_prefs"),
+                     param("bson_error_ptr", "error")]),
+
     future_function("bool",
                     "mongoc_collection_find_and_modify_with_opts",
                     [param("mongoc_collection_ptr", "collection"),

tests/mock_server/future-functions.c

@@ -110,6 +110,31 @@ background_mongoc_collection_aggregate (void *data)
    return NULL;
 }
 
+static void *
+background_mongoc_collection_count (void *data)
+{
+   future_t *future = (future_t *) data;
+   future_value_t return_value;
+
+   return_value.type = future_value_int64_t_type;
+
+   future_value_set_int64_t (
+      &return_value,
+      mongoc_collection_count (
+         future_value_get_mongoc_collection_ptr (future_get_param (future, 0)),
+         future_value_get_mongoc_query_flags_t (future_get_param (future, 1)),
+         future_value_get_const_bson_ptr (future_get_param (future, 2)),
+         future_value_get_int64_t (future_get_param (future, 3)),
+         future_value_get_int64_t (future_get_param (future, 4)),
+         future_value_get_const_mongoc_read_prefs_ptr (future_get_param (future, 5)),
+         future_value_get_bson_error_ptr (future_get_param (future, 6))
+      ));
+
+   future_resolve (future, return_value);
+
+   return NULL;
+}
+
 static void *
 background_mongoc_collection_find_and_modify_with_opts (void *data)
 {
@@ -478,6 +503,44 @@ future_collection_aggregate (
    return future;
 }
 
+future_t *
+future_collection_count (
+   mongoc_collection_ptr collection,
+   mongoc_query_flags_t flags,
+   const_bson_ptr query,
+   int64_t skip,
+   int64_t limit,
+   const_mongoc_read_prefs_ptr read_prefs,
+   bson_error_ptr error)
+{
+   future_t *future = future_new (future_value_int64_t_type,
+                                  7);
+   
+   future_value_set_mongoc_collection_ptr (
+      future_get_param (future, 0), collection);
+   
+   future_value_set_mongoc_query_flags_t (
+      future_get_param (future, 1), flags);
+   
+   future_value_set_const_bson_ptr (
+      future_get_param (future, 2), query);
+   
+   future_value_set_int64_t (
+      future_get_param (future, 3), skip);
+   
+   future_value_set_int64_t (
+      future_get_param (future, 4), limit);
+   
+   future_value_set_const_mongoc_read_prefs_ptr (
+      future_get_param (future, 5), read_prefs);
+   
+   future_value_set_bson_error_ptr (
+      future_get_param (future, 6), error);
+   
+   future_start (future, background_mongoc_collection_count);
+   return future;
+}
+
 future_t *
 future_collection_find_and_modify_with_opts (
    mongoc_collection_ptr collection,

tests/mock_server/future-functions.h

@@ -54,6 +54,19 @@ future_collection_aggregate (
 );
 
 
+future_t *
+future_collection_count (
+
+   mongoc_collection_ptr collection,
+   mongoc_query_flags_t flags,
+   const_bson_ptr query,
+   int64_t skip,
+   int64_t limit,
+   const_mongoc_read_prefs_ptr read_prefs,
+   bson_error_ptr error
+);
+
+
 future_t *
 future_collection_find_and_modify_with_opts (