| Index: third_party/WebKit/Source/core/streams/ReadableStreamController.h
|
| diff --git a/third_party/WebKit/Source/core/streams/ReadableStreamController.h b/third_party/WebKit/Source/core/streams/ReadableStreamController.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..435db846beb28d4e863245c39bba731ab03a9033
|
| --- /dev/null
|
| +++ b/third_party/WebKit/Source/core/streams/ReadableStreamController.h
|
| @@ -0,0 +1,121 @@
|
| +// Copyright 2016 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#ifndef ReadableStreamController_h
|
| +#define ReadableStreamController_h
|
| +
|
| +#include "bindings/core/v8/ScopedPersistent.h"
|
| +#include "bindings/core/v8/ScriptValue.h"
|
| +#include "bindings/core/v8/ToV8.h"
|
| +#include "bindings/core/v8/V8ScriptRunner.h"
|
| +#include "core/CoreExport.h"
|
| +#include "platform/heap/Handle.h"
|
| +#include "wtf/RefPtr.h"
|
| +#include <v8.h>
|
| +
|
| +namespace blink {
|
| +
|
| +class CORE_EXPORT ReadableStreamController final : public GarbageCollectedFinalized<ReadableStreamController> {
|
| +public:
|
| + DEFINE_INLINE_TRACE() {}
|
| +
|
| + explicit ReadableStreamController(ScriptValue stream)
|
| + : m_scriptState(stream.scriptState())
|
| + , m_stream(stream.isolate(), stream.v8Value())
|
| + {
|
| + m_stream.setWeak(&m_stream, ReadableStreamController::streamWeakCallback);
|
| + }
|
| +
|
| + // Users of the ReadableStreamController can call this to note that the stream has been canceled and thus they
|
| + // don't anticipate using the ReadableStreamController anymore. (close/desiredSize/enqueue/error will become no-ops
|
| + // afterward.)
|
| + void noteHasBeenCanceled()
|
| + {
|
| + m_stream.clear();
|
| + }
|
| +
|
| + bool isActive() const
|
| + {
|
| + return !m_stream.isEmpty();
|
| + }
|
| +
|
| + void close()
|
| + {
|
| + ScriptState* scriptState = m_scriptState.get();
|
| + ScriptState::Scope scope(scriptState); // will assert context is valid; do not call this method when the context is invalidated
|
| + v8::Isolate* isolate = scriptState->isolate();
|
| +
|
| + v8::Local<v8::Value> stream = m_stream.newLocal(isolate);
|
| + if (stream.IsEmpty())
|
| + return;
|
| +
|
| + v8::Local<v8::Value> args[] = { stream };
|
| + V8ScriptRunner::callExtraOrCrash(scriptState, "CloseReadableStream", args);
|
| +
|
| + m_stream.clear();
|
| + }
|
| +
|
| + double desiredSize() const
|
| + {
|
| + ScriptState* scriptState = m_scriptState.get();
|
| + ScriptState::Scope scope(scriptState); // will assert context is valid; do not call this method when the context is invalidated
|
| + v8::Isolate* isolate = scriptState->isolate();
|
| +
|
| + v8::Local<v8::Value> stream = m_stream.newLocal(isolate);
|
| + if (stream.IsEmpty())
|
| + return 0;
|
| +
|
| + v8::Local<v8::Value> args[] = { stream };
|
| + v8::Local<v8::Value> result = V8ScriptRunner::callExtraOrCrash(scriptState, "GetReadableStreamDesiredSize", args);
|
| +
|
| + return result.As<v8::Number>()->Value();
|
| + }
|
| +
|
| + template <typename ChunkType>
|
| + void enqueue(ChunkType chunk) const
|
| + {
|
| + ScriptState* scriptState = m_scriptState.get();
|
| + ScriptState::Scope scope(scriptState); // will assert context is valid; do not call this method when the context is invalidated
|
| + v8::Isolate* isolate = scriptState->isolate();
|
| +
|
| + v8::Local<v8::Value> stream = m_stream.newLocal(isolate);
|
| + if (stream.IsEmpty())
|
| + return;
|
| +
|
| + v8::Local<v8::Value> jsChunk = toV8(chunk, scriptState);
|
| + v8::Local<v8::Value> args[] = { stream, jsChunk };
|
| + V8ScriptRunner::callExtraOrCrash(scriptState, "EnqueueInReadableStream", args);
|
| + }
|
| +
|
| + template <typename ErrorType>
|
| + void error(ErrorType error)
|
| + {
|
| + ScriptState* scriptState = m_scriptState.get();
|
| + ScriptState::Scope scope(scriptState); // will assert context is valid; do not call this method when the context is invalidated
|
| + v8::Isolate* isolate = scriptState->isolate();
|
| +
|
| + v8::Local<v8::Value> stream = m_stream.newLocal(isolate);
|
| + if (stream.IsEmpty())
|
| + return;
|
| +
|
| + v8::Local<v8::Value> jsError = toV8(error, scriptState);
|
| + v8::Local<v8::Value> args[] = { stream, jsError };
|
| + V8ScriptRunner::callExtraOrCrash(scriptState, "ErrorReadableStream", args);
|
| +
|
| + m_stream.clear();
|
| + }
|
| +
|
| +private:
|
| + static void streamWeakCallback(const v8::WeakCallbackInfo<ScopedPersistent<v8::Value>>& weakInfo)
|
| + {
|
| + weakInfo.GetParameter()->clear();
|
| + }
|
| +
|
| + RefPtr<ScriptState> m_scriptState;
|
| + ScopedPersistent<v8::Value> m_stream;
|
| +};
|
| +
|
| +} // namespace blink
|
| +
|
| +#endif // ReadableStreamController_h
|
|
|