Add Logical partition num options for create collection (#467)

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>

client/collection.go

@@ -104,6 +104,7 @@ func (c *GrpcClient) CreateCollection(ctx context.Context, collSchema *entity.Sc
 		// default consistency level is strong
 		// to be consistent with previous version
 		ConsistencyLevel: common.ConsistencyLevel_Bounded,
+		NumPartitions:    0,
 	}
 	// apply options on request
 	for _, opt := range opts {

client/options.go

@@ -37,6 +37,13 @@ func WithCollectionProperty(key string, value string) CreateCollectionOption {
 	}
 }
 
+// WithPartitionNum returns a create collection options to set physical partition number when logical partition feature.
+func WithPartitionNum(partitionNums int64) CreateCollectionOption {
+	return func(req *server.CreateCollectionRequest) {
+		req.NumPartitions = partitionNums
+	}
+}
+
 // LoadCollectionOption is an option that is used to modify LoadCollectionRequest
 type LoadCollectionOption func(*server.LoadCollectionRequest)
 

client/options_test.go

@@ -12,6 +12,7 @@
 package client
 
 import (
+	"math/rand"
 	"testing"
 
 	common "github.com/milvus-io/milvus-proto/go-api/commonpb"
@@ -44,6 +45,19 @@ func TestLoadCollectionWithReplicaNumber(t *testing.T) {
 	assert.Equal(t, testMultiReplicaNumber, req.GetReplicaNumber())
 }
 
+func TestCreateCollectionWithPartitionNum(t *testing.T) {
+	partitionNum := rand.Int63n(1000) + 1
+	opt := WithPartitionNum(partitionNum)
+	assert.NotNil(t, opt)
+	req := &server.CreateCollectionRequest{}
+
+	assert.NotPanics(t, func() {
+		opt(req)
+	})
+
+	assert.Equal(t, partitionNum, req.GetNumPartitions())
+}
+
 func TestMakeSearchQueryOption(t *testing.T) {
 	c := &entity.Collection{
 		Name:             "999",