From 5eda39cb62526922fdba46b3ca7f1ef9fe526828 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Thu, 8 Sep 2022 11:04:43 +0200
Subject: [PATCH] rows affected info for postgresql

---
 plugins/dbgate-plugin-postgres/src/backend/drivers.js | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/plugins/dbgate-plugin-postgres/src/backend/drivers.js b/plugins/dbgate-plugin-postgres/src/backend/drivers.js
index 6aaa7aa6b..03a9f8de5 100644
--- a/plugins/dbgate-plugin-postgres/src/backend/drivers.js
+++ b/plugins/dbgate-plugin-postgres/src/backend/drivers.js
@@ -109,6 +109,16 @@ const drivers = driverBases.map(driverBase => ({
     });
 
     query.on('end', () => {
+      const { command, rowCount } = query._result || {};
+
+      if (command != 'SELECT' && _.isNumber(rowCount)) {
+        options.info({
+          message: `${rowCount} rows affected`,
+          time: new Date(),
+          severity: 'info',
+        });
+      }
+
       if (!wasHeader) {
         columns = extractPostgresColumns(query._result);
         if (columns && columns.length > 0) {