From b509ed03136e0b94c351d90cc4a195d4be71d544 Mon Sep 17 00:00:00 2001
From: daviszhen <60595215+daviszhen@users.noreply.github.com>
Date: Tue, 19 Oct 2021 21:16:07 +0800
Subject: [PATCH] Drain event channel first (#967)
---
pkg/frontend/load.go | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/pkg/frontend/load.go b/pkg/frontend/load.go
index 7abfcc329..7c8d28033 100644
--- a/pkg/frontend/load.go
+++ b/pkg/frontend/load.go
@@ -1472,16 +1472,13 @@ func (mce *MysqlCmdExecutor) LoadLoop(load *tree.Load, dbHandler engine.Database
wg.Wait()
- //wait write to quit
- handler.simdCsvWaitWriteRoutineToQuit.Wait()
-
/*
- drain event channel
- */
+ drain event channel
+ */
quit := false
for {
select {
- case ne = <- handler.simdCsvNotiyEventChan:
+ case ne = <- handler.simdCsvNotiyEventChan:
default:
quit = true
}
@@ -1498,6 +1495,9 @@ func (mce *MysqlCmdExecutor) LoadLoop(load *tree.Load, dbHandler engine.Database
}
}
+ //wait write to quit
+ handler.simdCsvWaitWriteRoutineToQuit.Wait()
+
//fmt.Printf("-----total row2col %s fillBlank %s toStorage %s\n",
// handler.row2col,handler.fillBlank,handler.toStorage)
//fmt.Printf("-----write batch %s reset batch %s\n",
--
GitLab