Index: Source/core/streams/ReadableStreamController.h |
diff --git a/Source/core/streams/ReadableStreamController.h b/Source/core/streams/ReadableStreamController.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..792241f923dea69a816fcc9c9535b9dfa16c26d2 |
--- /dev/null |
+++ b/Source/core/streams/ReadableStreamController.h |
@@ -0,0 +1,134 @@ |
+// Copyright 2015 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 "core/CoreExport.h" |
+#include "platform/heap/Handle.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() |
+ { |
+ if (m_stream.isEmpty()) |
+ return; |
+ |
+ ScriptState::Scope scope(m_scriptState); |
+ |
+ // TODO: should we be storing these somewhere so that we don't need to get the v8::Local<v8::Function> every |
+ // time? |
+ auto func = m_scriptState->getFromExtrasExports("CloseReadableStream").v8Value().As<v8::Function>(); |
+ |
+ auto isolate = m_scriptState->isolate(); |
+ auto context = m_scriptState->context(); |
+ auto undefined = v8::Undefined(isolate); |
+ v8::Local<v8::Value> args[] = { m_stream.newLocal(isolate) }; |
+ auto result = func->Call(context, undefined, arraysize(args), args); |
+ |
+ ASSERT_UNUSED(result, !result.IsEmpty()); |
+ m_stream.clear(); |
+ } |
+ |
+ double desiredSize() const |
+ { |
+ if (m_stream.isEmpty()) |
+ return 0; |
+ |
+ ScriptState::Scope scope(m_scriptState); |
+ |
+ auto func = m_scriptState->getFromExtrasExports("GetReadableStreamDesiredSize").v8Value().As<v8::Function>(); |
+ |
+ auto isolate = m_scriptState->isolate(); |
+ auto context = m_scriptState->context(); |
+ auto undefined = v8::Undefined(isolate); |
+ v8::Local<v8::Value> args[] = { m_stream.newLocal(isolate) }; |
+ auto result = func->Call(context, undefined, arraysize(args), args).ToLocalChecked(); |
+ |
+ return result.As<v8::Number>()->Value(); |
+ } |
+ |
+ template<typename ChunkType> |
+ void enqueue(ChunkType chunk) const |
+ { |
+ if (m_stream.isEmpty()) |
+ return; |
+ |
+ ScriptState::Scope scope(m_scriptState); |
+ |
+ auto func = m_scriptState->getFromExtrasExports("EnqueueInReadableStream").v8Value().As<v8::Function>(); |
+ |
+ auto isolate = m_scriptState->isolate(); |
+ auto context = m_scriptState->context(); |
+ auto undefined = v8::Undefined(isolate); |
+ auto chunkForV8 = toV8(chunk, context->Global(), isolate); |
+ v8::Local<v8::Value> args[] = { m_stream.newLocal(isolate), chunkForV8 }; |
+ |
+ auto result = func->Call(context, undefined, arraysize(args), args); |
+ |
+ ASSERT_UNUSED(result, !result.IsEmpty()); |
+ } |
+ |
+ template<typename ErrorType> |
+ void error(ErrorType e) |
+ { |
+ if (m_stream.isEmpty()) |
+ return; |
+ |
+ ScriptState::Scope scope(m_scriptState); |
+ |
+ auto func = m_scriptState->getFromExtrasExports("ErrorReadableStream").v8Value().As<v8::Function>(); |
+ |
+ auto isolate = m_scriptState->isolate(); |
+ auto context = m_scriptState->context(); |
+ auto undefined = v8::Undefined(isolate); |
+ auto errorForV8 = toV8(e, context->Global(), isolate); |
+ v8::Local<v8::Value> args[] = { m_stream.newLocal(isolate), errorForV8 }; |
+ auto result = func->Call(context, undefined, arraysize(args), args); |
+ |
+ ASSERT_UNUSED(result, !result.IsEmpty()); |
+ m_stream.clear(); |
+ } |
+ |
+private: |
+ static void streamWeakCallback(const v8::WeakCallbackInfo<ScopedPersistent<v8::Value>>& weakInfo) |
+ { |
+ weakInfo.GetParameter()->clear(); |
+ } |
+ |
+ ScopedPersistent<v8::Value> m_stream; |
+ ScriptState* m_scriptState; |
+}; |
+ |
+} // namespace blink |
+ |
+#endif // ReadableStreamController_h |