rstudio/src/cpp/session/http/SessionHttpConnectionQueue.cpp at master · usersbin/rstudio · GitHub

History

127 lines (106 loc) · 3.04 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

* SessionHttpConnectionQueue.cpp

* This program is licensed to you under the terms of version 3 of the

* GNU Affero General Public License. This program is distributed WITHOUT

* ANY EXPRESS OR IMPLIED WARRANTY, INCLUDING THOSE OF NON-INFRINGEMENT,

* MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. Please refer to the

* AGPL (http://www.gnu.org/licenses/agpl-3.0.txt) for more details.

#include <session/SessionHttpConnectionQueue.hpp>

#include <core/Log.hpp>

#include <core/Error.hpp>

#include <core/Thread.hpp>

#include <core/http/Request.hpp>

#include "SessionHttpLog.hpp"

using namespace core ;

namespace session {

void HttpConnectionQueue::enqueConnection(

boost::shared_ptr<HttpConnection> ptrConnection)

{

LOCK_MUTEX(*pMutex_)

{

// enque

queue_.push(ptrConnection);

}

END_LOCK_MUTEX

pWaitCondition_->notify_all();

}

boost::shared_ptr<HttpConnection> HttpConnectionQueue::doDequeConnection()

{

LOCK_MUTEX(*pMutex_)

{

if (!queue_.empty())

{

// remove it

boost::shared_ptr<HttpConnection> next = queue_.front();

queue_.pop();

// return it

return next;

}

else

{

return boost::shared_ptr<HttpConnection>();

}

END_LOCK_MUTEX

// keep compiler happy

return boost::shared_ptr<HttpConnection>();

}

boost::shared_ptr<HttpConnection> HttpConnectionQueue::dequeConnection()

{

// perform the deque

boost::shared_ptr<HttpConnection> connection = doDequeConnection();

// log if we got one

if (connection)

httpLog().addEntry(HttpLog::ConnectionDequeued, connection->requestId());

// return the connection

return connection;

}

boost::shared_ptr<HttpConnection> HttpConnectionQueue::dequeConnection(

const boost::posix_time::time_duration& waitDuration)

{

// first see if we already have one

boost::shared_ptr<HttpConnection> ptrConnection = dequeConnection();

if (ptrConnection)

return ptrConnection;

// now wait the specified interval for one to materialize

if (waitForConnection(waitDuration))

return dequeConnection();

else

return boost::shared_ptr<HttpConnection>();

}

std::string HttpConnectionQueue::peekNextConnectionUri()

{

LOCK_MUTEX(*pMutex_)

{

if (!queue_.empty())

return queue_.front()->request().uri();

else

return std::string();

}

END_LOCK_MUTEX

// keep compiler happy

return std::string();

}

bool HttpConnectionQueue::waitForConnection(

const boost::posix_time::time_duration& waitDuration)

{

using namespace boost;

try

{

unique_lock<mutex> lock(*pMutex_);

system_time timeoutTime = get_system_time() + waitDuration;

return pWaitCondition_->timed_wait(lock, timeoutTime);

}

catch(const thread_resource_error& e)

{

Error waitError(boost::thread_error::ec_from_exception(e), ERROR_LOCATION) ;

LOG_ERROR(waitError);

return false ;

}

} // namespace session

rstudio/src/cpp/session/http/SessionHttpConnectionQueue.cpp at master · usersbin/rstudio · GitHub

Uh oh!

Expand file tree

Breadcrumbs

SessionHttpConnectionQueue.cpp

Latest commit

History

Breadcrumbs

SessionHttpConnectionQueue.cpp

File metadata and controls