iggy_cli/commands/binary_partitions/
create_partitions.rs

1/* Licensed to the Apache Software Foundation (ASF) under one
2 * or more contributor license agreements.  See the NOTICE file
3 * distributed with this work for additional information
4 * regarding copyright ownership.  The ASF licenses this file
5 * to you under the Apache License, Version 2.0 (the
6 * "License"); you may not use this file except in compliance
7 * with the License.  You may obtain a copy of the License at
8 *
9 *   http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing,
12 * software distributed under the License is distributed on an
13 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14 * KIND, either express or implied.  See the License for the
15 * specific language governing permissions and limitations
16 * under the License.
17 */
18
19use crate::commands::cli_command::{CliCommand, PRINT_TARGET};
20use anyhow::Context;
21use async_trait::async_trait;
22use iggy_common::Client;
23use iggy_common::Identifier;
24use iggy_common::create_partitions::CreatePartitions;
25use tracing::{Level, event};
26
27pub struct CreatePartitionsCmd {
28    create_partition: CreatePartitions,
29}
30
31impl CreatePartitionsCmd {
32    pub fn new(stream_id: Identifier, topic_id: Identifier, partitions_count: u32) -> Self {
33        Self {
34            create_partition: CreatePartitions {
35                stream_id,
36                topic_id,
37                partitions_count,
38            },
39        }
40    }
41}
42
43#[async_trait]
44impl CliCommand for CreatePartitionsCmd {
45    fn explain(&self) -> String {
46        let mut partitions = String::from("partition");
47        if self.create_partition.partitions_count > 1 {
48            partitions.push('s');
49        };
50
51        format!(
52            "create {} {partitions} for topic with ID: {} and stream with ID: {}",
53            self.create_partition.partitions_count,
54            self.create_partition.topic_id,
55            self.create_partition.stream_id
56        )
57    }
58
59    async fn execute_cmd(&mut self, client: &dyn Client) -> anyhow::Result<(), anyhow::Error> {
60        let mut partitions = String::from("partition");
61        if self.create_partition.partitions_count > 1 {
62            partitions.push('s');
63        };
64
65        client
66            .create_partitions(
67                &self.create_partition.stream_id,
68                &self.create_partition.topic_id,
69                self.create_partition.partitions_count,
70            )
71            .await
72            .with_context(|| {
73                format!(
74                    "Problem creating {} {partitions} for topic with ID: {} and stream with ID: {}",
75                    self.create_partition.partitions_count,
76                    self.create_partition.topic_id,
77                    self.create_partition.stream_id
78                )
79            })?;
80
81        event!(target: PRINT_TARGET, Level::INFO,
82            "Created {} {partitions} for topic with ID: {} and stream with ID: {}",
83            self.create_partition.partitions_count,
84            self.create_partition.topic_id,
85            self.create_partition.stream_id,
86        );
87
88        Ok(())
89    }
90}
iggy_cli/commands/binary_partitions/create_partitions.rs

iggy_cli/commands/binary_partitions/
create_partitions.rs
