ThreadableWebSocketChannelClientWrapper.cpp [plain text]
#include "config.h"
#if ENABLE(WEB_SOCKETS)
#include "ThreadableWebSocketChannelClientWrapper.h"
#include "WebSocketChannelClient.h"
#include <wtf/PassRefPtr.h>
namespace WebCore {
ThreadableWebSocketChannelClientWrapper::ThreadableWebSocketChannelClientWrapper(WebSocketChannelClient* client)
: m_client(client)
, m_syncMethodDone(false)
, m_sent(false)
, m_bufferedAmount(0)
, m_suspended(false)
, m_pendingConnected(false)
, m_pendingClosed(false)
{
}
PassRefPtr<ThreadableWebSocketChannelClientWrapper> ThreadableWebSocketChannelClientWrapper::create(WebSocketChannelClient* client)
{
return adoptRef(new ThreadableWebSocketChannelClientWrapper(client));
}
void ThreadableWebSocketChannelClientWrapper::clearSyncMethodDone()
{
m_syncMethodDone = false;
}
void ThreadableWebSocketChannelClientWrapper::setSyncMethodDone()
{
m_syncMethodDone = true;
}
bool ThreadableWebSocketChannelClientWrapper::syncMethodDone() const
{
return m_syncMethodDone;
}
bool ThreadableWebSocketChannelClientWrapper::sent() const
{
return m_sent;
}
void ThreadableWebSocketChannelClientWrapper::setSent(bool sent)
{
m_sent = sent;
m_syncMethodDone = true;
}
unsigned long ThreadableWebSocketChannelClientWrapper::bufferedAmount() const
{
return m_bufferedAmount;
}
void ThreadableWebSocketChannelClientWrapper::setBufferedAmount(unsigned long bufferedAmount)
{
m_bufferedAmount = bufferedAmount;
m_syncMethodDone = true;
}
void ThreadableWebSocketChannelClientWrapper::clearClient()
{
m_client = 0;
}
void ThreadableWebSocketChannelClientWrapper::didConnect()
{
m_pendingConnected = true;
if (!m_suspended)
processPendingEvents();
}
void ThreadableWebSocketChannelClientWrapper::didReceiveMessage(const String& message)
{
m_pendingMessages.append(message);
if (!m_suspended)
processPendingEvents();
}
void ThreadableWebSocketChannelClientWrapper::didClose(unsigned long unhandledBufferedAmount)
{
m_pendingClosed = true;
m_bufferedAmount = unhandledBufferedAmount;
if (!m_suspended)
processPendingEvents();
}
void ThreadableWebSocketChannelClientWrapper::suspend()
{
m_suspended = true;
}
void ThreadableWebSocketChannelClientWrapper::resume()
{
m_suspended = false;
processPendingEvents();
}
void ThreadableWebSocketChannelClientWrapper::processPendingEvents()
{
ASSERT(!m_suspended);
if (m_pendingConnected) {
m_pendingConnected = false;
if (m_client)
m_client->didConnect();
}
Vector<String> messages;
messages.swap(m_pendingMessages);
for (Vector<String>::const_iterator iter = messages.begin(); iter != messages.end(); ++iter) {
if (m_client)
m_client->didReceiveMessage(*iter);
}
if (m_pendingClosed) {
m_pendingClosed = false;
if (m_client)
m_client->didClose(m_bufferedAmount);
}
}
}
#endif