ByteStreamProducer-handle-string-type

Signed-off-by: Youyuan Wu <[email protected]>
go-openapi · Apr 3, 2021 · ac4d104 · ac4d104
1 parent 6da6fd0
commit ac4d104
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/bytestream.go b/bytestream.go
@@ -129,6 +129,11 @@ func ByteStreamProducer(opts ...byteStreamOpt) Producer {
 		}
 
 		if data != nil {
+			if str, ok := data.(string); ok {
+				_, err := writer.Write([]byte(str))
+				return err
+			}
+
 			if e, ok := data.(error); ok {
 				_, err := writer.Write([]byte(e.Error()))
 				return err

diff --git a/bytestream_test.go b/bytestream_test.go
@@ -84,6 +84,12 @@ func TestByteStreamProducer(t *testing.T) {
 		rdr.Reset()
 	}
 
+	// string can also be used to produce
+	if assert.NoError(t, cons.Produce(&rdr, expected)) {
+		assert.Equal(t, expected, rdr.String())
+		rdr.Reset()
+	}
+
 	// binary slices can also be used to produce
 	if assert.NoError(t, cons.Produce(&rdr, []byte(expected))) {
 		assert.Equal(t, expected, rdr.String())