fix: proxy recover flag to client implementation

finkef · Apr 11, 2023 · abb6c2e · abb6c2e
1 parent 788ab33
commit abb6c2e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 6 deletions.
diff --git a/.changeset/four-phones-join.md b/.changeset/four-phones-join.md
@@ -0,0 +1,5 @@
+---
+"@model-ts/dynamodb": patch
+---
+
+fix: proxy recover flag to client implementation
diff --git a/packages/dynamodb/src/provider.ts b/packages/dynamodb/src/provider.ts
@@ -417,22 +417,24 @@ export const getProvider = (client: Client) => {
 
       load<
         M extends DynamoDBModelConstructor<any>,
-        Null extends boolean = false
+        Null extends boolean = false,
+        Recover extends boolean = false
       >(
         this: M,
         key: Key,
         params?: Omit<GetOperation<M>, "_model" | "_operation" | "key"> & {
           null?: Null
+          recover?: Recover
         }
       ) {
-        return client.load<M, Null>(
+        return client.load<M, Null, Recover>(
           {
             _model: this,
             _operation: "get",
             key,
             ...params,
           },
-          { null: params?.null }
+          { null: params?.null, recover: params?.recover }
         )
       },
 
@@ -629,21 +631,26 @@ export const getProvider = (client: Client) => {
         })
       },
 
-      load<M extends DynamoDBUnion, Null extends boolean = false>(
+      load<
+        M extends DynamoDBUnion,
+        Null extends boolean = false,
+        Recover extends boolean = false
+      >(
         this: M,
         key: Key,
         params?: Omit<GetOperation<M>, "_model" | "_operation" | "key"> & {
           null?: Null
+          recover?: Recover
         }
       ) {
-        return client.load<M, Null>(
+        return client.load<M, Null, Recover>(
           {
             _model: this,
             _operation: "get",
             key,
             ...params,
           },
-          { null: params?.null }
+          { null: params?.null, recover: params?.recover }
         )
       },