comparison base/Serialiser.cpp @ 1857:14c776dad920

Make Serialiser cancellable while waiting for its lock

     License, or (at your option) any later version.  See the file
     COPYING included with this distribution for more information.
 */
 
 #include "Serialiser.h"
+#include "Debug.h"
 
 #include <iostream>
 
 QMutex
 Serialiser::m_mapMutex;
 
 std::map<QString, QMutex *>
 Serialiser::m_mutexMap;
 
 Serialiser::Serialiser(QString id) :
-    m_id(id)
+    Serialiser(id, nullptr) { }
+
+Serialiser::Serialiser(QString id, const std::atomic<bool> *cancelled) :
+    m_id(id),
+    m_cancelled(cancelled),
+    m_locked(false)
 {
     m_mapMutex.lock();
     
     if (m_mutexMap.find(m_id) == m_mutexMap.end()) {
         m_mutexMap[m_id] = new QMutex;

 
     QMutex *idMutex = m_mutexMap[m_id];
 
     m_mapMutex.unlock();
 
-    idMutex->lock();
+    if (!m_cancelled) {
+        idMutex->lock();
+        m_locked = true;
+    } else {
+        // try to lock, polling the cancelled status occasionally
+        while (!m_locked && ! *m_cancelled) {
+            m_locked = idMutex->tryLock(500);
+            if (*m_cancelled) {
+                SVCERR << "Serialiser: cancelled!" << endl;
+            }
+        }
+    }
 }
 
 Serialiser::~Serialiser()
 {
     m_mapMutex.lock();
-    
-    m_mutexMap[m_id]->unlock();
+
+    if (m_locked) {
+        m_mutexMap[m_id]->unlock();
+    }
 
     m_mapMutex.unlock();
 }