Drain event channel first (#967)

matrixorigin · Oct 19, 2021 · b509ed0 · b509ed0
1 parent 6ff0fb6
commit b509ed0
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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",