accomodate new optional values

aerospike · Oct 15, 2024 · 4de4232 · 4de4232
1 parent be3410a
commit 4de4232
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 12 deletions.
diff --git a/client.go b/client.go
@@ -163,7 +163,7 @@ func (c *Client) put(
 
 	putReq := &protos.PutRequest{
 		Key:                protoKey,
-		WriteType:          writeType,
+		WriteType:          &writeType,
 		Fields:             fields,
 		IgnoreMemQueueFull: ignoreMemQueueFull,
 	}
@@ -786,7 +786,7 @@ func (c *Client) IndexCreate(
 			Name:      indexName,
 		},
 		Dimensions:           dimensions,
-		VectorDistanceMetric: vectorDistanceMetric,
+		VectorDistanceMetric: &vectorDistanceMetric,
 		Field:                vectorField,
 		SetFilter:            set,
 		Params:               params,

diff --git a/utils.go b/utils.go
@@ -8,6 +8,10 @@ import (
 	"github.com/aerospike/avs-client-go/protos"
 )
 
+func projectionTypePtr(t protos.ProjectionType) *protos.ProjectionType {
+	return &t
+}
+
 func createUserPassCredential(username, password string) *protos.Credentials {
 	return &protos.Credentials{
 		Username: username,
@@ -44,23 +48,23 @@ func createVectorSearchRequest(
 func createProjectionSpec(includeFields, excludeFields []string) *protos.ProjectionSpec {
 	spec := &protos.ProjectionSpec{
 		Include: &protos.ProjectionFilter{
-			Type: protos.ProjectionType_ALL,
+			Type: projectionTypePtr(protos.ProjectionType_ALL),
 		},
 		Exclude: &protos.ProjectionFilter{
-			Type: protos.ProjectionType_NONE,
+			Type: projectionTypePtr(protos.ProjectionType_NONE),
 		},
 	}
 
 	if includeFields != nil {
 		spec.Include = &protos.ProjectionFilter{
-			Type:   protos.ProjectionType_SPECIFIED,
+			Type:   projectionTypePtr(protos.ProjectionType_SPECIFIED),
 			Fields: includeFields,
 		}
 	}
 
 	if excludeFields != nil {
 		spec.Exclude = &protos.ProjectionFilter{
-			Type:   protos.ProjectionType_SPECIFIED,
+			Type:   projectionTypePtr(protos.ProjectionType_SPECIFIED),
 			Fields: excludeFields,
 		}
 	}

diff --git a/utils_test.go b/utils_test.go
@@ -20,11 +20,11 @@ func TestCreateProjectionSpec(t *testing.T) {
 			excludeFields: nil,
 			expectedProjectionSpec: &protos.ProjectionSpec{
 				Include: &protos.ProjectionFilter{
-					Type:   protos.ProjectionType_SPECIFIED,
+					Type:   projectionTypePtr(protos.ProjectionType_SPECIFIED),
 					Fields: []string{"field1", "field2"},
 				},
 				Exclude: &protos.ProjectionFilter{
-					Type: protos.ProjectionType_NONE,
+					Type: projectionTypePtr(protos.ProjectionType_NONE),
 				},
 			},
 		},
@@ -34,10 +34,10 @@ func TestCreateProjectionSpec(t *testing.T) {
 			excludeFields: []string{"field3", "field4"},
 			expectedProjectionSpec: &protos.ProjectionSpec{
 				Include: &protos.ProjectionFilter{
-					Type: protos.ProjectionType_ALL,
+					Type: projectionTypePtr(protos.ProjectionType_ALL),
 				},
 				Exclude: &protos.ProjectionFilter{
-					Type:   protos.ProjectionType_SPECIFIED,
+					Type:   projectionTypePtr(protos.ProjectionType_SPECIFIED),
 					Fields: []string{"field3", "field4"},
 				},
 			},
@@ -48,10 +48,10 @@ func TestCreateProjectionSpec(t *testing.T) {
 			excludeFields: nil,
 			expectedProjectionSpec: &protos.ProjectionSpec{
 				Include: &protos.ProjectionFilter{
-					Type: protos.ProjectionType_ALL,
+					Type: projectionTypePtr(protos.ProjectionType_ALL),
 				},
 				Exclude: &protos.ProjectionFilter{
-					Type: protos.ProjectionType_NONE,
+					Type: projectionTypePtr(protos.ProjectionType_NONE),
 				},
 			},
 		},