From 1141d5be7a4ff3a2095eb5deca22ab02a10badfa Mon Sep 17 00:00:00 2001
From: Jonathan Stevens <jonathanchris@gmail.com>
Date: Tue, 9 Apr 2024 10:50:20 -0400
Subject: [PATCH] Update SolrDocument.php

make some adjustments to properly get the original $id for the documents array
---
 src/Plugin/search_api/datasource/SolrDocument.php | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/Plugin/search_api/datasource/SolrDocument.php b/src/Plugin/search_api/datasource/SolrDocument.php
index f1aca822..33993fd8 100644
--- a/src/Plugin/search_api/datasource/SolrDocument.php
+++ b/src/Plugin/search_api/datasource/SolrDocument.php
@@ -192,8 +192,9 @@ public function loadMultiple(array $ids) {
         ->addCondition('search_api_id', $ids, 'IN')
         ->execute()
         ->getResultItems();
-      foreach ($results as $result) {
-        $documents[$id] = $this->getSolrDocumentFactory()->create($result);
+      foreach ($results as $id => $result) {
+        $original_id =  str_replace('solr_document/', '', $id);
+        $documents[$original_id] = $this->getSolrDocumentFactory()->create($result);
       }
     }
     catch (SearchApiException $e) {