From 103b587ab5c3c2c90cadc5a3eae7ae28d697f5c0 Mon Sep 17 00:00:00 2001
From: Frank Tao <frank.tao3@ontario.ca>
Date: Thu, 21 Jun 2018 22:32:08 -0400
Subject: [PATCH] #1008: Fixed sql looping issue

---
 .../ca/uhn/fhir/jpa/dao/SearchBuilder.java    | 29 +++++++++++++++++++
 .../uhn/fhir/jpa/dao/data/IForcedIdDao.java   |  5 +++-
 .../uhn/fhir/jpa/entity/BaseHasResource.java  |  2 +-
 3 files changed, 34 insertions(+), 2 deletions(-)

diff --git a/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/dao/SearchBuilder.java b/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/dao/SearchBuilder.java
index f43ea0aef284..978826d5bf38 100644
--- a/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/dao/SearchBuilder.java
+++ b/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/dao/SearchBuilder.java
@@ -87,6 +87,7 @@
 import ca.uhn.fhir.jpa.dao.data.IResourceTagDao;
 import ca.uhn.fhir.jpa.entity.BaseHasResource;
 import ca.uhn.fhir.jpa.entity.BaseResourceIndexedSearchParam;
+import ca.uhn.fhir.jpa.entity.ForcedId;
 import ca.uhn.fhir.jpa.entity.ResourceHistoryTable;
 import ca.uhn.fhir.jpa.entity.ResourceIndexedSearchParamDate;
 import ca.uhn.fhir.jpa.entity.ResourceIndexedSearchParamNumber;
@@ -1688,8 +1689,19 @@ private void doLoadPids(List<IBaseResource> theResourceListToPopulate, Set<Long>
 		//-- preload all tags with tag definition if any
 		Map<Long, Collection<ResourceTag>> tagMap = getResourceTagMap(resultList);
 		
+		//-- pre-load all forcedId
+		Map<Long, ForcedId> forcedIdMap = getForcedIdMap(pids);
+				
+		ForcedId forcedId = null;
+		Long resourceId = null;
 		for (ResourceTable next : resultList) {
 			Class<? extends IBaseResource> resourceType = context.getResourceDefinition(next.getResourceType()).getImplementingClass();
+			
+			resourceId = next.getId();
+			forcedId = forcedIdMap.get(resourceId);
+			if (forcedId != null)
+				next.setForcedId(forcedId);
+			
 			IBaseResource resource = theDao.toResource(resourceType, next, historyMap.get(next.getId()), tagMap.get(next.getId()), theForHistoryOperation);
 			if (resource == null) {
 				ourLog.warn("Unable to find resource {}/{}/_history/{} in database", next.getResourceType(), next.getIdDt().getIdPart(), next.getVersion());
@@ -1775,6 +1787,23 @@ private Map<Long, Collection<ResourceTag>> getResourceTagMap(List<ResourceTable>
 		return tagMap;		
 	}
 
+	//-- load all forcedId in to the map
+	private Map<Long, ForcedId> getForcedIdMap(Collection<Long> pids) {
+
+		Map<Long, ForcedId> forceIdMap = new HashMap<Long, ForcedId>();
+
+		if (pids.size() == 0)
+			return forceIdMap;
+
+		Collection<ForcedId> forceIdList = myForcedIdDao.findByResourcePids(pids);
+
+		for (ForcedId forcedId : forceIdList) {
+
+			forceIdMap.put(forcedId.getResourcePid(), forcedId);
+		}
+
+		return forceIdMap;
+	}
 	@Override
 	public void loadResourcesByPid(Collection<Long> theIncludePids, List<IBaseResource> theResourceListToPopulate, Set<Long> theRevIncludedPids, boolean theForHistoryOperation,
 											 EntityManager entityManager, FhirContext context, IDao theDao) {
diff --git a/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/dao/data/IForcedIdDao.java b/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/dao/data/IForcedIdDao.java
index 2d24d6c7b001..9e2dc7c63e06 100644
--- a/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/dao/data/IForcedIdDao.java
+++ b/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/dao/data/IForcedIdDao.java
@@ -1,5 +1,6 @@
 package ca.uhn.fhir.jpa.dao.data;
 
+import java.util.Collection;
 import java.util.List;
 
 /*
@@ -38,5 +39,7 @@ public interface IForcedIdDao extends JpaRepository<ForcedId, Long> {
 
 	@Query("SELECT f FROM ForcedId f WHERE f.myResourcePid = :resource_pid")
 	public ForcedId findByResourcePid(@Param("resource_pid") Long theResourcePid);
-	
+
+	@Query("SELECT f FROM ForcedId f WHERE f.myResourcePid in (:pids)")
+	Collection<ForcedId> findByResourcePids(@Param("pids") Collection<Long> pids);
 }
diff --git a/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/entity/BaseHasResource.java b/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/entity/BaseHasResource.java
index 42cbf10057fc..177220e1b370 100644
--- a/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/entity/BaseHasResource.java
+++ b/hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/entity/BaseHasResource.java
@@ -42,7 +42,7 @@ public abstract class BaseHasResource {
 	@OptimisticLock(excluded = true)
 	private FhirVersionEnum myFhirVersion;
 
-	@OneToOne(optional = true, fetch = FetchType.EAGER, cascade = {}, orphanRemoval = false)
+	@OneToOne(optional = true, fetch = FetchType.LAZY, cascade = {}, orphanRemoval = false)
 	@JoinColumn(name = "FORCED_ID_PID")
 	@OptimisticLock(excluded = true)
 	private ForcedId myForcedId;