use std::net::SocketAddr;
use clap::{command, Args};
use ockam_api::port_range::PortRange;
use ockam_multiaddr::MultiAddr;
use crate::kafka::util::{async_run, make_brokers_port_range, ArgOpts};
use crate::util::async_cmd;
use crate::{
kafka::{
kafka_consumer_default_addr, kafka_default_consumer_server, kafka_default_project_route,
},
node::NodeOpts,
util::parsers::socket_addr_parser,
CommandGlobalOpts,
};
#[derive(Clone, Debug, Args)]
pub struct CreateCommand {
#[command(flatten)]
node_opts: NodeOpts,
#[arg(long, default_value_t = kafka_consumer_default_addr())]
addr: String,
#[arg(long, default_value_t = kafka_default_consumer_server(), value_parser = socket_addr_parser)]
bootstrap_server: SocketAddr,
#[arg(long)]
brokers_port_range: Option<PortRange>,
#[arg(long, default_value_t = kafka_default_project_route())]
project_route: MultiAddr,
}
impl CreateCommand {
pub fn run(self, opts: CommandGlobalOpts) -> miette::Result<()> {
let cmd_name = self.name();
let arg_opts = ArgOpts {
endpoint: "/node/services/kafka_consumer".to_string(),
kafka_entity: "KafkaConsumer".to_string(),
node_opts: self.node_opts,
addr: self.addr,
bootstrap_server: self.bootstrap_server,
brokers_port_range: self
.brokers_port_range
.unwrap_or_else(|| make_brokers_port_range(&self.bootstrap_server)),
project_route: self.project_route,
};
async_cmd(&cmd_name, opts.clone(), |ctx| async move {
async_run(&ctx, opts, arg_opts).await
})
}
pub fn name(&self) -> String {
"create kafka consumer".into()
}
}