Files
luos/workflow/engine/classes/class.searchIndex.php
2012-11-30 10:47:17 -04:00

445 lines
17 KiB
PHP

<?php
/**
* ProcessMaker Open Source Edition
* Copyright (C) 2004 - 2012 Colosa Inc.23
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* For more information, contact Colosa Inc, 5304 Ventura Drive,
* Delray Beach, FL, 33484, USA, or email info@colosa.com.
*/
/**
* Class used as interface to have access to the search index services
*
* @author Herbert Saal Gutierrez
*
*/
class BpmnEngine_Services_SearchIndex
{
private $_solrIsEnabled = false;
private $_solrHost = "";
public function __construct ($solrIsEnabled = false, $solrHost = "")
{
// check if Zend Library is available
// if(class_exists("Zend_Registry")){
// $registry = Zend_Registry::getInstance();
// //check if configuration is enabled
// $this->solrIsEnabled = $registry->isRegistered('solrEnabled') &&
// $registry->get('solrEnabled') == 1;
// $this->solrHost =
// $registry->isRegistered('solrHost')?$registry->get('solrHost'):"";
// }
// else{
// //use the parameters to initialize class
$this->_solrIsEnabled = $solrIsEnabled;
$this->_solrHost = $solrHost;
// }
}
/**
* Verify if the Solr service is available
* @gearman = false
* @rest = false
* @background = false
*
* no input parameters @param[in]
*
* @param [out] bool true if index service is enabled false in other case
*/
public function isEnabled ()
{
// require_once (ROOT_PATH .
// '/businessLogic/modules/SearchIndexAccess/Solr.php');
require_once ('class.solr.php');
$solr = new BpmnEngine_SearchIndexAccess_Solr( $this->_solrIsEnabled, $this->_solrHost );
return $solr->isEnabled();
}
/**
* Get the list of facets in base to the specified query and filter
* @gearman = true
* @rest = false
* @background = false
*
* @param [in] Entity_FacetRequest facetRequestEntity Facet request entity
* @param [out] array FacetGroup
*/
public function getFacetsList ($facetRequestEntity)
{
require_once ('class.solr.php');
// require_once (ROOT_PATH .
// '/businessLogic/modules/SearchIndexAccess/Solr.php');
require_once ('entities/FacetGroup.php');
require_once ('entities/FacetItem.php');
require_once ('entities/SelectedFacetGroupItem.php');
require_once ('entities/FacetResult.php');
/**
* ***************************************************************
*/
// get array of selected facet groups
$facetRequestEntity->selectedFacetsString = str_replace( ',,', ',', $facetRequestEntity->selectedFacetsString );
// remove descriptions of selected facet groups
$aGroups = explode( ',', $facetRequestEntity->selectedFacetsString );
$aGroups = array_filter( $aGroups ); // remove empty items
$aSelectedFacetGroups = array ();
foreach ($aGroups as $i => $value) {
$gi = explode( ':::', $value );
$gr = explode( '::', $gi[0] );
$it = explode( '::', $gi[1] );
// create string for remove condition
$count = 0;
$removeCondition = str_replace( $value . ',', '', $facetRequestEntity->selectedFacetsString, $count );
if ($count == 0) {
$removeCondition = str_replace( $value, '', $facetRequestEntity->selectedFacetsString, $count );
}
$selectedFacetGroupData = array ('selectedFacetGroupName' => $gr[0],'selectedFacetGroupPrintName' => $gr[1],'selectedFacetItemName' => $it[0],'selectedFacetItemPrintName' => $it[1],'selectedFacetRemoveCondition' => $removeCondition
);
$aSelectedFacetGroups[] = Entity_SelectedFacetGroupItem::createForRequest( $selectedFacetGroupData );
}
/**
* ***************************************************************
*/
// convert request to index request
// create filters
$filters = array ();
if (! empty( $aSelectedFacetGroups )) {
// exclude facetFields and facetDates included in filter from the next
// list of facets
foreach ($aSelectedFacetGroups as $value) {
$facetRequestEntity->facetFields = array_diff( $facetRequestEntity->facetFields, array ($value->selectedFacetGroupName
) );
$facetRequestEntity->facetDates = array_diff( $facetRequestEntity->facetDates, array ($value->selectedFacetGroupName
) );
}
// $facetFields = array_diff($facetFields,
// $facetInterfaceRequestEntity->selectedFacetGroups);
// $facetDates = array_diff($facetDates,
// $facetInterfaceRequestEntity->selectedFacetGroups);
foreach ($aSelectedFacetGroups as $group) {
$filters[] = $group->selectedFacetGroupName . ':' . urlencode( $group->selectedFacetItemName );
}
}
$facetRequestEntity->filters = $filters;
$solr = new BpmnEngine_SearchIndexAccess_Solr( $this->_solrIsEnabled, $this->_solrHost );
// create list of facets
$facetsList = $solr->getFacetsList( $facetRequestEntity );
$numFound = $facetsList->response->numFound;
$facetCounts = $facetsList->facet_counts;
$facetGroups = array ();
// convert facet fields result to objects
/**
* *********************************************************************
*/
// include facet field results
$facetFieldsResult = $facetsList->facet_counts->facet_fields;
if (! empty( $facetFieldsResult )) {
foreach ($facetFieldsResult as $facetGroup => $facetvalues) {
if (count( $facetvalues ) > 0) // if the group have facets included
{
$data = array ('facetGroupName' => $facetGroup
);
$data['facetGroupPrintName'] = $facetGroup;
$data['facetGroupType'] = 'field';
$facetItems = array ();
for ($i = 0; $i < count( $facetvalues ); $i += 2) {
$dataItem = array ();
$dataItem['facetName'] = $facetvalues[$i];
$dataItem['facetPrintName'] = $facetvalues[$i];
$dataItem['facetCount'] = $facetvalues[$i + 1];
$dataItem['facetSelectCondition'] = $facetRequestEntity->selectedFacetsString . (empty( $facetRequestEntity->selectedFacetsString ) ? '' : ',') . $data['facetGroupName'] . '::' . $data['facetGroupPrintName'] . ':::' . $dataItem['facetName'] . '::' . $dataItem['facetPrintName'];
$newFacetItem = Entity_FacetItem::createForInsert( $dataItem );
$facetItems[] = $newFacetItem;
}
$data['facetItems'] = $facetItems;
$newFacetGroup = Entity_FacetGroup::createForInsert( $data );
$facetGroups[] = $newFacetGroup;
}
}
}
/**
* *********************************************************************
*/
// include facet date ranges results
$facetDatesResult = $facetsList->facet_counts->facet_dates;
if (! empty( $facetDatesResult )) {
foreach ($facetDatesResult as $facetGroup => $facetvalues) {
if (count( (array) $facetvalues ) > 3) // if the group have any facets included
// besides start, end and gap
{
$data = array ('facetGroupName' => $facetGroup
);
$data['facetGroupPrintName'] = $facetGroup;
$data['facetGroupType'] = 'daterange';
$facetItems = array ();
$facetvalueskeys = array_keys( (array) $facetvalues );
foreach ($facetvalueskeys as $i => $k) {
if ($k != 'gap' && $k != 'start' && $k != 'end') {
$dataItem = array ();
if ($i < count( $facetvalueskeys ) - 4) {
$dataItem['facetName'] = '[' . $k . '%20TO%20' . $facetvalueskeys[$i + 1] . ']';
$dataItem['facetPrintName'] = '[' . $k . '%20TO%20' . $facetvalueskeys[$i + 1] . ']';
} else {
// the last group
$dataItem['facetName'] = '[' . $k . '%20TO%20' . $facetvalues->end . ']';
$dataItem['facetPrintName'] = '[' . $k . '%20TO%20' . $facetvalues->end . ']';
}
$dataItem['facetCount'] = $facetvalues->$k;
$dataItem['facetSelectCondition'] = $facetRequestEntity->selectedFacetsString . (empty( $facetRequestEntity->selectedFacetsString ) ? '' : ',') . $data['facetGroupName'] . '::' . $data['facetGroupPrintName'] . ':::' . $dataItem['facetName'] . '::' . $dataItem['facetPrintName'];
$newFacetItem = Entity_FacetItem::createForInsert( $dataItem );
$facetItems[] = $newFacetItem;
}
}
$data['facetItems'] = $facetItems;
$newFacetGroup = Entity_FacetGroup::createForInsert( $data );
$facetGroups[] = $newFacetGroup;
}
}
}
// TODO:convert facet queries
// -----
/**
* ***************************************************************
*/
// Create a filter string used in the filter of results of a datatable
$filterText = ''; // the list of selected filters used for filtering result,
// send in ajax
foreach ($aSelectedFacetGroups as $selectedFacetGroup) {
$filterText .= $selectedFacetGroup->selectedFacetGroupName . ':' . urlencode( $selectedFacetGroup->selectedFacetItemName ) . ',';
}
$filterText = substr_replace( $filterText, '', - 1 );
// $filterText = ($filterText == '')?'':'&filterText='.$filterText;
/**
* ***************************************************************
*/
// Create result
$dataFacetResult = array ('aFacetGroups' => $facetGroups,'aSelectedFacetGroups' => $aSelectedFacetGroups,'sFilterText' => $filterText
);
$facetResult = Entity_FacetResult::createForRequest( $dataFacetResult );
return $facetResult;
}
/**
* Get the total number of documents in search server
*
* @param string $workspace
* @return integer number of documents
*
*/
public function getNumberDocuments ($workspace)
{
require_once ('class.solr.php');
// require_once (ROOT_PATH .
// '/businessLogic/modules/SearchIndexAccess/Solr.php');
$solr = new BpmnEngine_SearchIndexAccess_Solr( $this->_solrIsEnabled, $this->_solrHost );
// create list of facets
$numberDocuments = $solr->getNumberDocuments( $workspace );
return $numberDocuments;
}
/**
* Update document Index
*
* @param SolrUpdateDocumentEntity $solrUpdateDocumentEntity
*/
public function updateIndexDocument ($solrUpdateDocumentEntity)
{
G::LoadClass( 'solr' );
$solr = new BpmnEngine_SearchIndexAccess_Solr( $this->_solrIsEnabled, $this->_solrHost );
// create list of facets
$solr->updateDocument( $solrUpdateDocumentEntity );
}
/**
* Delete document from index
*
* @param string $workspace
* @param string $idQuery
*/
public function deleteDocumentFromIndex ($workspace, $idQuery)
{
G::LoadClass( 'solr' );
$solr = new BpmnEngine_SearchIndexAccess_Solr( $this->_solrIsEnabled, $this->_solrHost );
// create list of facets
$solr->deleteDocument( $workspace, $idQuery );
}
/**
* Commit index changes
*
* @param string $workspace
*/
public function commitIndexChanges ($workspace)
{
G::LoadClass( 'solr' );
$solr = new BpmnEngine_SearchIndexAccess_Solr( $this->_solrIsEnabled, $this->_solrHost );
// commit
$solr->commitChanges( $workspace );
}
/**
* Optimize index changes
*
* @param string $workspace
*/
public function optimizeIndexChanges ($workspace)
{
G::LoadClass( 'solr' );
$solr = new BpmnEngine_SearchIndexAccess_Solr( $this->_solrIsEnabled, $this->_solrHost );
// commit
$solr->optimizeChanges( $workspace );
}
/**
* Call Solr server to return the list of paginated pages.
*
* @param FacetRequest $solrRequestData
* @return Entity_SolrQueryResult
*/
public function getDataTablePaginatedList ($solrRequestData)
{
require_once ('class.solr.php');
require_once ('entities/SolrRequestData.php');
require_once ('entities/SolrQueryResult.php');
// prepare the list of sorted columns
// verify if the data of sorting is available
if (isset( $solrRequestData->sortCols[0] )) {
for ($i = 0; $i < $solrRequestData->numSortingCols; $i ++) {
// verify if column is sortable
if ($solrRequestData->includeCols[$solrRequestData->sortCols[$i]] != '' && $solrRequestData->sortableCols[$i] == "true") {
// change sorting column index to column names
$solrRequestData->sortCols[$i] = $solrRequestData->includeCols[$solrRequestData->sortCols[$i]];
// define the direction of the sorting columns
$solrRequestData->sortDir[$i] = $solrRequestData->sortDir[$i];
}
}
}
// remove placeholder fields
// the placeholder doesn't affect the the solr's response
// $solrRequestData->includeCols = array_diff($solrRequestData->includeCols,
// array(''));
// execute query
$solr = new BpmnEngine_SearchIndexAccess_Solr( $this->_solrIsEnabled, $this->_solrHost );
$solrPaginatedResult = $solr->executeQuery( $solrRequestData );
// get total number of documents in index
$numTotalDocs = $solr->getNumberDocuments( $solrRequestData->workspace );
// create the Datatable response of the query
$numFound = $solrPaginatedResult->response->numFound;
$docs = $solrPaginatedResult->response->docs;
// print_r($docs);
// insert list of names in docs result
$data = array ("sEcho" => '', // must be completed in response
"iTotalRecords" => intval( $numTotalDocs ), // we must get the
// total number of
// documents
"iTotalDisplayRecords" => $numFound,"aaData" => array ()
);
// copy result document or add placeholders to result
foreach ($docs as $i => $doc) {
$data['aaData'][$i] = array ();
foreach ($solrRequestData->includeCols as $columnName) {
if ($columnName == '') {
$data['aaData'][$i][] = ''; // placeholder
} else {
if (isset( $doc->$columnName )) {
$data['aaData'][$i][] = $doc->$columnName;
} else {
$data['aaData'][$i][] = '';
}
}
}
}
$solrQueryResponse = Entity_SolrQueryResult::createForRequest( $data );
//
return $solrQueryResponse;
}
/**
* Return the list of stored fields in the index.
*
* @param string $workspace
* @return array of index fields
*/
public function getIndexFields ($workspace)
{
require_once ('class.solr.php');
$solr = new BpmnEngine_SearchIndexAccess_Solr( $this->_solrIsEnabled, $this->_solrHost );
// print "SearchIndex!!!!";
// create list of facets
$solrFieldsData = $solr->getListIndexedStoredFields( $workspace );
// copy list of arrays
$listFields = array ();
foreach ($solrFieldsData->fields as $key => $fieldData) {
if (array_key_exists( 'dynamicBase', $fieldData )) {
$originalFieldName = substr( $key, 0, - strlen( $fieldData->dynamicBase ) + 1 );
// $listFields[strtolower($originalFieldName)] = $key; //in case of case insentive strings
// Maintain case sensitive variable names
$listFields[$originalFieldName] = $key;
} else {
// $listFields[strtolower($key)] = $key;
// Maintain case sensitive variable names
$listFields[$key] = $key;
}
}
return $listFields;
}
}