feat: VectorString helpers to build vector search queries

Author: rueian

binary.go

@@ -1,7 +1,9 @@
 package rueidis
 
 import (
+	"encoding/binary"
 	"encoding/json"
+	"math"
 	"unsafe"
 )
 
@@ -16,6 +18,54 @@ func BinaryString(bs []byte) string {
 	return unsafe.String(unsafe.SliceData(bs), len(bs))
 }
 
+// VectorString32 convert the provided []float32 into a string. Users can use this to build vector search queries:
+//
+//	client.B().FtSearch().Index("idx").Query("*=>[KNN 5 @vec $V]").
+//	    Params().Nargs(2).NameValue().NameValue("V", rueidis.VectorString32([]float32{1})).
+//	    Dialect(2).Build()
+func VectorString32(v []float32) string {
+	b := make([]byte, len(v)*4)
+	for i, e := range v {
+		i := i * 4
+		binary.LittleEndian.PutUint32(b[i:i+4], math.Float32bits(e))
+	}
+	return BinaryString(b)
+}
+
+// ToVector32 reverts VectorString32. User can use this to convert redis response back to []float32.
+func ToVector32(s string) []float32 {
+	bs := unsafe.Slice(unsafe.StringData(s), len(s))
+	vs := make([]float32, 0, len(bs)/4)
+	for i := 0; i < len(bs); i += 4 {
+		vs = append(vs, math.Float32frombits(binary.LittleEndian.Uint32(bs[i:i+4])))
+	}
+	return vs
+}
+
+// VectorString64 convert the provided []float64 into a string. Users can use this to build vector search queries:
+//
+//	client.B().FtSearch().Index("idx").Query("*=>[KNN 5 @vec $V]").
+//	    Params().Nargs(2).NameValue().NameValue("V", rueidis.VectorString64([]float64{1})).
+//	    Dialect(2).Build()
+func VectorString64(v []float64) string {
+	b := make([]byte, len(v)*8)
+	for i, e := range v {
+		i := i * 8
+		binary.LittleEndian.PutUint64(b[i:i+8], math.Float64bits(e))
+	}
+	return BinaryString(b)
+}
+
+// ToVector64 reverts VectorString64. User can use this to convert redis response back to []float64.
+func ToVector64(s string) []float64 {
+	bs := unsafe.Slice(unsafe.StringData(s), len(s))
+	vs := make([]float64, 0, len(bs)/8)
+	for i := 0; i < len(bs); i += 8 {
+		vs = append(vs, math.Float64frombits(binary.LittleEndian.Uint64(bs[i:i+8])))
+	}
+	return vs
+}
+
 // JSON convert the provided parameter into a JSON string. Users can use this JSON helper to work with RedisJSON commands.
 // For example:
 //

binary_test.go

@@ -2,6 +2,7 @@ package rueidis
 
 import (
 	"encoding/json"
+	"reflect"
 	"strings"
 	"testing"
 )
@@ -29,6 +30,32 @@ func TestJSONPanic(t *testing.T) {
 	JSON(a)
 }
 
+func TestVectorString32(t *testing.T) {
+	for _, test := range [][]float32{
+		{},
+		{0, 0, 0, 0},
+		{9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9},
+		{-1, -1, -1, -1, -1, -1, -1, -1, -1, -1},
+	} {
+		if !reflect.DeepEqual(test, ToVector32(VectorString32(test))) {
+			t.Fatalf("fail to convert %v", test)
+		}
+	}
+}
+
+func TestVectorString64(t *testing.T) {
+	for _, test := range [][]float64{
+		{},
+		{0, 0, 0, 0},
+		{9, 9, 9, 9, 9, 9, 9, 9, 9, 9, 9},
+		{-1, -1, -1, -1, -1, -1, -1, -1, -1, -1},
+	} {
+		if !reflect.DeepEqual(test, ToVector64(VectorString64(test))) {
+			t.Fatalf("fail to convert %v", test)
+		}
+	}
+}
+
 type recursive struct {
 	R *recursive
 }