Decode(Text|Binary) now accepts []byte instead of io.Reader

pgtype/textarray.go

@@ -2,6 +2,7 @@ package pgtype
 
 import (
 	"bytes"
+	"encoding/binary"
 	"fmt"
 	"io"
 
@@ -73,29 +74,17 @@ func (src *TextArray) AssignTo(dst interface{}) error {
 	return nil
 }
 
-func (dst *TextArray) DecodeText(r io.Reader) error {
-	size, err := pgio.ReadInt32(r)
-	if err != nil {
-		return err
-	}
-
-	if size == -1 {
+func (dst *TextArray) DecodeText(src []byte) error {
+	if src == nil {
 		*dst = TextArray{Status: Null}
 		return nil
 	}
 
-	buf := make([]byte, int(size))
-	_, err = io.ReadFull(r, buf)
+	uta, err := ParseUntypedTextArray(string(src))
 	if err != nil {
 		return err
 	}
 
-	uta, err := ParseUntypedTextArray(string(buf))
-	if err != nil {
-		return err
-	}
-
-	textElementReader := NewTextElementReader(r)
 	var elements []Text
 
 	if len(uta.Elements) > 0 {
@@ -103,8 +92,11 @@ func (dst *TextArray) DecodeText(r io.Reader) error {
 
 		for i, s := range uta.Elements {
 			var elem Text
-			textElementReader.Reset(s)
-			err = elem.DecodeText(textElementReader)
+			var elemSrc []byte
+			if s != "NULL" {
+				elemSrc = []byte(s)
+			}
+			err = elem.DecodeText(elemSrc)
 			if err != nil {
 				return err
 			}
@@ -118,19 +110,14 @@ func (dst *TextArray) DecodeText(r io.Reader) error {
 	return nil
 }
 
-func (dst *TextArray) DecodeBinary(r io.Reader) error {
-	size, err := pgio.ReadInt32(r)
-	if err != nil {
-		return err
-	}
-
-	if size == -1 {
+func (dst *TextArray) DecodeBinary(src []byte) error {
+	if src == nil {
 		*dst = TextArray{Status: Null}
 		return nil
 	}
 
 	var arrayHeader ArrayHeader
-	err = arrayHeader.DecodeBinary(r)
+	rp, err := arrayHeader.DecodeBinary(src)
 	if err != nil {
 		return err
 	}
@@ -148,7 +135,14 @@ func (dst *TextArray) DecodeBinary(r io.Reader) error {
 	elements := make([]Text, elementCount)
 
 	for i := range elements {
-		err = elements[i].DecodeBinary(r)
+		elemLen := int(int32(binary.BigEndian.Uint32(src[rp:])))
+		rp += 4
+		var elemSrc []byte
+		if elemLen >= 0 {
+			elemSrc = src[rp : rp+elemLen]
+			rp += elemLen
+		}
+		err = elements[i].DecodeBinary(elemSrc)
 		if err != nil {
 			return err
 		}
@@ -211,7 +205,12 @@ func (src *TextArray) EncodeText(w io.Writer) error {
 		}
 
 		textElementWriter.Reset()
-		if elem.String == "" && elem.Status == Present {
+		if elem.Status == Null {
+			_, err := io.WriteString(buf, `"NULL"`)
+			if err != nil {
+				return err
+			}
+		} else if elem.String == "" {
 			_, err := io.WriteString(buf, `""`)
 			if err != nil {
 				return err