Support range search (#519)

Signed-off-by: lixinguo <[email protected]>
Co-authored-by: lixinguo <[email protected]>

entity/index.go

@@ -67,6 +67,8 @@ type Index interface {
 type SearchParam interface {
 	// returns parameters for search/query
 	Params() map[string]interface{}
+	AddRadius(radius float64)
+	AddRangeFilter(rangeFilter float64)
 }
 
 type baseSearchParams struct {
@@ -81,6 +83,14 @@ func (sp *baseSearchParams) Params() map[string]interface{} {
 	return params
 }
 
+func (sp *baseSearchParams) AddRadius(radius float64) {
+	sp.params["radius"] = radius
+}
+
+func (sp *baseSearchParams) AddRangeFilter(rangeFilter float64) {
+	sp.params["range_filter"] = rangeFilter
+}
+
 func newBaseSearchParams() baseSearchParams {
 	return baseSearchParams{
 		params: make(map[string]interface{}),

entity/index_test.go

@@ -29,3 +29,15 @@ func TestGenericIndex(t *testing.T) {
 	assert.Equal(t, name, gi.Name())
 	assert.EqualValues(t, IvfFlat, gi.Params()[tIndexType])
 }
+
+func TestAddRadius(t *testing.T) {
+	params := newBaseSearchParams()
+	params.AddRadius(10)
+	assert.Equal(t, params.Params()["radius"], float64(10))
+}
+
+func TestAddRangeFilter(t *testing.T) {
+	params := newBaseSearchParams()
+	params.AddRangeFilter(20)
+	assert.Equal(t, params.Params()["range_filter"], float64(20))
+}