pqb 0.2.6 → 0.2.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pqb",
-  "version": "0.2.6",
+  "version": "0.2.7",
   "description": "Postgres query builder",
   "homepage": "https://porm.netlify.app/guide/query-builder.html",
   "repository": {

package/src/queryMethods/update.test.ts

@@ -413,4 +413,29 @@ describe('update', () => {
       expect(eq).toBe(true);
     });
   });
+
+  describe('chaining', () => {
+    it('should handle multiple updates with increment and decrement', () => {
+      const query = User.select('id')
+        .find(1)
+        .update({ name: 'name' })
+        .increment('id')
+        .update({ password: 'password' })
+        .decrement('age');
+
+      expectSql(
+        query.toSql(),
+        `
+          UPDATE "user"
+          SET "name" = $1,
+              "id" = "id" + $2,
+              "password" = $3,
+              "age" = "age" - $4
+          WHERE "user"."id" = $5
+          RETURNING "user"."id"
+        `,
+        ['name', 1, 'password', 1, 1],
+      );
+    });
+  });
 });

package/src/sql/update.ts

@@ -21,22 +21,20 @@ export const pushUpdateSql = (
 
   ctx.sql.push('SET');
 
+  const set: string[] = [];
   query.data.forEach((item) => {
     if (isRaw(item)) {
-      ctx.sql.push(getRaw(item, ctx.values));
+      set.push(getRaw(item, ctx.values));
     } else {
-      const set: string[] = [];
-
       for (const key in item) {
         const value = item[key];
         if (value !== undefined) {
           set.push(`${q(key)} = ${processValue(ctx.values, key, value)}`);
         }
       }
-
-      ctx.sql.push(set.join(', '));
     }
   });
+  ctx.sql.push(set.join(', '));
 
   pushWhereStatementSql(ctx, model, query, quotedAs);
   pushReturningSql(ctx, model, query, quotedAs);