diff --git a/client/options.go b/client/options.go index 9490fef4..38675248 100644 --- a/client/options.go +++ b/client/options.go @@ -13,6 +13,7 @@ package client import ( "fmt" + "strconv" "github.com/cockroachdb/errors" @@ -270,6 +271,24 @@ func IsBackup() BulkInsertOption { } } +// IsL0 specifies it is to import L0 segment binlog +func IsL0(isL0 bool) BulkInsertOption { + return func(req *milvuspb.ImportRequest) { + optionMap := entity.KvPairsMap(req.GetOptions()) + optionMap["l0_import"] = strconv.FormatBool(isL0) + req.Options = entity.MapKvPairs(optionMap) + } +} + +// SkipDiskQuotaCheck https://github.com/milvus-io/milvus/pull/35274 +func SkipDiskQuotaCheck(skipDiskQuotaCheck bool) BulkInsertOption { + return func(req *milvuspb.ImportRequest) { + optionMap := entity.KvPairsMap(req.GetOptions()) + optionMap["skip_disk_quota_check"] = strconv.FormatBool(skipDiskQuotaCheck) + req.Options = entity.MapKvPairs(optionMap) + } +} + type getOption struct { partitionNames []string outputFields []string