void
mongoc_bulk_operation_delete_one (mongoc_bulk_operation_t *bulk,
const bson_t *selector)
{
ENTRY;
mongoc_bulk_operation_remove_one (bulk, selector);
EXIT;
}
int
lua_mongo_collection_delete_one(lua_State *L)
{
collection_t *collection;
bson_t selector = BSON_INITIALIZER;
mongoc_bulk_operation_t *bulk_remove = NULL;
bson_error_t error;
bson_t reply;
bool throw_error = false;
bool ret;
collection = (collection_t *)luaL_checkudata(L, 1, "lua_mongoc_collection");
if ((lua_istable(L, 2))) {
if (!(lua_table_to_bson(L, &selector, 2, false, &error))) {
throw_error = true;
goto DONE;
}
} else {
luaL_error(L, "second input must be a table");
}
bulk_remove = mongoc_collection_create_bulk_operation(collection->c_collection,
true, NULL);
mongoc_bulk_operation_remove_one(bulk_remove, &selector);
ret = mongoc_bulk_operation_execute(bulk_remove, &reply, &error);
if (!(ret)) {
throw_error = true;
goto DONE;
}
if (!(generate_DeleteResult(L, &reply, ret, &error))) {
throw_error = true;
goto DONE;
}
DONE:
bson_destroy(&selector);
if (bulk_remove) {
mongoc_bulk_operation_destroy (bulk_remove);
}
if (throw_error) {
luaL_error(L, error.message);
}
return 1;
}
static void
bulk3 (mongoc_collection_t *collection)
{
mongoc_bulk_operation_t *bulk;
bson_error_t error;
bson_t *query;
bson_t *doc;
bson_t reply;
char *str;
bool ret;
/* false indicates unordered */
bulk = mongoc_collection_create_bulk_operation (collection, false, NULL);
/* Add a document */
doc = BCON_NEW ("_id", BCON_INT32 (1));
mongoc_bulk_operation_insert (bulk, doc);
bson_destroy (doc);
/* remove {_id: 2} */
query = BCON_NEW ("_id", BCON_INT32 (2));
mongoc_bulk_operation_remove_one (bulk, query);
bson_destroy (query);
/* insert {_id: 3} */
doc = BCON_NEW ("_id", BCON_INT32 (3));
mongoc_bulk_operation_insert (bulk, doc);
bson_destroy (doc);
/* replace {_id:4} {'i': 1} */
query = BCON_NEW ("_id", BCON_INT32 (4));
doc = BCON_NEW ("i", BCON_INT32 (1));
mongoc_bulk_operation_replace_one (bulk, query, doc, false);
bson_destroy (query);
bson_destroy (doc);
ret = mongoc_bulk_operation_execute (bulk, &reply, &error);
str = bson_as_json (&reply, NULL);
printf ("%s\n", str);
bson_free (str);
if (!ret) {
printf ("Error: %s\n", error.message);
}
bson_destroy (&reply);
mongoc_bulk_operation_destroy (bulk);
}
static void
bulk6 (mongoc_collection_t *collection)
{
mongoc_write_concern_t *wc;
mongoc_bulk_operation_t *bulk;
bson_error_t error;
bson_t *doc;
bson_t *selector;
bson_t reply;
char *str;
bool ret;
wc = mongoc_write_concern_new ();
mongoc_write_concern_set_w (wc, 0);
bulk = mongoc_collection_create_bulk_operation (collection, true, wc);
doc = BCON_NEW ("_id", BCON_INT32 (10));
mongoc_bulk_operation_insert (bulk, doc);
bson_destroy (doc);
selector = BCON_NEW ("_id", BCON_INT32 (11));
mongoc_bulk_operation_remove_one (bulk, selector);
bson_destroy (selector);
ret = mongoc_bulk_operation_execute (bulk, &reply, &error);
str = bson_as_json (&reply, NULL);
printf ("%s\n", str);
bson_free (str);
if (!ret) {
printf ("Error: %s\n", error.message);
}
bson_destroy (&reply);
mongoc_bulk_operation_destroy (bulk);
mongoc_write_concern_destroy (wc);
}
static void
test_index_offset (void)
{
mongoc_bulk_operation_t *bulk;
mongoc_collection_t *collection;
mongoc_client_t *client;
bson_error_t error;
bson_iter_t iter;
bson_iter_t ar;
bson_iter_t citer;
bson_t reply;
bson_t *sel;
bson_t *doc;
bool r;
client = mongoc_client_new (gTestUri);
assert (client);
collection = get_test_collection (client, "test_index_offset");
assert (collection);
doc = bson_new ();
BSON_APPEND_INT32 (doc, "abcd", 1234);
r = mongoc_collection_insert (collection, MONGOC_INSERT_NONE, doc, NULL, &error);
assert (r);
bson_destroy (doc);
bulk = mongoc_collection_create_bulk_operation (collection, true, NULL);
assert (bulk);
sel = BCON_NEW ("abcd", BCON_INT32 (1234));
doc = BCON_NEW ("$set", "{", "hello", "there", "}");
mongoc_bulk_operation_remove_one (bulk, sel);
mongoc_bulk_operation_update (bulk, sel, doc, true);
r = mongoc_bulk_operation_execute (bulk, &reply, &error);
assert (r);
assert (bson_iter_init_find (&iter, &reply, "nUpserted"));
assert (BSON_ITER_HOLDS_INT32 (&iter));
assert (bson_iter_int32 (&iter) == 1);
assert (bson_iter_init_find (&iter, &reply, "nMatched"));
assert (BSON_ITER_HOLDS_INT32 (&iter));
assert (bson_iter_int32 (&iter) == 0);
assert (bson_iter_init_find (&iter, &reply, "nRemoved"));
assert (BSON_ITER_HOLDS_INT32 (&iter));
assert (bson_iter_int32 (&iter) == 1);
assert (bson_iter_init_find (&iter, &reply, "nInserted"));
assert (BSON_ITER_HOLDS_INT32 (&iter));
assert (bson_iter_int32 (&iter) == 0);
if (bson_iter_init_find (&iter, &reply, "nModified")) {
assert (BSON_ITER_HOLDS_INT32 (&iter));
assert (bson_iter_int32 (&iter) == 0);
}
assert (bson_iter_init_find (&iter, &reply, "upserted"));
assert (BSON_ITER_HOLDS_ARRAY (&iter));
assert (bson_iter_recurse (&iter, &ar));
assert (bson_iter_next (&ar));
assert (BSON_ITER_HOLDS_DOCUMENT (&ar));
assert (bson_iter_recurse (&ar, &citer));
assert (bson_iter_next (&citer));
assert (BSON_ITER_IS_KEY (&citer, "index"));
assert (bson_iter_int32 (&citer) == 1);
assert (bson_iter_next (&citer));
assert (BSON_ITER_IS_KEY (&citer, "_id"));
assert (BSON_ITER_HOLDS_OID (&citer));
assert (!bson_iter_next (&citer));
assert (!bson_iter_next (&ar));
assert (bson_iter_init_find (&iter, &reply, "writeErrors"));
assert (BSON_ITER_HOLDS_ARRAY (&iter));
assert (bson_iter_recurse (&iter, &ar));
assert (!bson_iter_next (&ar));
bson_destroy (&reply);
r = mongoc_collection_drop (collection, &error);
assert (r);
mongoc_bulk_operation_destroy (bulk);
mongoc_collection_destroy (collection);
mongoc_client_destroy (client);
bson_destroy (doc);
bson_destroy (sel);
}
