fix

risingwavelabs · Feb 2, 2023 · e32e123 · e32e123
1 parent a142f80
commit e32e123
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/batch/src/rpc/service/task_service.rs b/src/batch/src/rpc/service/task_service.rs
@@ -29,7 +29,7 @@ use crate::task::{
     self, BatchEnvironment, BatchManager, BatchTaskExecution, ComputeNodeContext, TaskId,
 };
 
-const LOCAL_EXECUTE_BUFFER_SIZE: usize = 1024;
+const LOCAL_EXECUTE_BUFFER_SIZE: usize = 64;
 
 #[derive(Clone)]
 pub struct BatchServiceImpl {

diff --git a/src/batch/src/task/task_execution.rs b/src/batch/src/task/task_execution.rs
@@ -374,7 +374,9 @@ impl<C: BatchTaskContext> BatchTaskExecution<C> {
                     let err_str = e.to_string();
                     *failure.lock() = Some(e);
                     // There will be no more chunks, so send None.
-                    sender.send(None).await.unwrap();
+                    if let Err(_e) = sender.send(None).await {
+                        warn!("failed to send None to annotate end");
+                    }
                     if let Err(_e) = t_1
                         .change_state_notify(
                             TaskStatus::Failed,