12
12
// See the License for the specific language governing permissions and
13
13
// limitations under the License.
14
14
15
+ use anyhow:: Context ;
15
16
use either:: Either ;
16
17
use futures:: stream:: PollNext ;
17
18
use futures:: StreamExt ;
@@ -247,7 +248,7 @@ pub async fn stream_lookup_arrange_prev_epoch(
247
248
match input
248
249
. next ( )
249
250
. await
250
- . expect ( "unexpected close of barrier aligner" ) ?
251
+ . context ( "unexpected close of barrier aligner" ) ? ?
251
252
{
252
253
Either :: Left ( Message :: Watermark ( _) ) => {
253
254
todo ! ( "https://github.com/risingwavelabs/risingwave/issues/6042" )
@@ -298,7 +299,7 @@ pub async fn stream_lookup_arrange_this_epoch(
298
299
match input
299
300
. next ( )
300
301
. await
301
- . expect ( "unexpected close of barrier aligner" ) ?
302
+ . context ( "unexpected close of barrier aligner" ) ? ?
302
303
{
303
304
Either :: Left ( Message :: Chunk ( msg) ) => {
304
305
// Should wait until arrangement from this epoch is available.
@@ -333,7 +334,7 @@ pub async fn stream_lookup_arrange_this_epoch(
333
334
match input
334
335
. next ( )
335
336
. await
336
- . expect ( "unexpected close of barrier aligner" ) ?
337
+ . context ( "unexpected close of barrier aligner" ) ? ?
337
338
{
338
339
Either :: Left ( Message :: Chunk ( msg) ) => yield ArrangeMessage :: Stream ( msg) ,
339
340
Either :: Left ( Message :: Barrier ( b) ) => {
@@ -355,7 +356,7 @@ pub async fn stream_lookup_arrange_this_epoch(
355
356
match input
356
357
. next ( )
357
358
. await
358
- . expect ( "unexpected close of barrier aligner" ) ?
359
+ . context ( "unexpected close of barrier aligner" ) ? ?
359
360
{
360
361
Either :: Left ( _) => unreachable ! ( ) ,
361
362
Either :: Right ( Message :: Chunk ( chunk) ) => {
0 commit comments