From 0cec1b9448671ccc368f761304e295932018efcc Mon Sep 17 00:00:00 2001 From: groot Date: Fri, 22 Apr 2022 18:43:43 +0800 Subject: [PATCH] Fix bulkload bug (#16591) Signed-off-by: groot --- internal/datanode/data_node.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/internal/datanode/data_node.go b/internal/datanode/data_node.go index d2d8ffdc29..89976b689d 100644 --- a/internal/datanode/data_node.go +++ b/internal/datanode/data_node.go @@ -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 {