Skip to content

Commit

Permalink
chore: refactor handle_stream_read
Browse files Browse the repository at this point in the history
  • Loading branch information
ShiKaiWi committed Jun 1, 2023
1 parent 9d28fe6 commit b38a68d
Showing 1 changed file with 14 additions and 15 deletions.
29 changes: 14 additions & 15 deletions server/src/grpc/remote_engine_service/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -311,27 +311,26 @@ async fn handle_stream_read(
ctx: HandlerContext,
request: ReadRequest,
) -> Result<PartitionedStreams> {
let read_request: table_engine::remote::model::ReadRequest =
request.try_into().box_err().context(ErrWithCause {
code: StatusCode::BadRequest,
msg: "fail to convert read request",
})?;
let table_engine::remote::model::ReadRequest {
table: table_ident,
read_request,
} = request.try_into().box_err().context(ErrWithCause {
code: StatusCode::BadRequest,
msg: "fail to convert read request",
})?;

let begin = Instant::now();
let request_id = read_request.read_request.request_id;
let request_id = read_request.request_id;
info!(
"Handle stream read, request_id:{request_id}, table:{:?}, read_options:{:?}, read_order:{:?}, predicate:{:?} ",
read_request.table,
read_request.read_request.opts,
read_request.read_request.order,
read_request.read_request.predicate,
"Handle stream read, request_id:{request_id}, table:{table_ident:?}, read_options:{:?}, read_order:{:?}, predicate:{:?} ",
read_request.opts,
read_request.order,
read_request.predicate,
);

let table = find_table_by_identifier(&ctx, &read_request.table)?;
let table_ident = read_request.table;

let table = find_table_by_identifier(&ctx, &table_ident)?;
let streams = table
.partitioned_read(read_request.read_request)
.partitioned_read(read_request)
.await
.box_err()
.with_context(|| ErrWithCause {
Expand Down

0 comments on commit b38a68d

Please sign in to comment.