DocumentDb: Allow fine-grained customization of underlying client.

rfrappier · Sep 19, 2018 · c65190e · c65190e
1 parent d92843b
commit c65190e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 - SSMParameterStoreConfigService: Added set method.
 - DocumentDb: Added debug mode
 - DocumentDb: Exposed underlying client
+- DocumentDb: Allow fine-grained customization of underlying client
 
 ### Changed
 - [BREAKING] Removed maxresults in Azure KeyVaultConfigService options as it is not very useful (blocked at 25 anyway)

diff --git a/packages/uno-serverless-azure/src/services/documentdb.ts b/packages/uno-serverless-azure/src/services/documentdb.ts
@@ -93,8 +93,9 @@ export interface DocumentDbImplOptions {
   collectionCreationOptions?: Partial<Collection>;
   collectionId: string | Promise<string>;
   databaseId: string | Promise<string>;
-  debug?: boolean | Promise<boolean>;
+  debug?: boolean | Promise<boolean | undefined>;
   defaultConsistencyLevel?: ConsistencyLevel;
+  documentClient?: DocumentClient | Promise<DocumentClient | undefined>;
   endpoint: string | Promise<string>;
   primaryKey: string | Promise<string>;
 }
@@ -105,6 +106,11 @@ export class DocumentDbImpl implements DocumentDb, CheckHealth {
 
   private readonly lazyClient = lazyAsync(
     async () => {
+      const optionClient = await this.options.documentClient;
+      if (optionClient) {
+        return optionClient;
+      }
+
       return new DocumentClient(
         await this.options.endpoint,
         {