risingwave_frontend/handler/
cancel_job.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use pgwire::pg_response::{PgResponse, StatementType};
16use risingwave_common::types::Fields;
17use risingwave_pb::meta::cancel_creating_jobs_request::{CreatingJobIds, PbJobs};
18use risingwave_sqlparser::ast::JobIdents;
19
20use super::RwPgResponseBuilderExt;
21use crate::error::Result;
22use crate::handler::{HandlerArgs, RwPgResponse};
23
24pub(super) async fn handle_cancel(
25    handler_args: HandlerArgs,
26    jobs: JobIdents,
27) -> Result<RwPgResponse> {
28    let session = handler_args.session;
29
30    let canceled_jobs = session
31        .env()
32        .meta_client()
33        .cancel_creating_jobs(PbJobs::Ids(CreatingJobIds { job_ids: jobs.0 }))
34        .await?;
35    let rows = canceled_jobs
36        .into_iter()
37        .map(|id| CancelRow { id: id.to_string() });
38    Ok(PgResponse::builder(StatementType::CANCEL_COMMAND)
39        .rows(rows)
40        .into())
41}
42
43#[derive(Fields)]
44#[fields(style = "Title Case")]
45struct CancelRow {
46    id: String,
47}
risingwave_frontend/handler/cancel_job.rs

risingwave_frontend/handler/
cancel_job.rs
