Simplify storage context mgmt.
[shibboleth/cpp-xmltooling.git] / xmltooling / impl / MemoryStorageService.cpp
index 3a041ea..0179897 100644 (file)
@@ -40,26 +40,28 @@ namespace xmltooling {
         MemoryStorageService(const DOMElement* e);\r
         virtual ~MemoryStorageService();\r
         \r
-        void createString(const char* key, const char* value, time_t expiration);\r
-        bool readString(const char* key, string& value, time_t modifiedSince=0);\r
-        bool updateString(const char* key, const char* value=NULL, time_t expiration=0);\r
-        bool deleteString(const char* key);\r
+        void createString(const char* context, const char* key, const char* value, time_t expiration);\r
+        bool readString(const char* context, const char* key, string& value, time_t modifiedSince=0);\r
+        bool updateString(const char* context, const char* key, const char* value=NULL, time_t expiration=0);\r
+        bool deleteString(const char* context, const char* key);\r
         \r
-        void createText(const char* key, const char* value, time_t expiration) {\r
-            return createString(key, value, expiration);\r
+        void createText(const char* context, const char* key, const char* value, time_t expiration) {\r
+            return createString(context, key, value, expiration);\r
         }\r
-        bool readText(const char* key, string& value, time_t modifiedSince=0) {\r
-            return readString(key, value, modifiedSince);\r
+        bool readText(const char* context, const char* key, string& value, time_t modifiedSince=0) {\r
+            return readString(context, key, value, modifiedSince);\r
         }\r
-        bool updateText(const char* key, const char* value=NULL, time_t expiration=0) {\r
-            return updateString(key, value, expiration);\r
+        bool updateText(const char* context, const char* key, const char* value=NULL, time_t expiration=0) {\r
+            return updateString(context, key, value, expiration);\r
         }\r
-        bool deleteText(const char* key) {\r
-            return deleteString(key);\r
+        bool deleteText(const char* context, const char* key) {\r
+            return deleteString(context, key);\r
         }\r
         \r
-        void reap() {\r
-            shutdown_wait->signal();\r
+        void reap(const char* context);\r
+        void deleteContext(const char* context) {\r
+            Lock wrapper(contextLock);\r
+            m_contextMap.erase(context);\r
         }\r
 \r
     private:\r
@@ -72,9 +74,27 @@ namespace xmltooling {
             time_t modified, expiration;\r
         };\r
         \r
-        map<string,Record> m_dataMap;\r
-        multimap<time_t,string> m_expMap;\r
-        RWLock* m_lock;\r
+        struct XMLTOOL_DLLLOCAL Context {\r
+            Context() : m_lock(RWLock::create()) {}\r
+            Context(const Context& src) {\r
+                m_dataMap = src.m_dataMap;\r
+                m_expMap = src.m_expMap;\r
+                m_lock = RWLock::create();\r
+            }\r
+            ~Context() { delete m_lock; }\r
+            map<string,Record> m_dataMap;\r
+            multimap<time_t,string> m_expMap;\r
+            RWLock* m_lock;\r
+            unsigned long reap();\r
+        };\r
+\r
+        Context& getContext(const char* context) {\r
+            Lock wrapper(contextLock);\r
+            return m_contextMap[context];\r
+        }\r
+\r
+        map<string,Context> m_contextMap;\r
+        Mutex* contextLock;\r
         CondWait* shutdown_wait;\r
         Thread* cleanup_thread;\r
         static void* cleanup_fn(void*);\r
@@ -93,10 +113,10 @@ namespace xmltooling {
 static const XMLCh cleanupInterval[] = UNICODE_LITERAL_15(c,l,e,a,n,u,p,I,n,t,e,r,v,a,l);\r
 \r
 MemoryStorageService::MemoryStorageService(const DOMElement* e)\r
-    : m_lock(NULL), shutdown_wait(NULL), cleanup_thread(NULL), shutdown(false), m_cleanupInterval(0),\r
+    : contextLock(NULL), shutdown_wait(NULL), cleanup_thread(NULL), shutdown(false), m_cleanupInterval(0),\r
         m_log(Category::getInstance(XMLTOOLING_LOGCAT".StorageService"))\r
 {\r
-    m_lock = RWLock::create();\r
+    contextLock = Mutex::create();\r
     shutdown_wait = CondWait::create();\r
     cleanup_thread = Thread::create(&cleanup_fn, (void*)this);\r
 \r
@@ -115,8 +135,8 @@ MemoryStorageService::~MemoryStorageService()
     shutdown_wait->signal();\r
     cleanup_thread->join(NULL);\r
 \r
-    delete m_lock;\r
     delete shutdown_wait;\r
+    delete contextLock;\r
 }\r
 \r
 void* MemoryStorageService::cleanup_fn(void* cache_p)\r
@@ -149,20 +169,11 @@ void MemoryStorageService::cleanup()
         shutdown_wait->timedwait(mutex, m_cleanupInterval);\r
         if (shutdown)\r
             break;\r
-\r
-        // Lock the "database".\r
-        m_lock->wrlock();\r
-        \r
-        // Garbage collect any expired entries.\r
-        unsigned int count=0;\r
-        time_t now=time(NULL)-XMLToolingConfig::getConfig().clock_skew_secs;\r
-        multimap<time_t,string>::iterator stop=m_expMap.upper_bound(now);\r
-        for (multimap<time_t,string>::iterator i=m_expMap.begin(); i!=stop; m_expMap.erase(i++)) {\r
-            m_dataMap.erase(i->second);\r
-            ++count;\r
-        }\r
         \r
-        m_lock->unlock();\r
+        unsigned long count=0;\r
+        Lock wrapper(contextLock);\r
+        for (map<string,Context>::iterator i=m_contextMap.begin(); i!=m_contextMap.end(); ++i)\r
+            count += i->second.reap();\r
         \r
         if (count)\r
             m_log.info("purged %d record(s) from storage", count);\r
@@ -175,28 +186,55 @@ void MemoryStorageService::cleanup()
     Thread::exit(NULL);\r
 }\r
 \r
-void MemoryStorageService::createString(const char* key, const char* value, time_t expiration)\r
+void MemoryStorageService::reap(const char* context)\r
 {\r
-    // Lock the maps.\r
+    getContext(context).reap();\r
+}\r
+\r
+unsigned long MemoryStorageService::Context::reap()\r
+{\r
+    // Lock the "database".\r
     m_lock->wrlock();\r
     SharedLock wrapper(m_lock, false);\r
     \r
+    // Garbage collect any expired entries.\r
+    unsigned long count=0;\r
+    time_t now=time(NULL)-XMLToolingConfig::getConfig().clock_skew_secs;\r
+    multimap<time_t,string>::iterator stop=m_expMap.upper_bound(now);\r
+    for (multimap<time_t,string>::iterator i=m_expMap.begin(); i!=stop; m_expMap.erase(i++)) {\r
+        m_dataMap.erase(i->second);\r
+        ++count;\r
+    }\r
+\r
+    return count;\r
+}\r
+\r
+void MemoryStorageService::createString(const char* context, const char* key, const char* value, time_t expiration)\r
+{\r
+    Context& ctx = getContext(context);\r
+\r
+    // Lock the maps.\r
+    ctx.m_lock->wrlock();\r
+    SharedLock wrapper(ctx.m_lock, false);\r
+    \r
     // Check for a duplicate.\r
-    map<string,Record>::iterator i=m_dataMap.find(key);\r
-    if (i!=m_dataMap.end())\r
+    map<string,Record>::iterator i=ctx.m_dataMap.find(key);\r
+    if (i!=ctx.m_dataMap.end())\r
         throw IOException("attempted to insert a record with duplicate key ($1)", params(1,key));\r
     \r
-    m_dataMap[key]=Record(value,time(NULL),expiration);\r
-    m_expMap.insert(multimap<time_t,string>::value_type(expiration,key));\r
+    ctx.m_dataMap[key]=Record(value,time(NULL),expiration);\r
+    ctx.m_expMap.insert(multimap<time_t,string>::value_type(expiration,key));\r
     \r
     m_log.debug("inserted record (%s)", key);\r
 }\r
 \r
-bool MemoryStorageService::readString(const char* key, string& value, time_t modifiedSince)\r
+bool MemoryStorageService::readString(const char* context, const char* key, string& value, time_t modifiedSince)\r
 {\r
-    SharedLock wrapper(m_lock);\r
-    map<string,Record>::iterator i=m_dataMap.find(key);\r
-    if (i==m_dataMap.end())\r
+    Context& ctx = getContext(context);\r
+\r
+    SharedLock wrapper(ctx.m_lock);\r
+    map<string,Record>::iterator i=ctx.m_dataMap.find(key);\r
+    if (i==ctx.m_dataMap.end())\r
         return false;\r
     else if (modifiedSince >= i->second.modified)\r
         return false;\r
@@ -204,14 +242,16 @@ bool MemoryStorageService::readString(const char* key, string& value, time_t mod
     return true;\r
 }\r
 \r
-bool MemoryStorageService::updateString(const char* key, const char* value, time_t expiration)\r
+bool MemoryStorageService::updateString(const char* context, const char* key, const char* value, time_t expiration)\r
 {\r
+    Context& ctx = getContext(context);\r
+\r
     // Lock the maps.\r
-    m_lock->wrlock();\r
-    SharedLock wrapper(m_lock, false);\r
+    ctx.m_lock->wrlock();\r
+    SharedLock wrapper(ctx.m_lock, false);\r
 \r
-    map<string,Record>::iterator i=m_dataMap.find(key);\r
-    if (i==m_dataMap.end())\r
+    map<string,Record>::iterator i=ctx.m_dataMap.find(key);\r
+    if (i==ctx.m_dataMap.end())\r
         return false;\r
         \r
     if (value)\r
@@ -219,15 +259,16 @@ bool MemoryStorageService::updateString(const char* key, const char* value, time
         \r
     if (expiration && expiration != i->second.expiration) {\r
         // Update secondary map.\r
-        pair<multimap<time_t,string>::iterator,multimap<time_t,string>::iterator> range=m_expMap.equal_range(i->second.expiration);\r
+        pair<multimap<time_t,string>::iterator,multimap<time_t,string>::iterator> range =\r
+            ctx.m_expMap.equal_range(i->second.expiration);\r
         for (; range.first != range.second; ++range.first) {\r
             if (range.first->second == i->first) {\r
-                m_expMap.erase(range.first);\r
+                ctx.m_expMap.erase(range.first);\r
                 break;\r
             }\r
         }\r
         i->second.expiration = expiration;\r
-        m_expMap.insert(multimap<time_t,string>::value_type(expiration,key));\r
+       ctx.m_expMap.insert(multimap<time_t,string>::value_type(expiration,key));\r
     }\r
 \r
     i->second.modified = time(NULL);\r
@@ -235,25 +276,28 @@ bool MemoryStorageService::updateString(const char* key, const char* value, time
     return true;\r
 }\r
 \r
-bool MemoryStorageService::deleteString(const char* key)\r
+bool MemoryStorageService::deleteString(const char* context, const char* key)\r
 {\r
+    Context& ctx = getContext(context);\r
+\r
     // Lock the maps.\r
-    m_lock->wrlock();\r
-    SharedLock wrapper(m_lock, false);\r
+    ctx.m_lock->wrlock();\r
+    SharedLock wrapper(ctx.m_lock, false);\r
     \r
     // Find the record.\r
-    map<string,Record>::iterator i=m_dataMap.find(key);\r
-    if (i!=m_dataMap.end()) {\r
+    map<string,Record>::iterator i=ctx.m_dataMap.find(key);\r
+    if (i!=ctx.m_dataMap.end()) {\r
         // Now find the reversed index of expiration to key, so we can clear it.\r
-        pair<multimap<time_t,string>::iterator,multimap<time_t,string>::iterator> range=m_expMap.equal_range(i->second.expiration);\r
+        pair<multimap<time_t,string>::iterator,multimap<time_t,string>::iterator> range =\r
+            ctx.m_expMap.equal_range(i->second.expiration);\r
         for (; range.first != range.second; ++range.first) {\r
             if (range.first->second == i->first) {\r
-                m_expMap.erase(range.first);\r
+                ctx.m_expMap.erase(range.first);\r
                 break;\r
             }\r
         }\r
         // And finally delete the record itself.\r
-        m_dataMap.erase(i);\r
+        ctx.m_dataMap.erase(i);\r
         m_log.debug("deleted record (%s)", key);\r
         return true;\r
     }\r