Fixed several bugs in connections tree

Author: uglide

redis-desktop-manager/source/models/items/RedisKeyItem.cpp

@@ -1,4 +1,5 @@
 #include "RedisKeyItem.h"
+#include "RedisKeyNamespace.h"
 #include "RedisServerItem.h"
 #include "RedisServerDbItem.h"
 #include "KeyModel.h"
@@ -78,14 +79,16 @@ Command RedisKeyItem::getTypeCommand()
 
 ConnectionBridge * RedisKeyItem::getConnection()
 {
-    if (db == nullptr)
+    if (db == nullptr || db->server == nullptr)
         return nullptr;
 
     return db->server->connection;
 }
 
 void RedisKeyItem::remove()
 {
-    db->decreaseKeyCounter();
-    db->removeRow(this->row());
+    if (db != nullptr)
+        db->decreaseKeyCounter();
+
+    setEnabled(false); 
 }

redis-desktop-manager/source/models/items/RedisServerDbItem.cpp

@@ -57,22 +57,22 @@ void RedisServerDbItem::keysLoaded(const QVariant &keys, QObject *owner)
 
     rawKeys = keys.toStringList();
 
-    int resultSize = rawKeys.size();
+    keysCount = rawKeys.size();
 
-    if (resultSize == 0) {
+    if (keysCount == 0) {
         server->unlockUI();
         setNormalIcon();
         return;
     }
 
-    if (resultSize < keysCount) {
+    if (keysCount < keysCount) {
         server->error(QString("Loaded keys: %2 of %3. Error - %4 <br /> Check <a href='https://github.com/uglide/RedisDesktopManager/wiki/Known-issues'>documentation</a>")
-            .arg(resultSize)
+            .arg(keysCount)
             .arg(keysCount)
             .arg(server->connection->getLastError()));
     }
 
-
+    setDbText();
     renderKeys(rawKeys);
 
     setNormalIcon();