Post init provider registerer

src/main/java/ca/uhn/fhir/jpa/starter/Application.java

@@ -31,15 +31,15 @@ import org.springframework.web.servlet.DispatcherServlet;
 @ServletComponentScan(basePackageClasses = { RestfulServer.class })
 @SpringBootApplication(exclude = { ElasticsearchRestClientAutoConfiguration.class, ThymeleafAutoConfiguration.class })
 @Import({
+		StarterCrR4Config.class,
+		StarterCrDstu3Config.class,
 		SubscriptionSubmitterConfig.class,
 		SubscriptionProcessorConfig.class,
 		SubscriptionChannelConfig.class,
 		WebsocketDispatcherConfig.class,
 		MdmConfig.class,
 		JpaBatch2Config.class,
-	Batch2JobsConfig.class,
+		Batch2JobsConfig.class
-	StarterCrR4Config.class,
-	StarterCrDstu3Config.class
 })
 public class Application extends SpringBootServletInitializer {
 

src/main/java/ca/uhn/fhir/jpa/starter/common/FhirServerConfigCommon.java

@@ -14,9 +14,11 @@ import ca.uhn.fhir.jpa.subscription.match.deliver.email.IEmailSender;
 import ca.uhn.fhir.rest.server.mail.IMailSvc;
 import ca.uhn.fhir.rest.server.mail.MailConfig;
 import ca.uhn.fhir.rest.server.mail.MailSvc;
+import ca.uhn.fhir.rest.server.provider.ResourceProviderFactory;
+
 import com.google.common.base.Strings;
 import org.hl7.fhir.r4.model.Bundle.BundleType;
-
+import org.springframework.beans.factory.FactoryBean;
 import org.springframework.boot.env.YamlPropertySourceLoader;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -38,20 +40,28 @@ public class FhirServerConfigCommon {
 
 	private static final org.slf4j.Logger ourLog = org.slf4j.LoggerFactory.getLogger(FhirServerConfigCommon.class);
 
-
 	public FhirServerConfigCommon(AppProperties appProperties) {
-    ourLog.info("Server configured to " + (appProperties.getAllow_contains_searches() ? "allow" : "deny") + " contains searches");
+		ourLog.info("Server configured to " + (appProperties.getAllow_contains_searches() ? "allow" : "deny")
-    ourLog.info("Server configured to " + (appProperties.getAllow_multiple_delete() ? "allow" : "deny") + " multiple deletes");
+				+ " contains searches");
-    ourLog.info("Server configured to " + (appProperties.getAllow_external_references() ? "allow" : "deny") + " external references");
+		ourLog.info("Server configured to " + (appProperties.getAllow_multiple_delete() ? "allow" : "deny")
-    ourLog.info("Server configured to " + (appProperties.getDao_scheduling_enabled() ? "enable" : "disable") + " DAO scheduling");
+				+ " multiple deletes");
-    ourLog.info("Server configured to " + (appProperties.getDelete_expunge_enabled() ? "enable" : "disable") + " delete expunges");
+		ourLog.info("Server configured to " + (appProperties.getAllow_external_references() ? "allow" : "deny")
+				+ " external references");
+		ourLog.info("Server configured to " + (appProperties.getDao_scheduling_enabled() ? "enable" : "disable")
+				+ " DAO scheduling");
+		ourLog.info("Server configured to " + (appProperties.getDelete_expunge_enabled() ? "enable" : "disable")
+				+ " delete expunges");
 		ourLog.info("Server configured to " + (appProperties.getExpunge_enabled() ? "enable" : "disable") + " expunges");
-    ourLog.info("Server configured to " + (appProperties.getAllow_override_default_search_params() ? "allow" : "deny") + " overriding default search params");
+		ourLog.info("Server configured to " + (appProperties.getAllow_override_default_search_params() ? "allow" : "deny")
-    ourLog.info("Server configured to " + (appProperties.getAuto_create_placeholder_reference_targets() ? "allow" : "disable") + " auto-creating placeholder references");
+				+ " overriding default search params");
+		ourLog.info("Server configured to "
+				+ (appProperties.getAuto_create_placeholder_reference_targets() ? "allow" : "disable")
+				+ " auto-creating placeholder references");
 
 		if (appProperties.getSubscription().getEmail() != null) {
 			AppProperties.Subscription.Email email = appProperties.getSubscription().getEmail();
-      ourLog.info("Server is configured to enable email with host '" + email.getHost() + "' and port " + email.getPort());
+			ourLog.info(
+					"Server is configured to enable email with host '" + email.getHost() + "' and port " + email.getPort());
 			ourLog.info("Server will use '" + email.getFrom() + "' as the from email address");
 
 			if (!Strings.isNullOrEmpty(email.getUsername())) {
@@ -83,10 +93,15 @@ public class FhirServerConfigCommon {
 	public JpaStorageSettings jpaStorageSettings(AppProperties appProperties) {
 		JpaStorageSettings jpaStorageSettings = new JpaStorageSettings();
 
-    jpaStorageSettings.setIndexMissingFields(appProperties.getEnable_index_missing_fields() ? JpaStorageSettings.IndexEnabledEnum.ENABLED : JpaStorageSettings.IndexEnabledEnum.DISABLED);
+		jpaStorageSettings.setIndexMissingFields(
-    jpaStorageSettings.setAutoCreatePlaceholderReferenceTargets(appProperties.getAuto_create_placeholder_reference_targets());
+				appProperties.getEnable_index_missing_fields() ? JpaStorageSettings.IndexEnabledEnum.ENABLED
-    jpaStorageSettings.setEnforceReferentialIntegrityOnWrite(appProperties.getEnforce_referential_integrity_on_write());
+						: JpaStorageSettings.IndexEnabledEnum.DISABLED);
-    jpaStorageSettings.setEnforceReferentialIntegrityOnDelete(appProperties.getEnforce_referential_integrity_on_delete());
+		jpaStorageSettings
+				.setAutoCreatePlaceholderReferenceTargets(appProperties.getAuto_create_placeholder_reference_targets());
+		jpaStorageSettings
+				.setEnforceReferentialIntegrityOnWrite(appProperties.getEnforce_referential_integrity_on_write());
+		jpaStorageSettings
+				.setEnforceReferentialIntegrityOnDelete(appProperties.getEnforce_referential_integrity_on_delete());
 		jpaStorageSettings.setAllowContainsSearches(appProperties.getAllow_contains_searches());
 		jpaStorageSettings.setAllowMultipleDelete(appProperties.getAllow_multiple_delete());
 		jpaStorageSettings.setAllowExternalReferences(appProperties.getAllow_external_references());
@@ -98,13 +113,13 @@ public class FhirServerConfigCommon {
 
 		Integer maxFetchSize = appProperties.getMax_page_size();
 		jpaStorageSettings.setFetchSizeDefaultMaximum(maxFetchSize);
-    ourLog.info("Server configured to have a maximum fetch size of " + (maxFetchSize == Integer.MAX_VALUE ? "'unlimited'" : maxFetchSize));
+		ourLog.info("Server configured to have a maximum fetch size of "
+				+ (maxFetchSize == Integer.MAX_VALUE ? "'unlimited'" : maxFetchSize));
 
 		Long reuseCachedSearchResultsMillis = appProperties.getReuse_cached_search_results_millis();
 		jpaStorageSettings.setReuseCachedSearchResultsForMillis(reuseCachedSearchResultsMillis);
 		ourLog.info("Server configured to cache search results for {} milliseconds", reuseCachedSearchResultsMillis);
 
-
 		Long retainCachedSearchesMinutes = appProperties.getRetain_cached_searches_mins();
 		jpaStorageSettings.setExpireSearchResultsAfterMillis(retainCachedSearchesMinutes * 60 * 1000);
 
@@ -112,15 +127,18 @@ public class FhirServerConfigCommon {
 			// Subscriptions are enabled by channel type
 			if (appProperties.getSubscription().getResthook_enabled()) {
 				ourLog.info("Enabling REST-hook subscriptions");
-        jpaStorageSettings.addSupportedSubscriptionType(org.hl7.fhir.dstu2.model.Subscription.SubscriptionChannelType.RESTHOOK);
+				jpaStorageSettings
+						.addSupportedSubscriptionType(org.hl7.fhir.dstu2.model.Subscription.SubscriptionChannelType.RESTHOOK);
 			}
 			if (appProperties.getSubscription().getEmail() != null) {
 				ourLog.info("Enabling email subscriptions");
-        jpaStorageSettings.addSupportedSubscriptionType(org.hl7.fhir.dstu2.model.Subscription.SubscriptionChannelType.EMAIL);
+				jpaStorageSettings
+						.addSupportedSubscriptionType(org.hl7.fhir.dstu2.model.Subscription.SubscriptionChannelType.EMAIL);
 			}
 			if (appProperties.getSubscription().getWebsocket_enabled()) {
 				ourLog.info("Enabling websocket subscriptions");
-        jpaStorageSettings.addSupportedSubscriptionType(org.hl7.fhir.dstu2.model.Subscription.SubscriptionChannelType.WEBSOCKET);
+				jpaStorageSettings.addSupportedSubscriptionType(
+						org.hl7.fhir.dstu2.model.Subscription.SubscriptionChannelType.WEBSOCKET);
 			}
 		}
 
@@ -140,14 +158,13 @@ public class FhirServerConfigCommon {
 		jpaStorageSettings.setNormalizedQuantitySearchLevel(appProperties.getNormalized_quantity_search_level());
 		jpaStorageSettings.setIndexOnContainedResources(appProperties.getEnable_index_contained_resource());
 
-
-
 		if (appProperties.getAllowed_bundle_types() != null) {
-      jpaStorageSettings.setBundleTypesAllowedForStorage(appProperties.getAllowed_bundle_types().stream().map(BundleType::toCode).collect(Collectors.toSet()));
+			jpaStorageSettings.setBundleTypesAllowedForStorage(
+					appProperties.getAllowed_bundle_types().stream().map(BundleType::toCode).collect(Collectors.toSet()));
 		}
 
-	  jpaStorageSettings.setDeferIndexingForCodesystemsOfSize(appProperties.getDefer_indexing_for_codesystems_of_size());
+		jpaStorageSettings
-
+				.setDeferIndexingForCodesystemsOfSize(appProperties.getDefer_indexing_for_codesystems_of_size());
 
 		if (appProperties.getClient_id_strategy() == JpaStorageSettings.ClientIdStrategyEnum.ANY) {
 			jpaStorageSettings.setResourceServerIdStrategy(JpaStorageSettings.IdStrategyEnum.UUID);
@@ -173,7 +190,8 @@ public class FhirServerConfigCommon {
 		// Partitioning
 		if (appProperties.getPartitioning() != null) {
 			retVal.setPartitioningEnabled(true);
-      retVal.setIncludePartitionInSearchHashes(appProperties.getPartitioning().getPartitioning_include_in_search_hashes());
+			retVal.setIncludePartitionInSearchHashes(
+					appProperties.getPartitioning().getPartitioning_include_in_search_hashes());
 			if (appProperties.getPartitioning().getAllow_references_across_partitions()) {
 				retVal.setAllowReferencesAcrossPartitions(CrossPartitionReferenceMode.ALLOWED_UNQUALIFIED);
 			} else {
@@ -184,14 +202,13 @@ public class FhirServerConfigCommon {
 		return retVal;
 	}
 
-
 	@Primary
 	@Bean
-  public HibernatePropertiesProvider jpaStarterDialectProvider(LocalContainerEntityManagerFactoryBean myEntityManagerFactory) {
+	public HibernatePropertiesProvider jpaStarterDialectProvider(
+			LocalContainerEntityManagerFactoryBean myEntityManagerFactory) {
 		return new JpaHibernatePropertiesProvider(myEntityManagerFactory);
 	}
 
-
 	protected StorageSettings storageSettings(AppProperties appProperties, JpaStorageSettings jpaStorageSettings) {
 		jpaStorageSettings.setAllowContainsSearches(appProperties.getAllow_contains_searches());
 		jpaStorageSettings.setAllowExternalReferences(appProperties.getAllow_external_references());

src/main/java/ca/uhn/fhir/jpa/starter/cr/CrOperationProviderLoader.java

@@ -15,11 +15,14 @@ public class CrOperationProviderLoader {
 	private final ResourceProviderFactory myResourceProviderFactory;
 	private final CrOperationProviderFactory myCrProviderFactory;
 
+	private final PostInitProviderRegisterer myPostInitProviderRegister;
+
 	public CrOperationProviderLoader(FhirContext theFhirContext, ResourceProviderFactory theResourceProviderFactory,
-			CrOperationProviderFactory theCrProviderFactory) {
+			CrOperationProviderFactory theCrProviderFactory, PostInitProviderRegisterer thePostInitProviderRegister) {
 		myFhirContext = theFhirContext;
 		myResourceProviderFactory = theResourceProviderFactory;
 		myCrProviderFactory = theCrProviderFactory;
+		myPostInitProviderRegister = thePostInitProviderRegister;
 		loadProvider();
 	}
 

src/main/java/ca/uhn/fhir/jpa/starter/cr/PostInitProviderRegisterer.java

@@ -0,0 +1,51 @@
+package ca.uhn.fhir.jpa.starter.cr;
+
+import java.util.function.Supplier;
+
+import ca.uhn.fhir.rest.server.RestfulServer;
+import ca.uhn.fhir.rest.server.provider.IResourceProviderFactoryObserver;
+import ca.uhn.fhir.rest.server.provider.ResourceProviderFactory;
+
+public class PostInitProviderRegisterer {
+	public PostInitProviderRegisterer(RestfulServer restfulServer,
+			ResourceProviderFactory resourceProviderFactory) {
+		resourceProviderFactory.attach(new Observer(restfulServer));
+	}
+
+	private class Observer implements IResourceProviderFactoryObserver {
+		private RestfulServer restfulServer;
+
+		public Observer(RestfulServer restfulServer) {
+			this.restfulServer = restfulServer;
+		}
+
+		public void update(Supplier<Object> theSupplier) {
+			if (theSupplier == null) {
+				return;
+			}
+
+			var provider = theSupplier.get();
+			if (provider == null) {
+				return;
+			}
+
+			this.restfulServer.registerProvider(provider);
+
+		}
+
+		public void remove(Supplier<Object> theSupplier) {
+			if (theSupplier == null) {
+				return;
+			}
+
+			var provider = theSupplier.get();
+			if (provider == null) {
+				return;
+			}
+
+			this.restfulServer.unregisterProvider(provider);
+		}
+
+	}
+
+}

src/main/java/ca/uhn/fhir/jpa/starter/cr/StarterCrDstu3Config.java

@@ -30,6 +30,13 @@ import org.springframework.context.annotation.Import;
 @Conditional({ OnDSTU3Condition.class, CrConfigCondition.class })
 @Import({ CrDstu3Config.class })
 public class StarterCrDstu3Config {
+
+	@Bean
+	public PostInitProviderRegisterer postInitProviderRegisterer(RestfulServer theRestfulServer,
+			ResourceProviderFactory theResourceProviderFactory) {
+		return new PostInitProviderRegisterer(theRestfulServer, theResourceProviderFactory);
+	}
+
 	@Bean
 	public CrOperationProviderFactory crOperationProviderFactory() {
 		return new CrOperationProviderFactory();
@@ -38,8 +45,10 @@ public class StarterCrDstu3Config {
 	@Bean
 	public CrOperationProviderLoader crOperationProviderLoader(FhirContext theFhirContext,
 			ResourceProviderFactory theResourceProviderFactory,
-			CrOperationProviderFactory theCrlProviderFactory) {
+			CrOperationProviderFactory theCrOperationProviderFactory,
-		return new CrOperationProviderLoader(theFhirContext, theResourceProviderFactory, theCrlProviderFactory);
+			PostInitProviderRegisterer thePostInitProviderRegister) {
+		return new CrOperationProviderLoader(theFhirContext, theResourceProviderFactory, theCrOperationProviderFactory,
+				thePostInitProviderRegister);
 	}
 
 	@Bean

src/main/java/ca/uhn/fhir/jpa/starter/cr/StarterCrR4Config.java

@@ -1,5 +1,6 @@
 package ca.uhn.fhir.jpa.starter.cr;
 
+import ca.uhn.fhir.IHapiBootOrder;
 import ca.uhn.fhir.context.FhirContext;
 import ca.uhn.fhir.cr.common.IRepositoryFactory;
 import ca.uhn.fhir.cr.config.CrR4Config;
@@ -30,6 +31,13 @@ import org.springframework.context.annotation.Import;
 @Conditional({ OnR4Condition.class, CrConfigCondition.class })
 @Import({ CrR4Config.class })
 public class StarterCrR4Config {
+
+	@Bean
+	public PostInitProviderRegisterer postInitProviderRegisterer(RestfulServer theRestfulServer,
+			ResourceProviderFactory theResourceProviderFactory) {
+		return new PostInitProviderRegisterer(theRestfulServer, theResourceProviderFactory);
+	}
+
 	@Bean
 	public CrOperationProviderFactory crOperationProviderFactory() {
 		return new CrOperationProviderFactory();
@@ -38,8 +46,10 @@ public class StarterCrR4Config {
 	@Bean
 	public CrOperationProviderLoader crOperationProviderLoader(FhirContext theFhirContext,
 			ResourceProviderFactory theResourceProviderFactory,
-			CrOperationProviderFactory theCrlProviderFactory) {
+			CrOperationProviderFactory theCrOperationProviderFactory,
-		return new CrOperationProviderLoader(theFhirContext, theResourceProviderFactory, theCrlProviderFactory);
+			PostInitProviderRegisterer thePostInitProviderRegister) {
+		return new CrOperationProviderLoader(theFhirContext, theResourceProviderFactory, theCrOperationProviderFactory,
+				thePostInitProviderRegister);
 	}
 
 	@Bean