Allow lambda instance to provide an activity timeout time (#16409)

microsoft · Jul 20, 2023 · 8f6c793 · 8f6c793
1 parent c2b1b4e
commit 8f6c793
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/server/routerlicious/packages/lambdas-driver/src/document-router/documentPartition.ts b/server/routerlicious/packages/lambdas-driver/src/document-router/documentPartition.ts
@@ -164,6 +164,7 @@ export class DocumentPartition {
 	}
 
 	private updateActivityTime() {
-		this.activityTimeoutTime = Date.now() + this.activityTimeout;
+		this.activityTimeoutTime =
+			Date.now() + (this.lambda?.activityTimeout ?? this.activityTimeout);
 	}
 }
diff --git a/server/routerlicious/packages/services-core/src/lambdas.ts b/server/routerlicious/packages/services-core/src/lambdas.ts
@@ -74,6 +74,12 @@ export interface IContext {
 }
 
 export interface IPartitionLambda {
+	/**
+	 * Expire document partition after this long of no activity.
+	 * When undefined, the default global IDocumentLambdaServerConfiguration.partitionActivityTimeout is used.
+	 */
+	readonly activityTimeout?: number;
+
 	/**
 	 * Processes an incoming message
 	 */