risingwave_frontend/handler/
close_cursor.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_sqlparser::ast::CloseCursorStatement;
17
18use super::RwPgResponse;
19use crate::error::Result;
20use crate::handler::HandlerArgs;
21
22pub async fn handle_close_cursor(
23    handle_args: HandlerArgs,
24    stmt: CloseCursorStatement,
25) -> Result<RwPgResponse> {
26    let session = handle_args.session.clone();
27    let cursor_manager = session.get_cursor_manager();
28    if let Some(cursor_name) = stmt.cursor_name {
29        cursor_manager
30            .remove_cursor(&cursor_name.real_value())
31            .await?;
32    } else {
33        cursor_manager.remove_all_cursor().await;
34    }
35    Ok(PgResponse::empty_result(StatementType::CLOSE_CURSOR))
36}
risingwave_frontend/handler/close_cursor.rs

risingwave_frontend/handler/
close_cursor.rs
