1 files changed, 16 insertions, 3 deletions
diff --git a/SQLiteStudio3/guiSQLiteStudio/windows/editorwindow.cpp b/SQLiteStudio3/guiSQLiteStudio/windows/editorwindow.cpp
index 7856a5e..590a2fd 100644
--- a/SQLiteStudio3/guiSQLiteStudio/windows/editorwindow.cpp
+++ b/SQLiteStudio3/guiSQLiteStudio/windows/editorwindow.cpp
@@ -17,13 +17,13 @@
 #include "parser/lexer.h"
 #include "common/utils_sql.h"
 #include "parser/parser.h"
+#include "dbobjectdialogs.h"
+#include "dialogs/exportdialog.h"
 #include <QComboBox>
 #include <QDebug>
 #include <QStringListModel>
 #include <QActionGroup>
 #include <QMessageBox>
-#include <dbobjectdialogs.h>
-#include <dialogs/exportdialog.h>
 
 CFG_KEYS_DEFINE(EditorWindow)
 EditorWindow::ResultsDisplayMode EditorWindow::resultsDisplayMode;
@@ -94,6 +94,10 @@ void EditorWindow::init()
     initActions();
     updateShortcutTips();
 
+    Db* treeSelectedDb = DBTREE->getSelectedOpenDb();
+    if (treeSelectedDb)
+        dbCombo->setCurrentIndex(dbComboModel->getIndexForDb(treeSelectedDb));
+
     Db* currentDb = getCurrentDb();
     resultsModel->setDb(currentDb);
     ui->sqlEdit->setDb(currentDb);
@@ -484,7 +488,16 @@ void EditorWindow::executionSuccessful()
 {
     double secs = ((double)resultsModel->getExecutionTime()) / 1000;
     QString time = QString::number(secs, 'f', 3);
-    notifyInfo(tr("Query finished in %2 second(s).").arg(time));
+
+    if (resultsModel->wasDataModifyingQuery())
+    {
+        QString rowsAffected = QString::number(resultsModel->getTotalRowsAffected());
+        notifyInfo(tr("Query finished in %1 second(s). Rows affected: %2").arg(time, rowsAffected));
+    }
+    else
+    {
+        notifyInfo(tr("Query finished in %1 second(s).").arg(time));
+    }
 
     lastQueryHistoryId = CFG->addSqlHistory(resultsModel->getQuery(), resultsModel->getDb()->getName(), resultsModel->getExecutionTime(), 0);