Added EncodeText method to Vector and SparseVector

pgx/sparsevec.go

@@ -46,7 +46,7 @@ type encodePlanSparseVectorCodecText struct{}
 
 func (encodePlanSparseVectorCodecText) Encode(value any, buf []byte) (newBuf []byte, err error) {
 	v := value.(pgvector.SparseVector)
-	return append(buf, v.String()...), nil
+	return v.EncodeText(buf)
 }
 
 func (SparseVectorCodec) PlanScan(m *pgtype.Map, oid uint32, format int16, target any) pgtype.ScanPlan {

pgx/vector.go

@@ -46,7 +46,7 @@ type encodePlanVectorCodecText struct{}
 
 func (encodePlanVectorCodecText) Encode(value any, buf []byte) (newBuf []byte, err error) {
 	v := value.(pgvector.Vector)
-	return append(buf, v.String()...), nil
+	return v.EncodeText(buf)
 }
 
 func (VectorCodec) PlanScan(m *pgtype.Map, oid uint32, format int16, target any) pgtype.ScanPlan {

sparsevec.go

@@ -74,21 +74,9 @@ func (v SparseVector) Slice() []float32 {
 
 // String returns a string representation of the sparse vector.
 func (v SparseVector) String() string {
-	buf := make([]byte, 0, 13+27*len(v.indices))
-	buf = append(buf, '{')
-
-	for i := 0; i < len(v.indices); i++ {
-		if i > 0 {
-			buf = append(buf, ',')
-		}
-		buf = strconv.AppendInt(buf, int64(v.indices[i])+1, 10)
-		buf = append(buf, ':')
-		buf = strconv.AppendFloat(buf, float64(v.values[i]), 'f', -1, 32)
-	}
-
-	buf = append(buf, '}')
-	buf = append(buf, '/')
-	buf = strconv.AppendInt(buf, int64(v.dim), 10)
+	// should never throw an error
+	// but returning an empty string is fine if it does
+	buf, _ := v.EncodeText(nil)
 	return string(buf)
 }
 
@@ -168,6 +156,26 @@ func (v *SparseVector) DecodeBinary(buf []byte) error {
 	return nil
 }
 
+// EncodeText encodes a text representation of the sparse vector.
+func (v SparseVector) EncodeText(buf []byte) (newBuf []byte, err error) {
+	buf = slices.Grow(buf, 13+27*len(v.indices))
+	buf = append(buf, '{')
+
+	for i := 0; i < len(v.indices); i++ {
+		if i > 0 {
+			buf = append(buf, ',')
+		}
+		buf = strconv.AppendInt(buf, int64(v.indices[i])+1, 10)
+		buf = append(buf, ':')
+		buf = strconv.AppendFloat(buf, float64(v.values[i]), 'f', -1, 32)
+	}
+
+	buf = append(buf, '}')
+	buf = append(buf, '/')
+	buf = strconv.AppendInt(buf, int64(v.dim), 10)
+	return buf, nil
+}
+
 // statically assert that SparseVector implements sql.Scanner.
 var _ sql.Scanner = (*SparseVector)(nil)
 

vector.go

@@ -29,17 +29,9 @@ func (v Vector) Slice() []float32 {
 
 // String returns a string representation of the vector.
 func (v Vector) String() string {
-	buf := make([]byte, 0, 2+16*len(v.vec))
-	buf = append(buf, '[')
-
-	for i := 0; i < len(v.vec); i++ {
-		if i > 0 {
-			buf = append(buf, ',')
-		}
-		buf = strconv.AppendFloat(buf, float64(v.vec[i]), 'f', -1, 32)
-	}
-
-	buf = append(buf, ']')
+	// should never throw an error
+	// but returning an empty string is fine if it does
+	buf, _ := v.EncodeText(nil)
 	return string(buf)
 }
 
@@ -86,6 +78,20 @@ func (v *Vector) DecodeBinary(buf []byte) error {
 	return nil
 }
 
+// EncodeText encodes a text representation of the vector.
+func (v Vector) EncodeText(buf []byte) (newBuf []byte, err error) {
+	buf = slices.Grow(buf, 2+16*len(v.vec))
+	buf = append(buf, '[')
+	for i := 0; i < len(v.vec); i++ {
+		if i > 0 {
+			buf = append(buf, ',')
+		}
+		buf = strconv.AppendFloat(buf, float64(v.vec[i]), 'f', -1, 32)
+	}
+	buf = append(buf, ']')
+	return buf, nil
+}
+
 // statically assert that Vector implements sql.Scanner.
 var _ sql.Scanner = (*Vector)(nil)