markusos/simple-search

<?php namespace Search\Store;

use Search\Document;
use Search\Index\DocumentIndex;

class MongoDBDocumentStore implements DocumentStore
{

    /**
     * @var \MongoClient
     */
    private $connection;
    /**
     * @var \MongoCollection
     */
    private $documents;

    private $size;

    function __construct($host='localhost', $port=27017)
    {
        $this->connection = new \MongoClient('mongodb://'. $host .':'. $port);
        $this->documents = $this->connection->search->documents;
        $this->size = $this->documents->count();
    }

    /**
     * Initialize a DocumentIndex with all documents stored in the DocumentStore
     * @param DocumentIndex $index Search index to initialize
     * @return DocumentIndex initialized with documents form the DocumentStore
     */
    public function buildIndex(DocumentIndex $index)
    {
        $cursor = $this->documents->find();
        foreach ($cursor as $result) {
            $document = new Document($result['title'], $result['content']);
            $document->id = $result['id'];
            $document->tokens = $result['tokens'];
            $index->addDocument($document);
        }
        return $index;
    }

    /**
     * Add a new Document to the DocumentStore
     * @param Document $document Document to add
     */
    public function addDocument(Document $document)
    {
        $id = new \MongoId();
        $document->_id = $id;
        $this->documents->insert($document);
        unset($document->_id);
        $this->size++;
    }

    /**
     * Get a Document from the DocumentStore
     * @param integer $id ID of Document to get
     * @return Document Document matching $id
     */
    public function getDocument($id)
    {
        $documents = $this->getDocuments([$id]);
        if (isset($documents[$id])) {
            return $documents[$id];
        } else {
            return null;
        }
    }

    /**
     * Get a Document from the DocumentStore
     * @param array $ids IDs of Documents to get
     * @return array Documents matching $ids
     */
    public function getDocuments($ids)
    {
        $results = [];
        $documents = $this->documents->find(['id' => ['$in' => $ids]]);

        foreach ($documents as $result) {
            $document = new Document($result['title'], $result['content']);
            $document->id = $result['id'];
            $document->tokens = $result['tokens'];
            $results[$document->id] = $document;
        }

        return $results;
    }

    /**
     * Get the number of stored documents
     * @return integer Number of stored documents
     */
    public function size()
    {
        return $this->size;
    }

    /**
     * Clear all stored documents
     */
    public function clear()
    {
        $this->size = 0;
        $this->documents->drop();
    }
}