Merge branch 'devel' of https://github.com/arangodb/arangodb into devel

arangod/Scheduler/SocketTask.cpp

@@ -26,6 +26,7 @@
 
 #include "SocketTask.h"
 
+#include "Basics/MutexLocker.h"
 #include "Basics/StringBuffer.h"
 #include "Basics/socket-utils.h"
 #include "Endpoint/ConnectionInfo.h"
@@ -134,12 +135,14 @@ void SocketTask::addWriteBuffer(StringBuffer* buffer,
       LOG_TOPIC(TRACE, Logger::COMMUNICATION)
           << "SocketTask::addWriteBuffer - "
              "Statistics release: nullptr - "
-             "nothing to realease";
+             "nothing to release";
     }
 
     return;
   }
 
+  MUTEX_LOCKER(locker, _writeLock);
+
   if (_writeBuffer != nullptr) {
     _writeBuffers.push_back(buffer);
     _writeBuffersStats.push_back(stat);
@@ -156,19 +159,19 @@ void SocketTask::addWriteBuffer(StringBuffer* buffer,
   }
 
   if (_writeBuffer != nullptr) {
-    boost::system::error_code ec;
     size_t total = _writeBuffer->length();
     size_t written = 0;
 
     boost::system::error_code err;
     do {
-      ec.assign(boost::system::errc::success,
+      err.assign(boost::system::errc::success,
                  boost::system::generic_category());
       written = _peer->write(_writeBuffer, err);
       if(_writeBufferStatistics){
         _writeBufferStatistics->_sentBytes += written;
       }
       if (written == total) {
+        locker.unlock();
         completedWriteBuffer();
         return;
       }
@@ -182,18 +185,26 @@ void SocketTask::addWriteBuffer(StringBuffer* buffer,
       return;
     }
       
+    boost::system::error_code ec;
+    ec.assign(boost::system::errc::success,
+              boost::system::generic_category());
+
     auto self = shared_from_this();
     auto handler = [self, this](const boost::system::error_code& ec,
                                 std::size_t transferred) {
+      MUTEX_LOCKER(locker, _writeLock);
+
       if (_writeBufferStatistics){
         _writeBufferStatistics->_sentBytes += transferred;
       }
+
       if (ec) {
         LOG_TOPIC(DEBUG, Logger::COMMUNICATION)
             << "SocketTask::addWriterBuffer(async_write) - write on stream "
             << " failed with: " << ec.message();
         closeStream();
       } else {
+        locker.unlock();
         completedWriteBuffer();
       }
     };
@@ -205,6 +216,11 @@ void SocketTask::addWriteBuffer(StringBuffer* buffer,
 }
 
 void SocketTask::completedWriteBuffer() {
+  StringBuffer* buffer = nullptr;
+  TRI_request_statistics_t* statistics = nullptr;
+  {
+    MUTEX_LOCKER(locker, _writeLock);
+
     delete _writeBuffer;
     _writeBuffer = nullptr;
 
@@ -221,7 +237,7 @@ void SocketTask::completedWriteBuffer() {
 #ifdef DEBUG_STATISTICS
       LOG_TOPIC(TRACE, Logger::REQUESTS) << "SocketTask::addWriteBuffer - "
         "Statistics release: nullptr - "
-                                          "nothing to realease";
+        "nothing to release";
 #endif
     }
 
@@ -237,12 +253,12 @@ void SocketTask::completedWriteBuffer() {
       return;
     }
 
-  StringBuffer* buffer = _writeBuffers.front();
+    buffer = _writeBuffers.front();
     _writeBuffers.pop_front();
 
-  TRI_request_statistics_t* statistics = _writeBuffersStats.front();
+    statistics = _writeBuffersStats.front();
     _writeBuffersStats.pop_front();
-
+  }
   addWriteBuffer(buffer, statistics);
 }
 

arangod/Scheduler/SocketTask.h

@@ -29,6 +29,7 @@
 
 #include <boost/asio/ssl.hpp>
 
+#include "Basics/Mutex.h"
 #include "Basics/StringBuffer.h"
 #include "Basics/asio-helper.h"
 #include "Scheduler/Socket.h"
@@ -93,12 +94,15 @@ class SocketTask : virtual public Task, public ConnectionStatisticsAgent {
 
   basics::StringBuffer _readBuffer;
 
+ private:
+  Mutex _writeLock;
   basics::StringBuffer* _writeBuffer = nullptr;
   TRI_request_statistics_t* _writeBufferStatistics = nullptr;
 
   std::deque<basics::StringBuffer*> _writeBuffers;
   std::deque<TRI_request_statistics_t*> _writeBuffersStats;
 
+ protected:
   std::unique_ptr<Socket> _peer;
   boost::posix_time::milliseconds _keepAliveTimeout;
   boost::asio::deadline_timer _keepAliveTimer;

lib/Basics/MutexLocker.cpp

@@ -40,7 +40,7 @@ using namespace arangodb::basics;
 #ifdef TRI_SHOW_LOCK_TIME
 
 MutexLocker::MutexLocker(Mutex* mutex, char const* file, int line)
-    : _mutex(mutex), _file(file), _line(line), _time(0.0) {
+    : _mutex(mutex), isLocked(true), _isLocked(true), _file(file), _line(line), _time(0.0) {
   double t = TRI_microtime();
   _mutex->lock();
   _time = TRI_microtime() - t;
@@ -48,7 +48,9 @@ MutexLocker::MutexLocker(Mutex* mutex, char const* file, int line)
 
 #else
 
-MutexLocker::MutexLocker(Mutex* mutex) : _mutex(mutex) { _mutex->lock(); }
+MutexLocker::MutexLocker(Mutex* mutex) : _mutex(mutex), _isLocked(true) { 
+  _mutex->lock(); 
+}
 
 #endif
 
@@ -57,7 +59,9 @@ MutexLocker::MutexLocker(Mutex* mutex) : _mutex(mutex) { _mutex->lock(); }
 ////////////////////////////////////////////////////////////////////////////////
 
 MutexLocker::~MutexLocker() {
+  if (_isLocked) {
     _mutex->unlock();
+  }
 
 #ifdef TRI_SHOW_LOCK_TIME
   if (_time > TRI_SHOW_LOCK_THRESHOLD) {
@@ -65,3 +69,10 @@ MutexLocker::~MutexLocker() {
   }
 #endif
 }
+
+void MutexLocker::unlock() {
+  if (_isLocked) {
+    _mutex->unlock();
+    _isLocked = false;
+  }
+}

lib/Basics/MutexLocker.h

@@ -79,11 +79,16 @@ class MutexLocker {
 
 #endif
 
+
+  ~MutexLocker();
+  
+  bool isLocked() const { return _isLocked; }
+  
   //////////////////////////////////////////////////////////////////////////////
   /// @brief releases the lock
   //////////////////////////////////////////////////////////////////////////////
   
-  ~MutexLocker();
+  void unlock();
 
  private:
   //////////////////////////////////////////////////////////////////////////////
@@ -92,6 +97,12 @@ class MutexLocker {
 
   Mutex* _mutex;
   
+  //////////////////////////////////////////////////////////////////////////////
+  /// @brief whether or not the mutex is locked
+  //////////////////////////////////////////////////////////////////////////////
+
+  bool _isLocked;
+
 #ifdef TRI_SHOW_LOCK_TIME
 
   //////////////////////////////////////////////////////////////////////////////