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 |