/**
* The cursor constructor
* @param string|EMongoDocument $modelClass - The class name for the active record
* @param array|MongoCursor|EMongoCriteria $criteria - Either a condition array (without sort,limit and skip) or a MongoCursor Object
* @param array $fields
*/
public function __construct($modelClass, $criteria = array(), $fields = array())
{
// If $fields has something in it
if (!empty($fields)) {
$this->partial = true;
}
if (is_string($modelClass)) {
$this->modelClass = $modelClass;
$this->model = EMongoDocument::model($this->modelClass);
} elseif ($modelClass instanceof EMongoDocument) {
$this->modelClass = get_class($modelClass);
$this->model = $modelClass;
}
if ($criteria instanceof MongoCursor) {
$this->cursor = $criteria;
$this->cursor->reset();
} elseif ($criteria instanceof EMongoCriteria) {
$this->criteria = $criteria;
$this->cursor = $this->model->getCollection()->find($criteria->condition, $criteria->project)->sort($criteria->sort);
if ($criteria->skip > 0) {
$this->cursor->skip($criteria->skip);
}
if ($criteria->limit > 0) {
$this->cursor->limit($criteria->limit);
}
} else {
// Then we are doing an active query
$this->criteria = $criteria;
$this->cursor = $this->model->getCollection()->find($criteria, $fields);
}
}
/**
* @param \MongoCursor $cursor
* @param ODM $odm
* @param string|null $class
* @param array $sort
* @param int $limit
* @param int $offset
*/
public function __construct(\MongoCursor $cursor, ODM $odm, $class, array $sort = [], $limit = null, $offset = null)
{
$this->cursor = $cursor;
$this->odm = $odm;
$this->class = $class;
!empty($sort) && $this->cursor->sort($sort);
!empty($limit) && $this->cursor->limit($limit);
!empty($offset) && $this->cursor->skip($offset);
}
/**
* Returns an array of items for a page.
*
* @param int $offset Page offset
* @param int $itemCountPerPage Number of items per page
* @return array
*/
public function getItems($offset, $itemCountPerPage)
{
// $cursor = clone $this -> cursor;
$this->cursor->skip($offset);
$this->cursor->limit($itemCountPerPage);
$resultSet = clone $this->resultSetPrototype;
$resultSet->initialize($this->createResult($this->cursor));
return $resultSet;
}
/**
* @see CActiveDataProvider::fetchData()
* @return array
*/
public function fetchData()
{
$criteria = $this->getCriteria();
// I have not refactored this line considering that the condition may have changed from total item count to here, maybe.
$this->_builder = new EMongoQueryBuilder($this->model, isset($criteria['condition']) && is_array($criteria['condition']) ? $criteria['condition'] : [], isset($criteria['project']) && !empty($criteria['project']) ? $criteria['project'] : []);
$this->options['condition'] = isset($criteria['condition']) && is_array($criteria['condition']) ? $criteria['condition'] : [];
$this->options['projection'] = isset($criteria['project']) && !empty($criteria['project']) ? $criteria['project'] : [];
// If we have sort and limit and skip setup within the incoming criteria let's set it
if (isset($criteria['sort']) && is_array($criteria['sort'])) {
$this->_builder->sort($criteria['sort']);
}
if (isset($criteria['skip']) && is_int($criteria['skip'])) {
$this->_builder->skip($criteria['skip']);
}
if (isset($criteria['limit']) && is_int($criteria['limit'])) {
$this->_builder->limit($criteria['limit']);
}
if (isset($criteria['hint']) && (is_array($criteria['hint']) || is_string($criteria['hint']))) {
$this->_builder->hint($criteria['hint']);
}
if (($pagination = $this->getPagination()) !== false) {
$pagination->setItemCount($this->getTotalItemCount());
$this->_builder->limit($pagination->getLimit());
$this->_builder->skip($pagination->getOffset());
}
if (($sort = $this->getSort()) !== false) {
$sort = $sort->getOrderBy();
if (count($sort) > 0) {
$this->_builder->sort($sort);
}
}
// var_dump(iterator_to_array($this->_builder->find()));die;
return $this->_builder->queryAll(true);
}
/**
* Skip the specified number of documents
*
* @since v1.0.0
*/
public function skip($x = FALSE)
{
if ($x !== FALSE && is_numeric($x) && $x >= 1) {
return $this->_cursor->skip((int) $x);
}
return $this->_cursor;
}
/**
* Wrapper method for MongoCursor::skip().
*
* @see http://php.net/manual/en/mongocursor.skip.php
* @param integer $num
* @return self
*/
public function skip($num)
{
$num = (int) $num;
$this->skip = $num;
$this->mongoCursor->skip($num);
return $this;
}
/**
* @see CActiveDataProvider::fetchData()
* @return array
*/
public function fetchData()
{
$criteria = $this->getCriteria();
// I have not refactored this line considering that the condition may have changed from total item count to here, maybe.
$this->_cursor = $this->model->find(isset($criteria['condition']) && is_array($criteria['condition']) ? $criteria['condition'] : array(), isset($criteria['project']) && !empty($criteria['project']) ? $criteria['project'] : array());
// If we have sort and limit and skip setup within the incoming criteria let's set it
if (isset($criteria['sort']) && is_array($criteria['sort'])) {
$this->_cursor->sort($criteria['sort']);
}
if (isset($criteria['skip']) && is_int($criteria['skip'])) {
$this->_cursor->skip($criteria['skip']);
}
if (isset($criteria['limit']) && is_int($criteria['limit'])) {
$this->_cursor->limit($criteria['limit']);
}
if (isset($criteria['hint']) && (is_array($criteria['hint']) || is_string($criteria['hint']))) {
$this->_cursor->hint($criteria['hint']);
}
if (($pagination = $this->getPagination()) !== false) {
$pagination->setItemCount($this->getTotalItemCount());
$this->_cursor->limit($pagination->getLimit());
$this->_cursor->skip($pagination->getOffset());
}
if (($sort = $this->getSort()) !== false) {
$sort = $sort->getOrderBy();
if (count($sort) > 0) {
$this->_cursor->sort($sort);
}
}
return iterator_to_array($this->_cursor, false);
}
public function skip($offset)
{
if ($offset > 0 && $this->resource instanceof \MongoCursor) {
return $this->resource->skip($offset);
}
return $this->resource;
}
/**
* Performs proper skip and limit to get
* data package that can be wraped in paginator
*
* @param int $perPage
* @param int $page
* @param mixed $options Options you want to pass
*/
public function getPaginator($perPage = 10, $page = 1, $options = null)
{
$this->_checkCursor();
$total = $this->_cursor->count();
$this->_cursor->skip(($page - 1) * $perPage)->limit($perPage);
$result = $this->get();
return $this->_createPaginator($result, $total, $perPage, $page, $options);
}
/**
* Apply the limit,skip,sort options to the cursor before iterating on documents
*/
protected function _apply_cursor_controls()
{
if ($this->_Cursor) {
if ($this->_cursor_controls['limit'] !== null) {
$this->_Cursor->limit($this->_cursor_controls['limit']);
}
if ($this->_cursor_controls['skip'] !== null) {
$this->_Cursor->skip($this->_cursor_controls['skip']);
}
if (!empty($this->_cursor_controls['sort'])) {
$this->_Cursor->sort($this->_cursor_controls['sort']);
}
}
}
public function configureCursor()
{
if ($this->cursor === null) {
return;
}
if ($this->getOffset() > -1) {
$this->cursor->skip($this->getOffset());
}
if ($this->getLimit() > -1) {
$this->cursor->limit($this->getLimit());
}
if ($this->getOrder() !== null) {
$this->cursor->sort($this->getOrder());
}
}
/**
* Create MongoCursor from string query log
*
* @param string $queryString
* @return \MongoCursor|null
*/
protected function getCursorFromQueryLog($queryString)
{
$cursor = null;
$connection = $this->panel->getDb();
$connection->open();
if ($connection->isActive) {
$queryInfo = Json::decode($queryString);
$query = $this->prepareQuery(isset($queryInfo['query']['$query']) ? $queryInfo['query']['$query'] : $queryInfo['query']);
$cursor = new \MongoCursor($connection->mongoClient, $queryInfo['ns'], $query, $queryInfo['fields']);
$cursor->limit($queryInfo['limit']);
$cursor->skip($queryInfo['skip']);
if (isset($queryInfo['query']['$orderby'])) {
$cursor->sort($queryInfo['query']['$orderby']);
}
}
return $cursor;
}
/**
* {@inheritDoc}
*/
public function getCursor()
{
if (is_null($this->cursor)) {
$rawCollection = $this->connection->getRaw()->{$this->collection->getName()};
if (empty($this->criteria)) {
$this->cursor = $rawCollection->find();
} else {
$this->cursor = $rawCollection->find($this->criteria);
}
if (isset($this->sorts)) {
$this->cursor->sort($this->sorts);
}
if (isset($this->skip)) {
$this->cursor->skip($this->skip);
}
if (isset($this->limit)) {
$this->cursor->limit($this->limit);
}
}
return $this->cursor;
}
/**
*
* @return \MongoCursor
*/
private function getCursor()
{
if ($this->cursor) {
return $this->cursor;
}
$this->cursor = $this->collection->getMongoCollection()->find($this->expression->toArray(), $this->fields);
if ($this->skip) {
$this->cursor->skip($this->skip);
}
if ($this->limit) {
$this->cursor->limit($this->limit);
}
if ($this->options['batchSize']) {
$this->cursor->batchSize($this->options['batchSize']);
}
if ($this->options['clientTimeout']) {
$this->cursor->timeout($this->options['clientTimeout']);
}
if ($this->options['serverTimeout']) {
$this->cursor->maxTimeMS($this->options['clientTimeout']);
}
if ($this->sort) {
$this->cursor->sort($this->sort);
}
if ($this->hint) {
$this->cursor->hint($this->hint);
}
// log request
if ($this->client->hasLogger()) {
$this->client->getLogger()->debug(get_called_class() . ': ' . json_encode(array('collection' => $this->collection->getName(), 'query' => $this->expression->toArray(), 'project' => $this->fields, 'sort' => $this->sort)));
}
$this->cursor->rewind();
// define read preferences
if ($this->readPreference) {
$this->cursor->setReadPreference($this->readPreference['type'], $this->readPreference['tagsets']);
}
return $this->cursor;
}
public function skip($num)
{
parent::skip($num);
return $this;
}
public function skip($skip)
{
parent::skip($skip);
return $this;
}
/**
* Skip a $offset records
* {@see http://www.php.net/manual/en/mongocursor.skip.php}
* @param integer $skip new skip
* @since v1.3.4
*/
public function offset($offset)
{
$this->_cursor->skip($offset);
}
/**
* {@inheritDoc}
*
* @api
*/
public function getItems($offset, $limit)
{
$this->cursor->skip($offset);
$this->cursor->limit($limit);
return iterator_to_array($this->cursor);
}
/**
* Applies options to cursor.
*
* @param \MongoCursor $cursor
* @param array $options
*
* @return \MongoCursor
*/
private function applyOptionsToMongoCursor(\MongoCursor &$cursor, $options = [])
{
if (isset($options['limit']['limit'])) {
$cursor->limit($options['limit']['limit']);
}
if (isset($options['limit']['offset'])) {
$cursor->skip($options['limit']['offset']);
}
if (isset($options['sort'])) {
$cursor->sort($options['sort']);
}
return $cursor;
}
/**
* Skip the first $num results
*
* @param integer $num
* @return Result
*/
public function skip($num)
{
$this->myResultCursor->skip($num);
return $this;
}
/**
* Skips a number of results
* @link http://www.php.net/manual/en/mongocursor.skip.php
* @param int $num The number of results to skip.
* @throws MongoCursorException
* @return $this Returns this cursor
*/
public function skip($num)
{
$this->cursor->skip($num);
return $this;
}
