ryerraguntla commented on code in PR #4400: URL: https://github.com/apache/iggy/pull/4400#discussion_r4205817221
########## gateways/kafka/src/protocol/handlers/delete_topics.rs: ########## @@ -0,0 +1,307 @@ +// Licensed to the Apache Software Foundation (ASF) under one +// or more contributor license agreements. See the NOTICE file +// distributed with this work for additional information +// regarding copyright ownership. The ASF licenses this file +// to you under the Apache License, Version 2.0 (the +// "License"); you may not use this file except in compliance +// with the License. You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, +// software distributed under the License is distributed on an +// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +// KIND, either express or implied. See the License for the +// specific language governing permissions and limitations +// under the License. + +//! `DeleteTopics` (API key 20). +//! +//! Deletes the Iggy topic a Kafka topic name resolves to. Never deletes the backing Iggy +//! stream - see [`IggyBridge::delete_kafka_topic`]'s own doc comment for why: the shared +//! `default_stream` can hold other Kafka topics' data, and this bridge has no way to tell +//! whether the stream it just emptied is actually abandoned or just temporarily topic-less. +//! +//! `DeletableTopicResult.error_message` is only encoded on the wire at v5+ (`kafka_protocol`'s +//! own gate) - a client negotiating v1-v4 gets `error_code` only, with any text this module sets +//! silently dropped. Unlike `CreateTopics`, whose equivalent field is encoded at every version it +//! supports. + +use bytes::Bytes; +use iggy::prelude::IggyError; +use kafka_protocol::messages::delete_topics_response::DeletableTopicResult; +use kafka_protocol::messages::{DeleteTopicsRequest, DeleteTopicsResponse, TopicName}; +use kafka_protocol::protocol::StrBytes; + +use tokio::time::Instant; + +use crate::bridge::{BridgeError, IggyBridge}; +use crate::error::Result; +use crate::protocol::api::{ + API_KEY_DELETE_TOPICS, ApiVersionRange, ERROR_INVALID_REQUEST, ERROR_NONE, + ERROR_NOT_CONTROLLER, ERROR_POLICY_VIOLATION, ERROR_REQUEST_TIMED_OUT, GatewayState, + HandleOutcome, +}; +use crate::protocol::bounds_guard::validate_delete_topics_shape; +use crate::protocol::handlers::{ + clamp_request_timeout, decode_guarded, encode_message, handle_versioned_request, + is_supported_version, respond_or_close, unsupported_version_response, +}; + +/// This bridge never advertises v6 (the `topics: Vec<DeleteTopicState>`, topic-id-based shape). +/// +/// It has no concept of a Kafka topic id, only the Kafka-side name `TopicMapping` resolves. v1 +/// is the crate's own floor (`kafka_protocol` does not implement v0 for this message). +pub const RANGE: ApiVersionRange = ApiVersionRange { + api_key: API_KEY_DELETE_TOPICS, + min_version: 1, + max_version: 5, +}; + +/// Cap on topic names one `DeleteTopics` request may address through the bridge. +/// +/// Same rationale as `create_topics::MAX_BRIDGE_BACKED_TOPICS`, but counted differently: every +/// name here costs its own bridge round trip against the single lockstep `IggyClient` every +/// Kafka connection on this gateway shares, duplicates included, so this charges the raw count +/// rather than the distinct one `CreateTopics` charges. +const MAX_BRIDGE_BACKED_TOPICS: usize = 100; + +pub async fn handle(state: &GatewayState, api_version: i16, body: Bytes) -> HandleOutcome { + let Some(bridge) = &state.bridge else { + return handle_versioned_request( + API_KEY_DELETE_TOPICS, + api_version, + body, + |v, b| { + decode_guarded::<DeleteTopicsRequest>(v, b, |v, b| { + validate_delete_topics_shape(v, b, state.max_frame_size) + }) + }, + encode_response, + encode_error_response, + "DeleteTopics", + ); + }; + + if !is_supported_version(API_KEY_DELETE_TOPICS, api_version) { + return unsupported_version_response(API_KEY_DELETE_TOPICS, api_version, |version| { + encode_error_response(version, ERROR_INVALID_REQUEST) + }); + } + + let req = match decode_guarded::<DeleteTopicsRequest>(api_version, body, |v, b| { + validate_delete_topics_shape(v, b, state.max_frame_size) + }) { + Ok(req) => req, + Err(error) => { + // debug!, not warn!: attacker-controlled, not operator-actionable. + tracing::debug!(%error, "Failed to decode DeleteTopics request"); + return respond_or_close( + encode_error_response(api_version, ERROR_INVALID_REQUEST), + "DeleteTopics", + ); + } + }; + + // RANGE caps at v5, so req.topics (the v6 topic-id shape) is always empty - topic_names is + // the only populated field at any version this bridge advertises. + if req.topic_names.len() > MAX_BRIDGE_BACKED_TOPICS { + tracing::warn!( + requested_topics = req.topic_names.len(), + max = MAX_BRIDGE_BACKED_TOPICS, + "DeleteTopics request addresses too many topics; rejecting" + ); + let message = StrBytes::from(format!( + "this gateway addresses at most {MAX_BRIDGE_BACKED_TOPICS} topic names per DeleteTopics request" + )); + let results = req + .topic_names + .iter() + .map(|name| { + DeletableTopicResult::default() + .with_name(Some(name.clone())) + .with_error_code(ERROR_POLICY_VIOLATION) + .with_error_message(Some(message.clone())) + }) + .collect(); + let resp = DeleteTopicsResponse::default().with_responses(results); + return respond_or_close(encode_message(&resp, api_version, 256), "DeleteTopics"); + } + + let deadline = Instant::now() + clamp_request_timeout(req.timeout_ms); + let results = delete_all_topics(bridge, &req.topic_names, deadline).await; + let resp = DeleteTopicsResponse::default().with_responses(results); + respond_or_close(encode_message(&resp, api_version, 256), "DeleteTopics") +} + +/// Deletes every requested topic, independently of the others. +/// +/// `deadline` bounds each topic's own bridge work individually (`timeout_at`), not the whole +/// loop - same reasoning as `create_topics::create_all_topics`: a single `timeout` around the +/// entire call would discard every already-resolved result the moment one topic's call ran +/// long, answering a retriable code even for topics that had already deleted cleanly. +/// +/// A name repeated within the same request is not specially rejected the way `CreateTopics` +/// rejects a duplicate: deleting the same topic twice has no race to protect against the way +/// creating it twice does - the first occurrence deletes it, the second then answers +/// `UNKNOWN_TOPIC_OR_PARTITION` because by the time it runs, the topic genuinely doesn't exist +/// any more. That's the correct answer for a duplicate, not a special case. +async fn delete_all_topics( + bridge: &IggyBridge, + topic_names: &[TopicName], + deadline: Instant, +) -> Vec<DeletableTopicResult> { + let mut results = Vec::with_capacity(topic_names.len()); + for name in topic_names { + let result = match tokio::time::timeout_at(deadline, delete_one_topic(bridge, name)).await { Review Comment: fixed -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
