Add setType method to arSolrQuery

Added a setType method to arSolrQuery which appends the provided type to each of the search fields.
artefactual · Jun 18, 2024 · ca14a5b · ca14a5b
1 parent f78050c
commit ca14a5b
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/lib/task/search/arSolrSearchTask.class.php b/lib/task/search/arSolrSearchTask.class.php
@@ -94,7 +94,7 @@ private function runSolrQuery($solrInstance, $queryText, $rows, $start, $fields)
             $query->setFields(arSolrPluginUtil::getBoostedSearchFields($newFields));
         }
 
-        $docs = $solrInstance->search($query);
+        $docs = $solrInstance->search($query, 'QubitInformationObject');
         if ($docs) {
             foreach ($docs as $resp) {
                 $this->log(sprintf('%s - %s', $resp->id, $resp->{'i18n.en.title'}[0]));

diff --git a/plugins/arSolrPlugin/lib/arSolrPlugin.class.php b/plugins/arSolrPlugin/lib/arSolrPlugin.class.php
@@ -276,9 +276,10 @@ public function getSolrCollection()
         return $this->solrClientOptions['collection'];
     }
 
-    public function search($query)
+    public function search($query, $type)
     {
         $url = $this->getSolrUrl().'/solr/'.$this->getSolrCollection().'/query';
+        $query->setType($type);
         $response = arSolrPlugin::makeHttpRequest($url, 'POST', json_encode($query->getQueryParams()));
 
         return $response->response->docs;

diff --git a/plugins/arSolrPlugin/lib/query/arSolrQuery.class.php b/plugins/arSolrPlugin/lib/query/arSolrQuery.class.php
@@ -166,4 +166,13 @@ public function addAggregation($agg): self
 
         return $this;
     }
+
+    public function setType($type)
+    {
+        $newFieldsArr = [];
+        foreach ($this->fields as $field) {
+            array_push($newFieldsArr, "{$type}.{$field}");
+        }
+        $this->setFields($newFieldsArr);
+    }
 }