streamWriter

Author: Michael Lauer

streams.go

@@ -6,6 +6,7 @@ import (
 	"sync"
 )
 
+// streamReader is really sort of a piper. Maybe
 type streamReader struct {
 	buffer  bytes.Buffer
 	lock    sync.Mutex
@@ -50,3 +51,35 @@ func (sr *streamReader) Close() error {
 	sr.gotData.Signal()
 	return nil
 }
+
+// streamWriter writes data as FCGI records.
+type streamWriter struct {
+	w    io.Writer
+	tp   recordType
+	id   requestId
+	lock sync.Mutex
+}
+
+func newStreamWriter(w io.Writer, tp recordType, id requestId) *streamWriter {
+	return &streamWriter{w: w, tp: tp, id: id}
+}
+
+func (sw *streamWriter) Write(data []byte) (int, error) {
+	sw.lock.Lock()
+	defer sw.lock.Unlock()
+	if len(data) == 0 {
+		return 0, nil
+	}
+	rec := record{sw.tp, sw.id, data}
+	err := writeRecord(sw.w, rec)
+	// How much did we actually write? Just say nothing if we got an error.
+	if err != nil {
+		return 0, err
+	}
+	return len(data), nil
+}
+
+func (sw *streamWriter) Close() error {
+	// Close means writing an empty string
+	return writeRecord(sw.w, record{sw.tp, sw.id, nil})
+}

streams_test.go

@@ -1,6 +1,7 @@
 package gofcgisrv
 
 import (
+	"bytes"
 	"io"
 	"io/ioutil"
 	"strings"
@@ -49,3 +50,22 @@ func TestStreamReader(t *testing.T) {
 		t.Errorf("Read %s with error %v", read, err)
 	}
 }
+
+func TestStreamWriter(t *testing.T) {
+	var buffer bytes.Buffer
+	sw := newStreamWriter(&buffer, fcgiStdout, 3)
+	io.WriteString(sw, "Foo!")
+	io.WriteString(sw, "This is data")
+	io.WriteString(sw, "\000\001abc")
+	sw.Close()
+
+	expected := "\001\006\000\003\000\004\004\000Foo!\000\000\000\000" +
+		"\001\006\000\003\000\x0c\004\000This is data\000\000\000\000" +
+		"\001\006\000\003\000\x05\003\000\000\001abc\000\000\000" +
+		"\001\006\000\003\000\x00\000\000"
+
+	str := string(buffer.Bytes())
+	if str != expected {
+		t.Errorf("Got\n%q\nnot\n%q", str, expected)
+	}
+}