ThreadableWebSocketChannelClientWrapper.h [plain text]
#ifndef ThreadableWebSocketChannelClientWrapper_h
#define ThreadableWebSocketChannelClientWrapper_h
#if ENABLE(WEB_SOCKETS)
#include "PlatformString.h"
#include "WebSocketChannelClient.h"
#include <wtf/PassRefPtr.h>
#include <wtf/Threading.h>
#include <wtf/Vector.h>
namespace WebCore {
class String;
class ThreadableWebSocketChannelClientWrapper : public ThreadSafeShared<ThreadableWebSocketChannelClientWrapper> {
public:
static PassRefPtr<ThreadableWebSocketChannelClientWrapper> create(WebSocketChannelClient* client)
{
return adoptRef(new ThreadableWebSocketChannelClientWrapper(client));
}
void clearSyncMethodDone()
{
m_syncMethodDone = false;
}
void setSyncMethodDone()
{
m_syncMethodDone = true;
}
bool syncMethodDone() const
{
return m_syncMethodDone;
}
bool sent() const
{
return m_sent;
}
void setSent(bool sent)
{
m_sent = sent;
m_syncMethodDone = true;
}
unsigned long bufferedAmount() const
{
return m_bufferedAmount;
}
void setBufferedAmount(unsigned long bufferedAmount)
{
m_bufferedAmount = bufferedAmount;
m_syncMethodDone = true;
}
void clearClient()
{
m_client = 0;
}
void didConnect()
{
m_pendingConnected = true;
if (!m_suspended)
processPendingEvents();
}
void didReceiveMessage(const String& msg)
{
m_pendingMessages.append(msg);
if (!m_suspended)
processPendingEvents();
}
void didClose(unsigned long unhandledBufferedAmount)
{
m_pendingClosed = true;
m_bufferedAmount = unhandledBufferedAmount;
if (!m_suspended)
processPendingEvents();
}
void suspend()
{
m_suspended = true;
}
void resume()
{
m_suspended = false;
processPendingEvents();
}
protected:
ThreadableWebSocketChannelClientWrapper(WebSocketChannelClient* client)
: m_client(client)
, m_syncMethodDone(false)
, m_sent(false)
, m_bufferedAmount(0)
, m_suspended(false)
, m_pendingConnected(false)
, m_pendingClosed(false)
{
}
void 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);
}
}
WebSocketChannelClient* m_client;
bool m_syncMethodDone;
bool m_sent;
unsigned long m_bufferedAmount;
bool m_suspended;
bool m_pendingConnected;
Vector<String> m_pendingMessages;
bool m_pendingClosed;
};
}
#endif // ENABLE(WEB_SOCKETS)
#endif // ThreadableWebSocketChannelClientWrapper_h