WorkerThreadableWebSocketChannel.h [plain text]
#pragma once
#include "ThreadableWebSocketChannel.h"
#include "WebSocketChannelClient.h"
#include "WorkerGlobalScope.h"
#include <wtf/RefCounted.h>
#include <wtf/RefPtr.h>
#include <wtf/text/WTFString.h>
namespace WebCore {
class ScriptExecutionContext;
class ThreadableWebSocketChannelClientWrapper;
class WorkerGlobalScope;
class WorkerLoaderProxy;
class WorkerRunLoop;
class WorkerThreadableWebSocketChannel : public RefCounted<WorkerThreadableWebSocketChannel>, public ThreadableWebSocketChannel {
WTF_MAKE_FAST_ALLOCATED;
public:
static Ref<ThreadableWebSocketChannel> create(WorkerGlobalScope& workerGlobalScope, WebSocketChannelClient& client, const String& taskMode, SocketProvider& provider)
{
return adoptRef(*new WorkerThreadableWebSocketChannel(workerGlobalScope, client, taskMode, provider));
}
virtual ~WorkerThreadableWebSocketChannel();
void connect(const URL&, const String& protocol) override;
String subprotocol() override;
String extensions() override;
ThreadableWebSocketChannel::SendResult send(const String& message) override;
ThreadableWebSocketChannel::SendResult send(const JSC::ArrayBuffer&, unsigned byteOffset, unsigned byteLength) override;
ThreadableWebSocketChannel::SendResult send(Blob&) override;
unsigned bufferedAmount() const override;
void close(int code, const String& reason) override;
void fail(const String& reason) override;
void disconnect() override; void suspend() override;
void resume() override;
class Peer : public WebSocketChannelClient {
WTF_MAKE_NONCOPYABLE(Peer); WTF_MAKE_FAST_ALLOCATED;
public:
Peer(Ref<ThreadableWebSocketChannelClientWrapper>&&, WorkerLoaderProxy&, ScriptExecutionContext&, const String& taskMode, SocketProvider&);
~Peer();
void connect(const URL&, const String& protocol);
void send(const String& message);
void send(const JSC::ArrayBuffer&);
void send(Blob&);
void bufferedAmount();
void close(int code, const String& reason);
void fail(const String& reason);
void disconnect();
void suspend();
void resume();
void didConnect() final;
void didReceiveMessage(const String& message) final;
void didReceiveBinaryData(Vector<uint8_t>&&) final;
void didUpdateBufferedAmount(unsigned bufferedAmount) final;
void didStartClosingHandshake() final;
void didClose(unsigned unhandledBufferedAmount, ClosingHandshakeCompletionStatus, unsigned short code, const String& reason) final;
void didReceiveMessageError() final;
void didUpgradeURL() final;
private:
Ref<ThreadableWebSocketChannelClientWrapper> m_workerClientWrapper;
WorkerLoaderProxy& m_loaderProxy;
RefPtr<ThreadableWebSocketChannel> m_mainWebSocketChannel;
String m_taskMode;
};
using RefCounted<WorkerThreadableWebSocketChannel>::ref;
using RefCounted<WorkerThreadableWebSocketChannel>::deref;
protected:
void refThreadableWebSocketChannel() override { ref(); }
void derefThreadableWebSocketChannel() override { deref(); }
private:
class Bridge : public RefCounted<Bridge> {
public:
static Ref<Bridge> create(Ref<ThreadableWebSocketChannelClientWrapper>&& workerClientWrapper, Ref<WorkerGlobalScope>&& workerGlobalScope, const String& taskMode, Ref<SocketProvider>&& provider)
{
return adoptRef(*new Bridge(WTFMove(workerClientWrapper), WTFMove(workerGlobalScope), taskMode, WTFMove(provider)));
}
~Bridge();
void initialize();
void connect(const URL&, const String& protocol);
ThreadableWebSocketChannel::SendResult send(const String& message);
ThreadableWebSocketChannel::SendResult send(const JSC::ArrayBuffer&, unsigned byteOffset, unsigned byteLength);
ThreadableWebSocketChannel::SendResult send(Blob&);
unsigned bufferedAmount();
void close(int code, const String& reason);
void fail(const String& reason);
void disconnect();
void suspend();
void resume();
using RefCounted<Bridge>::ref;
using RefCounted<Bridge>::deref;
private:
Bridge(Ref<ThreadableWebSocketChannelClientWrapper>&&, Ref<WorkerGlobalScope>&&, const String& taskMode, Ref<SocketProvider>&&);
static void setWebSocketChannel(ScriptExecutionContext*, Bridge* thisPtr, Peer*, Ref<ThreadableWebSocketChannelClientWrapper>&&);
static void mainThreadInitialize(ScriptExecutionContext&, WorkerLoaderProxy&, Ref<ThreadableWebSocketChannelClientWrapper>&&, const String& taskMode, Ref<SocketProvider>&&);
void clearClientWrapper();
void setMethodNotCompleted();
void waitForMethodCompletion();
Ref<ThreadableWebSocketChannelClientWrapper> m_workerClientWrapper;
RefPtr<WorkerGlobalScope> m_workerGlobalScope;
WorkerLoaderProxy& m_loaderProxy;
String m_taskMode;
Peer* m_peer { nullptr };
Ref<SocketProvider> m_socketProvider;
};
WEBCORE_EXPORT WorkerThreadableWebSocketChannel(WorkerGlobalScope&, WebSocketChannelClient&, const String& taskMode, SocketProvider&);
class WorkerGlobalScopeDidInitializeTask;
Ref<WorkerGlobalScope> m_workerGlobalScope;
Ref<ThreadableWebSocketChannelClientWrapper> m_workerClientWrapper;
RefPtr<Bridge> m_bridge;
Ref<SocketProvider> m_socketProvider;
};
}