kafrust 0.3.3

A pure Rust Kafka client with no librdkafka or C client binding dependency.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

mod common;

use kafrust::{
    AdminClient, ClientConfig, CreateTopicsOptions, DeleteTopicsOptions, Error, NewTopic,
};

#[tokio::main]
async fn main() -> kafrust::Result<()> {
    let topic = std::env::var("KAFRUST_ADMIN_TOPIC").map_err(|_| Error::InvalidConfiguration {
        field: "KAFRUST_ADMIN_TOPIC",
        reason: "the authorization example requires a unique topic name",
    })?;
    let expected_error = parse_i16_env("KAFRUST_EXPECT_CREATE_TOPICS_ERROR")?;
    let config = common::apply_security(
        ClientConfig::new(common::bootstrap_servers_from_env())
            .client_id("kafrust-admin-create-topics-authorization-example"),
    )?;
    let admin = AdminClient::new(config);

    let result = admin
        .create_topics(&[NewTopic::new(&topic, 1, 1)], CreateTopicsOptions::new())
        .await?;
    let outcome = result
        .topics()
        .first()
        .ok_or(Error::Unsupported("CreateTopics returned no topic outcome"))?;
    if outcome.error_code() != expected_error {
        return Err(Error::Broker {
            code: outcome.error_code(),
            context: format!(
                "CreateTopics returned {}, expected {}",
                outcome.error_code(),
                expected_error
            ),
        });
    }

    if expected_error == 0 {
        if !outcome.is_success() {
            return Err(Error::Broker {
                code: outcome.error_code(),
                context: "CreateTopics success expectation was not successful".to_owned(),
            });
        }
        if !admin
            .list_topics()
            .await?
            .iter()
            .any(|listed| listed.name() == topic)
        {
            return Err(Error::UnknownTopicOrPartition {
                topic,
                partition: -1,
            });
        }
        let deleted = admin
            .delete_topics(&[topic.clone()], DeleteTopicsOptions::new())
            .await?;
        let delete_outcome = deleted
            .topics()
            .first()
            .ok_or(Error::Unsupported("DeleteTopics returned no topic outcome"))?;
        if !delete_outcome.is_success() {
            return Err(Error::Broker {
                code: delete_outcome.error_code(),
                context: "cleanup DeleteTopics failed".to_owned(),
            });
        }
        println!("CreateTopics allowed for {topic}; cleanup completed");
        return Ok(());
    }

    if outcome.is_success() {
        return Err(Error::Broker {
            code: expected_error,
            context: "CreateTopics succeeded despite the expected authorization error".to_owned(),
        });
    }
    if admin
        .list_topics()
        .await?
        .iter()
        .any(|listed| listed.name() == topic)
    {
        return Err(Error::Broker {
            code: expected_error,
            context: "CreateTopics changed topic state despite the expected error".to_owned(),
        });
    }
    println!("CreateTopics denied with expected error {expected_error}; topic state unchanged");
    Ok(())
}

fn parse_i16_env(name: &'static str) -> kafrust::Result<i16> {
    let value = std::env::var(name).map_err(|_| Error::InvalidConfiguration {
        field: name,
        reason: "the expected broker error code is required",
    })?;
    value.parse().map_err(|_| Error::InvalidConfiguration {
        field: name,
        reason: "value must be a signed 16-bit integer",
    })
}