Added entityType field

Stinger-Soft · Apr 11, 2017 · ed6e4b5 · ed6e4b5
1 parent 35848d2
commit ed6e4b5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/Model/Document.php b/Model/Document.php
@@ -14,14 +14,14 @@
 use StingerSoft\EntitySearchBundle\Model\DocumentAdapter;
 use Solarium\QueryType\Select\Result\DocumentInterface;
 
-class Document extends DocumentAdapter{
-
-
+class Document extends DocumentAdapter {
+
 	public static function createFromSolariumResult(DocumentInterface $solrDocument) {
 		$document = new Document();
 		foreach($solrDocument->getFields() as $key => $value) {
 			$document->addField($key, $value);
 		}
+		$document->setEntityType($document->getFieldValue('entityType'));
 		$document->setEntityClass($document->getFieldValue('clazz'));
 		$document->setEntityId($document->getFieldValue('internalId'));
 		return $document;

diff --git a/Services/SearchService.php b/Services/SearchService.php
@@ -107,6 +107,7 @@ public function saveDocument(\StingerSoft\EntitySearchBundle\Model\Document $doc
 		$doc->id = $this->createIdFromDocument($document);
 		$doc->internalId = json_encode($document->getEntityId());
 		$doc->clazz = $document->getEntityClass();
+		$doc->entityType = $document->getEntityType();
 
 		foreach($document->getFields() as $key => $value) {
 			$doc->setField($key, $value);
@@ -260,10 +261,10 @@ public function getIndexSize() {
 	}
 
 	protected function escapeFacetKey($facetKey) {
-		return $facetKey == Document::FIELD_TYPE ? 'clazz' : $facetKey;
+		return $facetKey == Document::FIELD_TYPE ? 'entityType' : $facetKey;
 	}
 
 	protected function unescapeFacetKey($facetKey) {
-		return $facetKey == 'clazz' ? Document::FIELD_TYPE : $facetKey;
+		return $facetKey == 'entityType' ? Document::FIELD_TYPE : $facetKey;
 	}
 }