From 655434ed69d07e997159772e28a6517e0d523de7 Mon Sep 17 00:00:00 2001
From: Jackie Zhang <zhajie@microsoft.com>
Date: Tue, 14 Jan 2025 22:23:39 -0800
Subject: [PATCH] Fix customized simplifiedCustomDataRetriever injection.
 (#23547)

## Description

Fix customized simplifiedCustomDataRetriever injection, as a follow-up
of PR https://github.com/microsoft/FluidFramework/pull/23354
---
 server/historian/packages/historian-base/src/app.ts          | 1 +
 .../historian/packages/historian-base/src/runnerFactory.ts   | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/server/historian/packages/historian-base/src/app.ts b/server/historian/packages/historian-base/src/app.ts
index 7dc061148598..b86b2db8dc6f 100644
--- 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
index 7be227c68ddd..eb6425e80b05 100644
--- 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,
 		);
 	}
 }