Merge pull request #280 from rbhuva/main

fix: return affected row on update ql
sapmentors · Mar 19, 2022 · 90956ac · 90956ac
2 parents bbdef88 + e423555
commit 90956ac
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/__tests__/lib/pg/ql.test.js b/__tests__/lib/pg/ql.test.js
@@ -182,4 +182,17 @@ describe('QL to PostgreSQL', () => {
  expect(beers[0].modifiedAt).toMatch(timestampRegex)
  })
  })
+
+ describe('UPDATE', () => {
+ beforeEach(async () => {
+ await deploy(this._model, {}).to(this._dbProperties)
+ })
+ test('-> Get affected rows ', async () => {
+ const { Beers } = cds.entities('csw')
+ const affectedRows = await cds.run(
+ UPDATE(Beers).set({ name: 'TEST' }).where({ ID: '9e1704e3-6fd0-4a5d-bfb1-13ac47f7976b' })
+ )
+ expect(affectedRows).toStrictEqual(1)
+ })
+ })
 })
diff --git a/lib/pg/execute.js b/lib/pg/execute.js
@@ -212,7 +212,9 @@ async function _executeSQLReturningRows(dbc, sql, values, isOne, postMapper, pro
 
  let rawResult = await dbc.query(sql, values)
  let result
- if (isOne && rawResult.rows.length > 0) {
+ if (rawResult.command === 'UPDATE') {
+ result = rawResult.rowCount
+ } else if (isOne && rawResult.rows.length > 0) {
  result = rawResult.rows[0]
  } else if (isOne && rawResult.rows.length === 0) {
  result = null