feat(update): add native document client options support ✨

neoxia · Sep 26, 2019 · 6d7c36f · 6d7c36f
1 parent 08544fb
commit 6d7c36f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/base-model.ts b/src/base-model.ts
@@ -399,18 +399,21 @@ export default abstract class Model<T> {
   public async update(
     pk: Key,
     actions: IUpdateActions,
+    options?: Partial<DocumentClient.UpdateItemInput>,
   ): Promise<PromiseResult<DocumentClient.UpdateItemOutput, AWSError>>;
 
   public async update(
     pk: Key,
     sk: Key,
     actions: IUpdateActions,
+    options?: Partial<DocumentClient.UpdateItemInput>,
   ): Promise<PromiseResult<DocumentClient.UpdateItemOutput, AWSError>>;
 
   public async update(
     pk: Key,
     sk_actions: Key | IUpdateActions,
     actions?: IUpdateActions,
+    options?: Partial<DocumentClient.UpdateItemInput>,
   ): Promise<PromiseResult<DocumentClient.UpdateItemOutput, AWSError>> {
     // Handle overloading
     const sk: Key = isKey(sk_actions) ? sk_actions : null;
@@ -422,7 +425,9 @@ export default abstract class Model<T> {
       Key: this.buildKeys(pk, sk),
       AttributeUpdates: buildUpdateActions(updateActions),
     };
-
+    if (options) {
+      Object.assign(params, options);
+    }
     return this.documentClient.update(params).promise();
   }