Add operation to allow users to be added to group

WARNING! Lazy implementation everywhere, does not handle out-of-order
packets

1 files changed, 40 insertions, 3 deletions

diff --git a/src/DatabaseStorage.cpp b/src/DatabaseStorage.cpp
index 638eac0..7c86d18 100644
--- a/src/DatabaseStorage.cpp
+++ b/src/DatabaseStorage.cpp
@@ -1,6 +1,6 @@
 #include "../include/DatabaseStorage.hpp"
-#include "../include/Group.hpp"
 #include "../include/User.hpp"
+#include "../include/Group.hpp"
 #include <cstring>
 #include <chrono>
 
@@ -8,6 +8,8 @@ using namespace std;
 
 namespace odhtdb
 {
+    const u64 QUARANTINE_STORAGE_TIME_MICROSECONDS = 60 * 1.0e6;
+    
     DatabaseStorageObject::DatabaseStorageObject(DataView &_data, u64 _timestamp, const Signature::PublicKey &_creatorPublicKey) : 
         data(_data), createdTimestamp(_timestamp), creatorPublicKey(_creatorPublicKey)
     {
@@ -37,11 +39,11 @@ namespace odhtdb
         databaseStorageObjectList->data = DataView(new u8[dataSize], dataSize);
         memcpy(databaseStorageObjectList->data.data, data, dataSize);
         storageMap[hash] = databaseStorageObjectList;
+        groupByIdMap[creatorGroup->getId()] = creatorGroup;
         
         for(auto user : creatorGroup->getUsers())
         {
-            userPublicKeyNodeMap[user->getPublicKey()] = hash;
-            publicKeyUserMap[user->getPublicKey()] = user;
+            addUser((User*)user, hash);
         }
     }
     
@@ -69,6 +71,12 @@ namespace odhtdb
         quarantineStorageMap[creatorPublicKey].emplace_back(databaseQuarantineStorageObject);
     }
     
+    void DatabaseStorage::addUser(User *user, const Hash &hash)
+    {
+        userPublicKeyNodeMap[user->getPublicKey()] = hash;
+        publicKeyUserMap[user->getPublicKey()] = user;
+    }
+    
     const DatabaseStorageObjectList* DatabaseStorage::getStorage(const Hash &hash) const
     {
         auto it = storageMap.find(hash);
@@ -93,4 +101,33 @@ namespace odhtdb
             return it->second;
         return nullptr;
     }
+    
+    Group* DatabaseStorage::getGroupById(uint8_t groupId[16])
+    {
+        auto it = groupByIdMap.find(DataView(groupId, 16));
+        if(it != groupByIdMap.end())
+            return it->second;
+        return nullptr;
+    }
+    
+    void DatabaseStorage::update()
+    {
+        auto time = chrono::high_resolution_clock::now().time_since_epoch();
+        auto timeMicroseconds = chrono::duration_cast<chrono::microseconds>(time).count();
+        for(auto mapIt = quarantineStorageMap.begin(); mapIt != quarantineStorageMap.end(); )
+        {
+            for(auto vecIt = mapIt->second.begin(); vecIt != mapIt->second.end(); )
+            {
+                if(timeMicroseconds - (*vecIt)->storedTimestamp > QUARANTINE_STORAGE_TIME_MICROSECONDS)
+                    vecIt = mapIt->second.erase(vecIt);
+                else
+                    ++vecIt;
+            }
+            
+            if(mapIt->second.empty())
+                mapIt = quarantineStorageMap.erase(mapIt);
+            else
+                ++mapIt;
+        }
+    }
 }