-/*\r
- * Copyright 2001-2005 Internet2\r
- * \r
- * Licensed under the Apache License, Version 2.0 (the "License");\r
- * you may not use this file except in compliance with the License.\r
- * You may obtain a copy of the License at\r
- *\r
- * http://www.apache.org/licenses/LICENSE-2.0\r
- *\r
- * Unless required by applicable law or agreed to in writing, software\r
- * distributed under the License is distributed on an "AS IS" BASIS,\r
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
- * See the License for the specific language governing permissions and\r
- * limitations under the License.\r
- */\r
-\r
-/**\r
- * MemoryStorageService.cpp\r
- * \r
- * In-memory "persistent" storage, suitable for simple applications.\r
- */\r
-\r
-#include "internal.h"\r
-#include "util/NDC.h"\r
-#include "util/StorageService.h"\r
-#include "util/Threads.h"\r
-#include "util/XMLHelper.h"\r
-\r
-#include <log4cpp/Category.hh>\r
-#include <xercesc/util/XMLUniDefs.hpp>\r
-\r
-using namespace xmltooling;\r
-using namespace log4cpp;\r
-using namespace std;\r
-\r
-namespace xmltooling {\r
- class XMLTOOL_DLLLOCAL MemoryStorageService : public StorageService\r
- {\r
- public:\r
- MemoryStorageService(const DOMElement* e);\r
- virtual ~MemoryStorageService();\r
- \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* pvalue=NULL, time_t* pexpiration=NULL);\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* context, const char* key, const char* value, time_t expiration) {\r
- return createString(context, key, value, expiration);\r
- }\r
- bool readText(const char* context, const char* key, string* pvalue=NULL, time_t* pexpiration=NULL) {\r
- return readString(context, key, pvalue, pexpiration);\r
- }\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* context, const char* key) {\r
- return deleteString(context, key);\r
- }\r
- \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
- void cleanup();\r
- \r
- struct XMLTOOL_DLLLOCAL Record {\r
- Record() : expiration(0) {}\r
- Record(string s, time_t t) : data(s), expiration(t) {}\r
- string data;\r
- time_t expiration;\r
- };\r
- \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
- bool shutdown;\r
- int m_cleanupInterval;\r
- Category& m_log;\r
- };\r
-\r
- StorageService* XMLTOOL_DLLLOCAL MemoryStorageServiceFactory(const DOMElement* const & e)\r
- {\r
- return new MemoryStorageService(e);\r
- }\r
-\r
-};\r
-\r
-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
- : contextLock(NULL), shutdown_wait(NULL), cleanup_thread(NULL), shutdown(false), m_cleanupInterval(0),\r
- m_log(Category::getInstance(XMLTOOLING_LOGCAT".StorageService"))\r
-{\r
- const XMLCh* tag=e ? e->getAttributeNS(NULL,cleanupInterval) : NULL;\r
- if (tag && *tag) {\r
- m_cleanupInterval = XMLString::parseInt(tag);\r
- }\r
- if (!m_cleanupInterval)\r
- m_cleanupInterval=300;\r
-\r
- contextLock = Mutex::create();\r
- shutdown_wait = CondWait::create();\r
- cleanup_thread = Thread::create(&cleanup_fn, (void*)this);\r
-}\r
-\r
-MemoryStorageService::~MemoryStorageService()\r
-{\r
- // Shut down the cleanup thread and let it know...\r
- shutdown = true;\r
- shutdown_wait->signal();\r
- cleanup_thread->join(NULL);\r
-\r
- delete shutdown_wait;\r
- delete contextLock;\r
-}\r
-\r
-void* MemoryStorageService::cleanup_fn(void* cache_p)\r
-{\r
- MemoryStorageService* cache = reinterpret_cast<MemoryStorageService*>(cache_p);\r
-\r
-#ifndef WIN32\r
- // First, let's block all signals \r
- Thread::mask_all_signals();\r
-#endif\r
-\r
- // Now run the cleanup process.\r
- cache->cleanup();\r
- return NULL;\r
-}\r
-\r
-void MemoryStorageService::cleanup()\r
-{\r
-#ifdef _DEBUG\r
- NDC ndc("cleanup");\r
-#endif\r
- \r
-\r
- Mutex* mutex = Mutex::create();\r
- mutex->lock();\r
-\r
- m_log.info("cleanup thread started...running every %d seconds", m_cleanupInterval);\r
-\r
- while (!shutdown) {\r
- shutdown_wait->timedwait(mutex, m_cleanupInterval);\r
- if (shutdown)\r
- break;\r
- \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
- }\r
-\r
- m_log.info("cleanup thread finished");\r
-\r
- mutex->unlock();\r
- delete mutex;\r
- Thread::exit(NULL);\r
-}\r
-\r
-void MemoryStorageService::reap(const char* context)\r
-{\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
- multimap<time_t,string>::iterator stop=m_expMap.upper_bound(time(NULL));\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=ctx.m_dataMap.find(key);\r
- if (i!=ctx.m_dataMap.end()) {\r
- // Not yet expired?\r
- if (time(NULL) < i->second.expiration)\r
- throw IOException("attempted to insert a record with duplicate key ($1)", params(1,key));\r
- // It's dead, so we can just remove it now and create the new record.\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 =\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
- ctx.m_expMap.erase(range.first);\r
- break;\r
- }\r
- }\r
- // And finally delete the record itself.\r
- ctx.m_dataMap.erase(i);\r
- }\r
- \r
- ctx.m_dataMap[key]=Record(value,expiration);\r
- ctx.m_expMap.insert(multimap<time_t,string>::value_type(expiration,key));\r
- \r
- m_log.debug("inserted record (%s) in context (%s)", key, context);\r
-}\r
-\r
-bool MemoryStorageService::readString(const char* context, const char* key, string* pvalue, time_t* pexpiration)\r
-{\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 (time(NULL) >= i->second.expiration)\r
- return false;\r
- if (pvalue)\r
- *pvalue = i->second.data;\r
- if (pexpiration)\r
- *pexpiration = i->second.expiration;\r
- return true;\r
-}\r
-\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
- ctx.m_lock->wrlock();\r
- SharedLock wrapper(ctx.m_lock, false);\r
-\r
- map<string,Record>::iterator i=ctx.m_dataMap.find(key);\r
- if (i==ctx.m_dataMap.end())\r
- return false;\r
- else if (time(NULL) >= i->second.expiration)\r
- return false;\r
- \r
- if (value)\r
- i->second.data = value;\r
- \r
- if (expiration && expiration != i->second.expiration) {\r
- // Update secondary map.\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
- ctx.m_expMap.erase(range.first);\r
- break;\r
- }\r
- }\r
- i->second.expiration = expiration;\r
- ctx.m_expMap.insert(multimap<time_t,string>::value_type(expiration,key));\r
- }\r
-\r
- m_log.debug("updated record (%s) in context (%s)", key, context);\r
- return true;\r
-}\r
-\r
-bool MemoryStorageService::deleteString(const char* context, const char* key)\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
- // Find the record.\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 =\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
- ctx.m_expMap.erase(range.first);\r
- break;\r
- }\r
- }\r
- // And finally delete the record itself.\r
- ctx.m_dataMap.erase(i);\r
- m_log.debug("deleted record (%s) in context (%s)", key, context);\r
- return true;\r
- }\r
-\r
- m_log.debug("deleting record (%s) in context (%s)....not found", key, context);\r
- return false;\r
-}\r
+/*
+ * Copyright 2001-2005 Internet2
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/**
+ * MemoryStorageService.cpp
+ *
+ * In-memory "persistent" storage, suitable for simple applications.
+ */
+
+#include "internal.h"
+#include "util/NDC.h"
+#include "util/StorageService.h"
+#include "util/Threads.h"
+#include "util/XMLHelper.h"
+
+#include <log4cpp/Category.hh>
+#include <xercesc/util/XMLUniDefs.hpp>
+
+using namespace xmltooling;
+using namespace log4cpp;
+using namespace std;
+
+namespace xmltooling {
+ class XMLTOOL_DLLLOCAL MemoryStorageService : public StorageService
+ {
+ public:
+ MemoryStorageService(const DOMElement* e);
+ virtual ~MemoryStorageService();
+
+ void createString(const char* context, const char* key, const char* value, time_t expiration);
+ bool readString(const char* context, const char* key, string* pvalue=NULL, time_t* pexpiration=NULL);
+ bool updateString(const char* context, const char* key, const char* value=NULL, time_t expiration=0);
+ bool deleteString(const char* context, const char* key);
+
+ void createText(const char* context, const char* key, const char* value, time_t expiration) {
+ return createString(context, key, value, expiration);
+ }
+ bool readText(const char* context, const char* key, string* pvalue=NULL, time_t* pexpiration=NULL) {
+ return readString(context, key, pvalue, pexpiration);
+ }
+ bool updateText(const char* context, const char* key, const char* value=NULL, time_t expiration=0) {
+ return updateString(context, key, value, expiration);
+ }
+ bool deleteText(const char* context, const char* key) {
+ return deleteString(context, key);
+ }
+
+ void reap(const char* context);
+ void deleteContext(const char* context) {
+ Lock wrapper(contextLock);
+ m_contextMap.erase(context);
+ }
+
+ private:
+ void cleanup();
+
+ struct XMLTOOL_DLLLOCAL Record {
+ Record() : expiration(0) {}
+ Record(string s, time_t t) : data(s), expiration(t) {}
+ string data;
+ time_t expiration;
+ };
+
+ struct XMLTOOL_DLLLOCAL Context {
+ Context() : m_lock(RWLock::create()) {}
+ Context(const Context& src) {
+ m_dataMap = src.m_dataMap;
+ m_expMap = src.m_expMap;
+ m_lock = RWLock::create();
+ }
+ ~Context() { delete m_lock; }
+ map<string,Record> m_dataMap;
+ multimap<time_t,string> m_expMap;
+ RWLock* m_lock;
+ unsigned long reap();
+ };
+
+ Context& getContext(const char* context) {
+ Lock wrapper(contextLock);
+ return m_contextMap[context];
+ }
+
+ map<string,Context> m_contextMap;
+ Mutex* contextLock;
+ CondWait* shutdown_wait;
+ Thread* cleanup_thread;
+ static void* cleanup_fn(void*);
+ bool shutdown;
+ int m_cleanupInterval;
+ Category& m_log;
+ };
+
+ StorageService* XMLTOOL_DLLLOCAL MemoryStorageServiceFactory(const DOMElement* const & e)
+ {
+ return new MemoryStorageService(e);
+ }
+
+};
+
+static const XMLCh cleanupInterval[] = UNICODE_LITERAL_15(c,l,e,a,n,u,p,I,n,t,e,r,v,a,l);
+
+MemoryStorageService::MemoryStorageService(const DOMElement* e)
+ : contextLock(NULL), shutdown_wait(NULL), cleanup_thread(NULL), shutdown(false), m_cleanupInterval(0),
+ m_log(Category::getInstance(XMLTOOLING_LOGCAT".StorageService"))
+{
+ const XMLCh* tag=e ? e->getAttributeNS(NULL,cleanupInterval) : NULL;
+ if (tag && *tag) {
+ m_cleanupInterval = XMLString::parseInt(tag);
+ }
+ if (!m_cleanupInterval)
+ m_cleanupInterval=300;
+
+ contextLock = Mutex::create();
+ shutdown_wait = CondWait::create();
+ cleanup_thread = Thread::create(&cleanup_fn, (void*)this);
+}
+
+MemoryStorageService::~MemoryStorageService()
+{
+ // Shut down the cleanup thread and let it know...
+ shutdown = true;
+ shutdown_wait->signal();
+ cleanup_thread->join(NULL);
+
+ delete shutdown_wait;
+ delete contextLock;
+}
+
+void* MemoryStorageService::cleanup_fn(void* cache_p)
+{
+ MemoryStorageService* cache = reinterpret_cast<MemoryStorageService*>(cache_p);
+
+#ifndef WIN32
+ // First, let's block all signals
+ Thread::mask_all_signals();
+#endif
+
+ // Now run the cleanup process.
+ cache->cleanup();
+ return NULL;
+}
+
+void MemoryStorageService::cleanup()
+{
+#ifdef _DEBUG
+ NDC ndc("cleanup");
+#endif
+
+
+ Mutex* mutex = Mutex::create();
+ mutex->lock();
+
+ m_log.info("cleanup thread started...running every %d seconds", m_cleanupInterval);
+
+ while (!shutdown) {
+ shutdown_wait->timedwait(mutex, m_cleanupInterval);
+ if (shutdown)
+ break;
+
+ unsigned long count=0;
+ Lock wrapper(contextLock);
+ for (map<string,Context>::iterator i=m_contextMap.begin(); i!=m_contextMap.end(); ++i)
+ count += i->second.reap();
+
+ if (count)
+ m_log.info("purged %d record(s) from storage", count);
+ }
+
+ m_log.info("cleanup thread finished");
+
+ mutex->unlock();
+ delete mutex;
+ Thread::exit(NULL);
+}
+
+void MemoryStorageService::reap(const char* context)
+{
+ getContext(context).reap();
+}
+
+unsigned long MemoryStorageService::Context::reap()
+{
+ // Lock the "database".
+ m_lock->wrlock();
+ SharedLock wrapper(m_lock, false);
+
+ // Garbage collect any expired entries.
+ unsigned long count=0;
+ multimap<time_t,string>::iterator stop=m_expMap.upper_bound(time(NULL));
+ for (multimap<time_t,string>::iterator i=m_expMap.begin(); i!=stop; m_expMap.erase(i++)) {
+ m_dataMap.erase(i->second);
+ ++count;
+ }
+
+ return count;
+}
+
+void MemoryStorageService::createString(const char* context, const char* key, const char* value, time_t expiration)
+{
+ Context& ctx = getContext(context);
+
+ // Lock the maps.
+ ctx.m_lock->wrlock();
+ SharedLock wrapper(ctx.m_lock, false);
+
+ // Check for a duplicate.
+ map<string,Record>::iterator i=ctx.m_dataMap.find(key);
+ if (i!=ctx.m_dataMap.end()) {
+ // Not yet expired?
+ if (time(NULL) < i->second.expiration)
+ throw IOException("attempted to insert a record with duplicate key ($1)", params(1,key));
+ // It's dead, so we can just remove it now and create the new record.
+ // Now find the reversed index of expiration to key, so we can clear it.
+ pair<multimap<time_t,string>::iterator,multimap<time_t,string>::iterator> range =
+ ctx.m_expMap.equal_range(i->second.expiration);
+ for (; range.first != range.second; ++range.first) {
+ if (range.first->second == i->first) {
+ ctx.m_expMap.erase(range.first);
+ break;
+ }
+ }
+ // And finally delete the record itself.
+ ctx.m_dataMap.erase(i);
+ }
+
+ ctx.m_dataMap[key]=Record(value,expiration);
+ ctx.m_expMap.insert(multimap<time_t,string>::value_type(expiration,key));
+
+ m_log.debug("inserted record (%s) in context (%s)", key, context);
+}
+
+bool MemoryStorageService::readString(const char* context, const char* key, string* pvalue, time_t* pexpiration)
+{
+ Context& ctx = getContext(context);
+
+ SharedLock wrapper(ctx.m_lock);
+ map<string,Record>::iterator i=ctx.m_dataMap.find(key);
+ if (i==ctx.m_dataMap.end())
+ return false;
+ else if (time(NULL) >= i->second.expiration)
+ return false;
+ if (pvalue)
+ *pvalue = i->second.data;
+ if (pexpiration)
+ *pexpiration = i->second.expiration;
+ return true;
+}
+
+bool MemoryStorageService::updateString(const char* context, const char* key, const char* value, time_t expiration)
+{
+ Context& ctx = getContext(context);
+
+ // Lock the maps.
+ ctx.m_lock->wrlock();
+ SharedLock wrapper(ctx.m_lock, false);
+
+ map<string,Record>::iterator i=ctx.m_dataMap.find(key);
+ if (i==ctx.m_dataMap.end())
+ return false;
+ else if (time(NULL) >= i->second.expiration)
+ return false;
+
+ if (value)
+ i->second.data = value;
+
+ if (expiration && expiration != i->second.expiration) {
+ // Update secondary map.
+ pair<multimap<time_t,string>::iterator,multimap<time_t,string>::iterator> range =
+ ctx.m_expMap.equal_range(i->second.expiration);
+ for (; range.first != range.second; ++range.first) {
+ if (range.first->second == i->first) {
+ ctx.m_expMap.erase(range.first);
+ break;
+ }
+ }
+ i->second.expiration = expiration;
+ ctx.m_expMap.insert(multimap<time_t,string>::value_type(expiration,key));
+ }
+
+ m_log.debug("updated record (%s) in context (%s)", key, context);
+ return true;
+}
+
+bool MemoryStorageService::deleteString(const char* context, const char* key)
+{
+ Context& ctx = getContext(context);
+
+ // Lock the maps.
+ ctx.m_lock->wrlock();
+ SharedLock wrapper(ctx.m_lock, false);
+
+ // Find the record.
+ map<string,Record>::iterator i=ctx.m_dataMap.find(key);
+ if (i!=ctx.m_dataMap.end()) {
+ // Now find the reversed index of expiration to key, so we can clear it.
+ pair<multimap<time_t,string>::iterator,multimap<time_t,string>::iterator> range =
+ ctx.m_expMap.equal_range(i->second.expiration);
+ for (; range.first != range.second; ++range.first) {
+ if (range.first->second == i->first) {
+ ctx.m_expMap.erase(range.first);
+ break;
+ }
+ }
+ // And finally delete the record itself.
+ ctx.m_dataMap.erase(i);
+ m_log.debug("deleted record (%s) in context (%s)", key, context);
+ return true;
+ }
+
+ m_log.debug("deleting record (%s) in context (%s)....not found", key, context);
+ return false;
+}