Fix customized simplifiedCustomDataRetriever injection. (#23547)

## Description Fix customized simplifiedCustomDataRetriever injection, as a follow-up of PR #23354
microsoft · Jan 15, 2025 · 655434e · 655434e
1 parent aae275e
commit 655434e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/server/historian/packages/historian-base/src/app.ts b/server/historian/packages/historian-base/src/app.ts
@@ -112,6 +112,7 @@ export function create(
 		revokedTokenChecker,
 		denyList,
 		ephemeralDocumentTTLSec,
+		simplifiedCustomDataRetriever,
 	);
 	app.use(apiRoutes.git.blobs);
 	app.use(apiRoutes.git.refs);

diff --git a/server/historian/packages/historian-base/src/runnerFactory.ts b/server/historian/packages/historian-base/src/runnerFactory.ts
@@ -206,6 +206,9 @@ export class HistorianResourcesFactory implements core.IResourcesFactory<Histori
 			tenantManager,
 			gitCache,
 		);
+		const simplifiedCustomDataRetriever =
+			customizations?.simplifiedCustomDataRetriever ??
+			new historianServices.SimplifiedCustomDataRetriever();
 
 		// Token revocation
 		const revokedTokenChecker: core.IRevokedTokenChecker | undefined =
@@ -232,6 +235,7 @@ export class HistorianResourcesFactory implements core.IResourcesFactory<Histori
 			denyList,
 			ephemeralDocumentTTLSec,
 			customizations?.readinessCheck,
+			simplifiedCustomDataRetriever,
 		);
 	}
 }
@@ -253,6 +257,7 @@ export class HistorianRunnerFactory implements core.IRunnerFactory<HistorianReso
 			resources.denyList,
 			resources.ephemeralDocumentTTLSec,
 			resources.readinessCheck,
+			resources.simplifiedCustomDataRetriever,
 		);
 	}
 }