1
0
mirror of synced 2024-12-13 14:56:01 +03:00
doctrine2/lib/Doctrine/Search.php

268 lines
8.7 KiB
PHP
Raw Normal View History

2007-07-09 17:26:20 +04:00
<?php
/*
* $Id$
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* This software consists of voluntary contributions made by many individuals
* and is licensed under the LGPL. For more information, see
* <http://www.phpdoctrine.org>.
2007-07-09 17:26:20 +04:00
*/
/**
* Doctrine_Search
*
* @package Doctrine
* @subpackage Search
* @author Konsta Vesterinen <kvesteri@cc.hut.fi>
2007-07-09 17:26:20 +04:00
* @license http://www.opensource.org/licenses/lgpl-license.php LGPL
* @version $Revision$
* @link www.phpdoctrine.com
* @since 1.0
*/
2007-11-29 01:56:45 +03:00
class Doctrine_Search extends Doctrine_Record_Generator
2007-07-09 17:26:20 +04:00
{
2007-10-15 00:32:48 +04:00
const INDEX_FILES = 0;
2007-10-15 02:11:48 +04:00
const INDEX_TABLES = 1;
2007-10-15 00:32:48 +04:00
protected $_options = array('generateFiles' => false,
2007-10-15 02:11:48 +04:00
'type' => self::INDEX_TABLES,
2007-10-13 12:39:26 +04:00
'className' => '%CLASS%Index',
2007-10-13 12:55:06 +04:00
'generatePath' => false,
'table' => null,
2007-10-13 12:55:06 +04:00
'batchUpdates' => false,
2007-10-15 00:32:48 +04:00
'pluginTable' => false,
2007-10-15 02:11:48 +04:00
'fields' => array(),
'connection' => null,
'children' => array());
/**
* __construct
*
* @param array $options
* @return void
*/
2007-07-11 17:20:39 +04:00
public function __construct(array $options)
{
$this->_options = Doctrine_Lib::arrayDeepMerge($this->_options, $options);
if ( ! isset($this->_options['analyzer'])) {
$this->_options['analyzer'] = new Doctrine_Search_Analyzer_Standard();
}
2007-10-15 02:11:48 +04:00
if ( ! isset($this->_options['connection'])) {
$this->_options['connection'] = Doctrine_Manager::connection();
}
2007-07-11 17:20:39 +04:00
}
2007-10-17 02:10:49 +04:00
/**
* search
*
* @param string $query
* @return Doctrine_Collection The collection of search results
*/
2007-10-17 02:10:49 +04:00
public function search($query)
{
$q = new Doctrine_Search_Query($this->_table);
2007-10-17 02:10:49 +04:00
2007-10-18 01:17:01 +04:00
$q->query($query);
2007-10-17 02:10:49 +04:00
2007-10-18 01:17:01 +04:00
return $this->_options['connection']->fetchAll($q->getSql(), $q->getParams());;
2007-10-17 02:10:49 +04:00
}
2007-07-11 17:20:39 +04:00
/**
* analyze
*
* @param string $text
* @return void
*/
public function analyze($text)
{
2007-07-12 02:03:47 +04:00
return $this->_options['analyzer']->analyze($text);
}
2007-07-25 00:38:21 +04:00
/**
* updateIndex
* updates the index
*
* @param Doctrine_Record $record
* @return integer
*/
public function updateIndex(array $data)
2007-07-12 02:03:47 +04:00
{
$this->initialize($this->_options['table']);
$fields = $this->getOption('fields');
2007-07-12 02:03:47 +04:00
$class = $this->getOption('className');
$name = $this->getOption('table')->getComponentName();
$conn = $this->getOption('table')->getConnection();
$identifier = $this->_options['table']->getIdentifier();
2007-10-19 00:39:37 +04:00
$q = Doctrine_Query::create()->delete()
->from($class);
foreach ((array) $identifier as $id) {
$q->addWhere($id . ' = ?', array($data[$id]));
}
$q->execute();
2007-07-12 02:03:47 +04:00
2007-10-13 12:55:06 +04:00
if ($this->_options['batchUpdates'] === true) {
$index = new $class();
2007-10-18 23:37:50 +04:00
foreach ((array) $this->_options['table']->getIdentifier() as $id) {
$index->$id = $data[$id];
}
$index->save();
2007-10-13 12:55:06 +04:00
} else {
foreach ($fields as $field) {
$value = $data[$field];
$terms = $this->analyze($value);
2007-10-13 12:55:06 +04:00
foreach ($terms as $pos => $term) {
$index = new $class();
2007-10-13 12:55:06 +04:00
$index->keyword = $term;
$index->position = $pos;
$index->field = $field;
foreach ((array) $this->_options['table']->getIdentifier() as $id) {
$index->$id = $data[$id];
}
2007-10-13 12:55:06 +04:00
$index->save();
}
2007-07-12 02:03:47 +04:00
}
}
}
2007-09-29 19:00:08 +04:00
/**
* readTableData
*
* @param mixed $limit
* @param mixed $offset
* @return Doctrine_Collection The collection of results
*/
2007-10-15 00:32:48 +04:00
public function readTableData($limit = null, $offset = null)
{
$this->initialize($this->_options['table']);
$conn = $this->_options['table']->getConnection();
$tableName = $this->_options['table']->getTableName();
$id = $this->_options['table']->getIdentifier();
2007-10-15 02:11:48 +04:00
$query = 'SELECT * FROM ' . $conn->quoteIdentifier($tableName)
. ' WHERE ' . $conn->quoteIdentifier($id)
. ' IN (SELECT ' . $conn->quoteIdentifier($id)
. ' FROM ' . $conn->quoteIdentifier($this->_table->getTableName())
2007-10-15 02:11:48 +04:00
. ' WHERE keyword IS NULL)';
$query = $conn->modifyLimitQuery($query, $limit, $offset);
return $conn->fetchAll($query);
2007-10-15 00:32:48 +04:00
}
2007-10-15 00:32:48 +04:00
/**
* batchUpdateIndex
*
* @param mixed $limit
* @param mixed $offset
* @return void
*/
2007-10-18 01:17:01 +04:00
public function batchUpdateIndex($limit = null, $offset = null)
2007-10-13 21:52:16 +04:00
{
$this->initialize($this->_options['table']);
$id = $this->_options['table']->getIdentifier();
2007-10-15 02:11:48 +04:00
$class = $this->_options['className'];
$fields = $this->_options['fields'];
$conn = $this->_options['connection'];
2007-10-13 21:52:16 +04:00
try {
2007-09-29 19:00:08 +04:00
2007-10-13 21:52:16 +04:00
$conn->beginTransaction();
2007-10-15 02:11:48 +04:00
$rows = $this->readTableData($limit, $offset);
2007-10-15 00:32:48 +04:00
foreach ($rows as $row) {
$ids[] = $row[$id];
}
$conn->exec('DELETE FROM '
. $conn->quoteIdentifier($this->_table->getTableName())
2007-10-15 00:32:48 +04:00
. ' WHERE ' . $conn->quoteIdentifier($id) . ' IN (' . implode(', ', $ids) . ')');
2007-10-13 21:52:16 +04:00
foreach ($rows as $row) {
foreach ($fields as $field) {
$data = $row[$field];
$terms = $this->analyze($data);
foreach ($terms as $pos => $term) {
$index = new $class();
$index->keyword = $term;
$index->position = $pos;
$index->field = $field;
2007-10-15 00:32:48 +04:00
foreach ((array) $id as $identifier) {
$index->$identifier = $row[$identifier];
2007-10-13 21:52:16 +04:00
}
$index->save();
}
2007-09-29 19:00:08 +04:00
}
}
2007-10-13 21:52:16 +04:00
$conn->commit();
} catch (Doctrine_Exception $e) {
$conn->rollback();
2007-09-29 19:00:08 +04:00
}
}
2007-09-29 18:45:38 +04:00
/**
* buildDefinition
*
* @return void
*/
public function setTableDefinition()
2007-07-11 15:27:02 +04:00
{
if ( ! isset($this->_options['table'])) {
throw new Doctrine_Plugin_Exception("Unknown option 'table'.");
}
$componentName = $this->_options['table']->getComponentName();
2007-07-25 00:38:21 +04:00
$className = $this->getOption('className');
2007-10-13 21:52:16 +04:00
2007-07-13 01:50:31 +04:00
if (class_exists($className)) {
return false;
}
2007-07-09 17:26:20 +04:00
2007-07-11 15:27:02 +04:00
$columns = array('keyword' => array('type' => 'string',
'length' => 200,
2007-07-18 00:59:09 +04:00
'primary' => true,
),
2007-07-11 15:27:02 +04:00
'field' => array('type' => 'string',
'length' => 50,
2007-07-18 00:59:09 +04:00
'primary' => true),
2007-07-11 15:27:02 +04:00
'position' => array('type' => 'integer',
2007-07-18 00:59:09 +04:00
'length' => 8,
'primary' => true,
));
2007-07-11 15:27:02 +04:00
$this->hasColumns($columns);
2007-07-11 15:27:02 +04:00
}
2007-11-29 01:56:45 +03:00
}