| #ifndef SRC_STREAM_PIPE_H_ |
| #define SRC_STREAM_PIPE_H_ |
| |
| #if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS |
| |
| #include "stream_base.h" |
| #include "allocated_buffer.h" |
| |
| namespace node { |
| |
| class StreamPipe : public AsyncWrap { |
| public: |
| StreamPipe(StreamBase* source, StreamBase* sink, v8::Local<v8::Object> obj); |
| ~StreamPipe() override; |
| |
| void Unpipe(bool is_in_deletion = false); |
| |
| static void New(const v8::FunctionCallbackInfo<v8::Value>& args); |
| static void Start(const v8::FunctionCallbackInfo<v8::Value>& args); |
| static void Unpipe(const v8::FunctionCallbackInfo<v8::Value>& args); |
| static void IsClosed(const v8::FunctionCallbackInfo<v8::Value>& args); |
| static void PendingWrites(const v8::FunctionCallbackInfo<v8::Value>& args); |
| |
| SET_NO_MEMORY_INFO() |
| SET_MEMORY_INFO_NAME(StreamPipe) |
| SET_SELF_SIZE(StreamPipe) |
| |
| private: |
| inline StreamBase* source(); |
| inline StreamBase* sink(); |
| |
| int pending_writes_ = 0; |
| bool is_reading_ = false; |
| bool is_eof_ = false; |
| bool is_closed_ = true; |
| bool sink_destroyed_ = false; |
| bool source_destroyed_ = false; |
| bool uses_wants_write_ = false; |
| |
| // Set a default value so that when we’re coming from Start(), we know |
| // that we don’t want to read just yet. |
| // This will likely need to be changed when supporting streams without |
| // `OnStreamWantsWrite()` support. |
| size_t wanted_data_ = 0; |
| |
| void ProcessData(size_t nread, AllocatedBuffer&& buf); |
| |
| class ReadableListener : public StreamListener { |
| public: |
| uv_buf_t OnStreamAlloc(size_t suggested_size) override; |
| void OnStreamRead(ssize_t nread, const uv_buf_t& buf) override; |
| void OnStreamDestroy() override; |
| }; |
| |
| class WritableListener : public StreamListener { |
| public: |
| uv_buf_t OnStreamAlloc(size_t suggested_size) override; |
| void OnStreamRead(ssize_t nread, const uv_buf_t& buf) override; |
| void OnStreamAfterWrite(WriteWrap* w, int status) override; |
| void OnStreamAfterShutdown(ShutdownWrap* w, int status) override; |
| void OnStreamWantsWrite(size_t suggested_size) override; |
| void OnStreamDestroy() override; |
| }; |
| |
| ReadableListener readable_listener_; |
| WritableListener writable_listener_; |
| }; |
| |
| } // namespace node |
| |
| #endif |
| |
| #endif // SRC_STREAM_PIPE_H_ |