bool
_mongoc_cursor_cursorid_next (mongoc_cursor_t *cursor, const bson_t **bson)
{
mongoc_cursor_cursorid_t *cid;
bool refreshed = false;
ENTRY;
*bson = NULL;
cid = (mongoc_cursor_cursorid_t *) cursor->iface_data;
BSON_ASSERT (cid);
if (!cursor->sent) {
if (!_mongoc_cursor_cursorid_prime (cursor)) {
GOTO (done);
}
}
again:
/* Two paths:
* - Mongo 3.2+, sent "getMore" cmd, we're reading reply's "nextBatch" array
* - Mongo 2.6 to 3, after "aggregate" or similar command we sent OP_GETMORE,
* we're reading the raw reply
*/
if (cid->in_batch) {
_mongoc_cursor_cursorid_read_from_batch (cursor, bson);
if (*bson) {
GOTO (done);
}
cid->in_batch = false;
} else if (cid->in_reader) {
_mongoc_read_from_buffer (cursor, bson);
if (*bson) {
GOTO (done);
}
cid->in_reader = false;
}
if (!refreshed && mongoc_cursor_get_id (cursor)) {
if (!_mongoc_cursor_cursorid_get_more (cursor)) {
GOTO (done);
}
refreshed = true;
GOTO (again);
}
done:
if (!*bson && mongoc_cursor_get_id (cursor) == 0) {
cursor->done = 1;
}
RETURN (*bson != NULL);
}
bool
_mongoc_cursor_prepare_getmore_command (mongoc_cursor_t *cursor,
bson_t *command)
{
const char *collection;
int collection_len;
ENTRY;
_mongoc_cursor_collection (cursor, &collection, &collection_len);
bson_init (command);
bson_append_int64 (command, "getMore", 7, mongoc_cursor_get_id (cursor));
bson_append_utf8 (command, "collection", 10, collection, collection_len);
if (cursor->batch_size) {
bson_append_int64 (command, "batchSize", 9, cursor->batch_size);
}
/* Find, getMore And killCursors Commands Spec: "In the case of a tailable
cursor with awaitData == true the driver MUST provide a Cursor level
option named maxAwaitTimeMS (See CRUD specification for details). The
maxTimeMS option on the getMore command MUST be set to the value of the
option maxAwaitTimeMS. If no maxAwaitTimeMS is specified, the driver
SHOULD not set maxTimeMS on the getMore command."
*/
if (cursor->flags & MONGOC_QUERY_TAILABLE_CURSOR &&
cursor->flags & MONGOC_QUERY_AWAIT_DATA &&
cursor->max_await_time_ms) {
bson_append_int32 (command, "maxTimeMS", 9, cursor->max_await_time_ms);
}
RETURN (true);
}
/* Check if the cursor is exhausted (i.e. ID is zero) and free any reference to
* the session. Calling this function during iteration will allow an implicit
* session to return to the pool immediately after a getMore indicates that the
* server has no more results to return. */
static void php_phongo_cursor_free_session_if_exhausted(php_phongo_cursor_t* cursor) /* {{{ */
{
if (mongoc_cursor_get_id(cursor->cursor)) {
return;
}
if (!Z_ISUNDEF(cursor->session)) {
zval_ptr_dtor(&cursor->session);
ZVAL_UNDEF(&cursor->session);
}
} /* }}} */
bool
_mongoc_cursor_prepare_getmore_command (mongoc_cursor_t *cursor,
bson_t *command)
{
const char *collection;
int collection_len;
int64_t batch_size;
bool await_data;
int32_t max_await_time_ms;
ENTRY;
_mongoc_cursor_collection (cursor, &collection, &collection_len);
bson_init (command);
bson_append_int64 (command, "getMore", 7, mongoc_cursor_get_id (cursor));
bson_append_utf8 (command, "collection", 10, collection, collection_len);
batch_size = mongoc_cursor_get_batch_size (cursor);
/* See find, getMore, and killCursors Spec for batchSize rules */
if (batch_size) {
bson_append_int64 (
command, MONGOC_CURSOR_BATCH_SIZE, MONGOC_CURSOR_BATCH_SIZE_LEN, abs (_mongoc_n_return (cursor)));
}
/* Find, getMore And killCursors Commands Spec: "In the case of a tailable
cursor with awaitData == true the driver MUST provide a Cursor level
option named maxAwaitTimeMS (See CRUD specification for details). The
maxTimeMS option on the getMore command MUST be set to the value of the
option maxAwaitTimeMS. If no maxAwaitTimeMS is specified, the driver
SHOULD not set maxTimeMS on the getMore command."
*/
await_data = _mongoc_cursor_get_opt_bool (cursor, MONGOC_CURSOR_TAILABLE) &&
_mongoc_cursor_get_opt_bool (cursor, MONGOC_CURSOR_AWAIT_DATA);
if (await_data) {
max_await_time_ms =
(int32_t) mongoc_cursor_get_max_await_time_ms (cursor);
if (max_await_time_ms) {
bson_append_int32 (
command, MONGOC_CURSOR_MAX_TIME_MS, MONGOC_CURSOR_MAX_TIME_MS_LEN, max_await_time_ms);
}
}
RETURN (true);
}
Object HHVM_METHOD(MongoDBDriverCursor, getId)
{
static Class* c_cursor;
int64_t cursorid;
MongoDBDriverCursorData* data = Native::data<MongoDBDriverCursorData>(this_);
cursorid = mongoc_cursor_get_id(data->cursor);
/* Prepare result */
c_cursor = Unit::lookupClass(s_MongoDriverCursorId_className.get());
assert(c_cursor);
Object obj = Object{c_cursor};
MongoDBDriverCursorIdData* cursorid_data = Native::data<MongoDBDriverCursorIdData>(obj.get());
cursorid_data->id = cursorid;
return obj;
}
