Fix bulkload bug (#16591)

Signed-off-by: groot <yihua.mo@zilliz.com>
This commit is contained in:
groot 2022-04-22 18:43:43 +08:00 committed by GitHub
parent 3f5733b738
commit 0cec1b9448
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -850,7 +850,8 @@ func (node *DataNode) Import(ctx context.Context, req *datapb.ImportTaskRequest)
_ = idAllocator.Start()
defer idAllocator.Close()
importWrapper := importutil.NewImportWrapper(ctx, schema, 2, Params.DataNodeCfg.FlushInsertBufferSize, idAllocator, node.chunkManager,
segmentSize := int64(Params.DataCoordCfg.SegmentMaxSize) * 1024 * 1024
importWrapper := importutil.NewImportWrapper(ctx, schema, 2, segmentSize, idAllocator, node.chunkManager,
importFlushReqFunc(node, req, importResult, schema, ts), importResult, reportFunc)
err = importWrapper.Import(req.GetImportTask().GetFiles(), req.GetImportTask().GetRowBased(), false)
if err != nil {