fixing table methods

winglang · Sep 12, 2023 · afda10c · afda10c
1 parent d634827
commit afda10c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/libs/wingsdk/src/shared-aws/table.inflight.ts b/libs/wingsdk/src/shared-aws/table.inflight.ts
@@ -27,7 +27,10 @@ export class TableClient implements ITableClient {
       const command = new PutItemCommand({
         TableName: this.tableName,
         Item: marshall(insertRow),
-        ConditionExpression: `attribute_not_exists(${this.primaryKey})`,
+        ConditionExpression: `attribute_not_exists(#primary_key)`,
+        ExpressionAttributeNames: {
+          "#primary_key": this.primaryKey,
+        },
       });
       await this.client.send(command);
     } catch (e) {

diff --git a/libs/wingsdk/src/target-tf-aws/table.ts b/libs/wingsdk/src/target-tf-aws/table.ts
@@ -56,7 +56,10 @@ export class Table extends ex.Table {
       throw new Error("tables can only be bound by tfaws.Function for now");
     }
 
-    if (ops.includes(ex.TableInflightMethods.INSERT)) {
+    if (
+      ops.includes(ex.TableInflightMethods.INSERT) ||
+      ops.includes(ex.TableInflightMethods.UPSERT)
+    ) {
       host.addPolicyStatements([
         {
           actions: ["dynamodb:PutItem"],