This is an automated cherry-pick of pingcap#11011

Signed-off-by: ti-chi-bot <[email protected]>
ti-chi-bot · May 14, 2024 · 9c223e9 · 9c223e9
1 parent 7bd0eb4
commit 9c223e9
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 13 deletions.
diff --git a/cdc/redo/writer/memory/encoding_worker.go b/cdc/redo/writer/memory/encoding_worker.go
@@ -104,7 +104,7 @@ type encodingWorkerGroup struct {
 	workerNum  int
 	nextWorker atomic.Uint64
 
-	closed chan struct{}
+	closed chan error
 }
 
 func newEncodingWorkerGroup(cfg *writer.LogWriterConfig) *encodingWorkerGroup {
@@ -121,19 +121,20 @@ func newEncodingWorkerGroup(cfg *writer.LogWriterConfig) *encodingWorkerGroup {
 		inputChs:   inputChs,
 		outputCh:   make(chan *polymorphicRedoEvent, redo.DefaultEncodingOutputChanSize),
 		workerNum:  workerNum,
-		closed:     make(chan struct{}),
+		closed:     make(chan error, 1),
 	}
 }
 
 func (e *encodingWorkerGroup) Run(ctx context.Context) (err error) {
 	defer func() {
-		close(e.closed)
+		log.Warn("redo encoding workers closed",
+			zap.String("namespace", e.changefeed.Namespace),
+			zap.String("changefeed", e.changefeed.ID),
+			zap.Error(err))
 		if err != nil && errors.Cause(err) != context.Canceled {
-			log.Warn("redo fileWorkerGroup closed with error",
-				zap.String("namespace", e.changefeed.Namespace),
-				zap.String("changefeed", e.changefeed.ID),
-				zap.Error(err))
+			e.closed <- err
 		}
+		close(e.closed)
 	}()
 	eg, egCtx := errgroup.WithContext(ctx)
 	for i := 0; i < e.workerNum; i++ {
@@ -183,8 +184,13 @@ func (e *encodingWorkerGroup) input(
 	select {
 	case <-ctx.Done():
 		return ctx.Err()
+<<<<<<< HEAD
 	case <-e.closed:
 		return errors.ErrRedoWriterStopped.GenWithStackByArgs("encoding worker is closed")
+=======
+	case err := <-e.closed:
+		return errors.WrapError(errors.ErrRedoWriterStopped, err, "encoding worker is closed")
+>>>>>>> 52b4301388 (redo(ticdc): return internal error in redo writer (#11011))
 	case e.inputChs[idx] <- event:
 		return nil
 	}
@@ -196,8 +202,13 @@ func (e *encodingWorkerGroup) output(
 	select {
 	case <-ctx.Done():
 		return ctx.Err()
+<<<<<<< HEAD
 	case <-e.closed:
 		return errors.ErrRedoWriterStopped.GenWithStackByArgs("encoding worker is closed")
+=======
+	case err := <-e.closed:
+		return errors.WrapError(errors.ErrRedoWriterStopped, err, "encoding worker is closed")
+>>>>>>> 52b4301388 (redo(ticdc): return internal error in redo writer (#11011))
 	case e.outputCh <- event:
 		return nil
 	}
@@ -221,8 +232,13 @@ func (e *encodingWorkerGroup) FlushAll(ctx context.Context) error {
 	select {
 	case <-ctx.Done():
 		return ctx.Err()
+<<<<<<< HEAD
 	case <-e.closed:
 		return errors.ErrRedoWriterStopped.GenWithStackByArgs("encoding worker is closed")
+=======
+	case err := <-e.closed:
+		return errors.WrapError(errors.ErrRedoWriterStopped, err, "encoding worker is closed")
+>>>>>>> 52b4301388 (redo(ticdc): return internal error in redo writer (#11011))
 	case <-flushCh:
 	}
 	return nil
@@ -245,8 +261,13 @@ func (e *encodingWorkerGroup) broadcastAndWaitEncoding(ctx context.Context) erro
 		select {
 		case <-ctx.Done():
 			return ctx.Err()
+<<<<<<< HEAD
 		case <-e.closed:
 			return errors.ErrRedoWriterStopped.GenWithStackByArgs("encoding worker is closed")
+=======
+		case err := <-e.closed:
+			return errors.WrapError(errors.ErrRedoWriterStopped, err, "encoding worker is closed")
+>>>>>>> 52b4301388 (redo(ticdc): return internal error in redo writer (#11011))
 		case <-ch:
 		}
 	}

diff --git a/cdc/redo/writer/memory/file_worker.go b/cdc/redo/writer/memory/file_worker.go
@@ -155,12 +155,10 @@ func (f *fileWorkerGroup) Run(
 ) (err error) {
 	defer func() {
 		f.close()
-		if err != nil && errors.Cause(err) != context.Canceled {
-			log.Warn("redo file workers closed with error",
-				zap.String("namespace", f.cfg.ChangeFeedID.Namespace),
-				zap.String("changefeed", f.cfg.ChangeFeedID.ID),
-				zap.Error(err))
-		}
+		log.Warn("redo file workers closed",
+			zap.String("namespace", f.cfg.ChangeFeedID.Namespace),
+			zap.String("changefeed", f.cfg.ChangeFeedID.ID),
+			zap.Error(err))
 	}()
 
 	eg, egCtx := errgroup.WithContext(ctx)

diff --git a/cdc/redo/writer/memory/mem_log_writer_test.go b/cdc/redo/writer/memory/mem_log_writer_test.go
@@ -88,11 +88,19 @@ func testWriteEvents(t *testing.T, events []writer.RedoEvent) {
 	require.NoError(t, err)
 
 	require.ErrorIs(t, lw.Close(), context.Canceled)
+<<<<<<< HEAD
 	require.Eventually(t, func() bool {
 		err = lw.WriteEvents(ctx, events...)
 		return err != nil
 	}, 2*time.Second, 10*time.Millisecond)
 	require.ErrorContains(t, err, "redo log writer stopped")
 	err = lw.FlushLog(ctx)
 	require.ErrorContains(t, err, "redo log writer stopped")
+=======
+
+	err = lw.WriteEvents(ctx, events...)
+	require.NoError(t, err)
+	err = lw.FlushLog(ctx)
+	require.NoError(t, err)
+>>>>>>> 52b4301388 (redo(ticdc): return internal error in redo writer (#11011))
 }