milvus/tests/go_client/testcases/helper/collection_helper.go

package helper

import (
	"github.com/milvus-io/milvus/client/v2/entity"
)

type CreateCollectionParams struct {
	CollectionFieldsType CollectionFieldsType // collection fields type
}

func NewCreateCollectionParams(collectionFieldsType CollectionFieldsType) *CreateCollectionParams {
	return &CreateCollectionParams{
		CollectionFieldsType: collectionFieldsType,
	}
}

type CreateCollectionOpt func(opt *createCollectionOpt)

type createCollectionOpt struct {
	shardNum             int32
	enabledDynamicSchema bool

	consistencyLevel *entity.ConsistencyLevel
	numPartitions    int64
	properties       map[string]any
}

func TWithShardNum(shardNum int32) CreateCollectionOpt {
	return func(opt *createCollectionOpt) {
		opt.shardNum = shardNum
	}
}

func TWithProperties(properties map[string]any) CreateCollectionOpt {
	return func(opt *createCollectionOpt) {
		opt.properties = properties
	}
}

func TWithConsistencyLevel(consistencyLevel entity.ConsistencyLevel) CreateCollectionOpt {
	return func(opt *createCollectionOpt) {
		opt.consistencyLevel = &consistencyLevel
	}
}

func TWithNullablePartitions(numPartitions int64) CreateCollectionOpt {
	return func(opt *createCollectionOpt) {
		opt.numPartitions = numPartitions
	}
}