Add copy functions for read or write only streams

Signed-off-by: Derek McGowan <derek@mcgstyle.net> (github: dmcgowan)

Author: dmcgowan

inmem.go

@@ -300,6 +300,10 @@ func (w *pipeSender) copyValue(v interface{}) (interface{}, error) {
 		}
 	case io.ReadWriteCloser:
 		return w.copyByteStream(val)
+	case io.ReadCloser:
+		return w.copyByteReadStream(val)
+	case io.WriteCloser:
+		return w.copyByteWriteStream(val)
 	case Sender:
 		return w.copySender(val)
 	case Receiver:
@@ -355,6 +359,31 @@ func (w *pipeSender) copyByteStream(stream io.ReadWriteCloser) (io.ReadWriteClos
 	return streamCopy, nil
 }
 
+func (w *pipeSender) copyByteReadStream(stream io.ReadCloser) (io.ReadCloser, error) {
+	streamCopy, err := w.session.newByteStream()
+	if err != nil {
+		return nil, err
+	}
+	go func() {
+		io.Copy(streamCopy, stream)
+		streamCopy.Close()
+		stream.Close()
+	}()
+	return streamCopy, nil
+}
+
+func (w *pipeSender) copyByteWriteStream(stream io.WriteCloser) (io.WriteCloser, error) {
+	streamCopy, err := w.session.newByteStream()
+	if err != nil {
+		return nil, err
+	}
+	go func() {
+		io.Copy(stream, streamCopy)
+		stream.Close()
+	}()
+	return streamCopy, nil
+}
+
 func (w *pipeSender) copyChannelMessage(m map[string]interface{}) (interface{}, error) {
 	mCopy := make(map[string]interface{})
 	for k, v := range m {

spdy/copy.go

@@ -38,6 +38,10 @@ func (c *channel) copyValue(v interface{}) (interface{}, error) {
 		// Do nothing until socket support is added
 	case io.ReadWriteCloser:
 		return c.copyByteStream(val)
+	case io.ReadCloser:
+		return c.copyByteReadStream(val)
+	case io.WriteCloser:
+		return c.copyByteWriteStream(val)
 	case libchan.Sender:
 		return c.copySender(val)
 	case libchan.Receiver:
@@ -93,6 +97,31 @@ func (c *channel) copyByteStream(stream io.ReadWriteCloser) (io.ReadWriteCloser,
 	return streamCopy, nil
 }
 
+func (c *channel) copyByteReadStream(stream io.ReadCloser) (io.ReadCloser, error) {
+	streamCopy, err := c.session.createByteStream()
+	if err != nil {
+		return nil, err
+	}
+	go func() {
+		io.Copy(streamCopy, stream)
+		streamCopy.Close()
+		stream.Close()
+	}()
+	return streamCopy, nil
+}
+
+func (c *channel) copyByteWriteStream(stream io.WriteCloser) (io.WriteCloser, error) {
+	streamCopy, err := c.session.createByteStream()
+	if err != nil {
+		return nil, err
+	}
+	go func() {
+		io.Copy(stream, streamCopy)
+		stream.Close()
+	}()
+	return streamCopy, nil
+}
+
 func (c *channel) copyChannelMessage(m map[string]interface{}) (interface{}, error) {
 	mCopy := make(map[string]interface{})
 	for k, v := range m {