Add list contain function

Author: adibrastegarnia

pkg/client/list/list.go

@@ -18,6 +18,7 @@ import (
 	"context"
 	"encoding/base64"
 	"errors"
+
 	"github.com/atomix/api/proto/atomix/headers"
 	api "github.com/atomix/api/proto/atomix/list"
 	"github.com/atomix/go-client/pkg/client/primitive"
@@ -58,6 +59,9 @@ type List interface {
 	// Len gets the length of the list
 	Len(ctx context.Context) (int, error)
 
+	// Contains checks whether the list contains a value
+	Contains(ctx context.Context, value []byte) (bool, error)
+
 	// Slice returns a slice of the list from the given start index to the given end index
 	Slice(ctx context.Context, from int, to int) (List, error)
 
@@ -258,6 +262,25 @@ func (l *list) Remove(ctx context.Context, index int) ([]byte, error) {
 	}
 }
 
+func (l *list) Contains(ctx context.Context, value []byte) (bool, error) {
+	response, err := l.instance.DoQuery(ctx, func(ctx context.Context, conn *grpc.ClientConn, header *headers.RequestHeader) (*headers.ResponseHeader, interface{}, error) {
+		client := api.NewListServiceClient(conn)
+		request := &api.ContainsRequest{
+			Header: header,
+			Value:  base64.StdEncoding.EncodeToString(value),
+		}
+		response, err := client.Contains(ctx, request)
+		if err != nil {
+			return nil, nil, err
+		}
+		return response.Header, response, nil
+	})
+	if err != nil {
+		return false, err
+	}
+	return bool(response.(*api.ContainsResponse).Contains), nil
+}
+
 func (l *list) Len(ctx context.Context) (int, error) {
 	response, err := l.instance.DoQuery(ctx, func(ctx context.Context, conn *grpc.ClientConn, header *headers.RequestHeader) (*headers.ResponseHeader, interface{}, error) {
 		client := api.NewListServiceClient(conn)

pkg/client/list/list_test.go

@@ -16,10 +16,11 @@ package list
 
 import (
 	"context"
+	"testing"
+
 	"github.com/atomix/go-client/pkg/client/primitive"
 	"github.com/atomix/go-client/pkg/client/test"
 	"github.com/stretchr/testify/assert"
-	"testing"
 )
 
 func TestListOperations(t *testing.T) {
@@ -53,6 +54,10 @@ func TestListOperations(t *testing.T) {
 	assert.NoError(t, err)
 	assert.Equal(t, "foo", string(value))
 
+	contain, err := list.Contains(context.TODO(), []byte("foo"))
+	assert.NoError(t, err)
+	assert.Equal(t, true, contain)
+
 	err = list.Append(context.TODO(), []byte("bar"))
 	assert.NoError(t, err)
 

pkg/client/list/slice.go

@@ -17,6 +17,7 @@ package list
 import (
 	"context"
 	"errors"
+
 	"github.com/atomix/go-client/pkg/client/primitive"
 )
 
@@ -96,6 +97,11 @@ func (l *slicedList) Len(ctx context.Context) (int, error) {
 	return size, nil
 }
 
+func (l *slicedList) Contains(ctx context.Context, value []byte) (bool, error) {
+	contain, err := l.list.Contains(ctx, value)
+	return contain, err
+}
+
 func (l *slicedList) Slice(ctx context.Context, from int, to int) (List, error) {
 	if l.from != nil {
 		from += *l.from