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

use std::collections::HashMap;
use std::time::SystemTime;

use crate::grpc_wrapper::raw_topic_service::common::codecs::RawSupportedCodecs;
use ydb_grpc::google_proto_workaround::protobuf::Timestamp;
use ydb_grpc::ydb_proto::topic::{Consumer, SupportedCodecs};

#[derive(serde::Serialize, Clone)]
pub(crate) struct RawConsumer {
    pub name: String,
    pub important: bool,
    pub read_from: Option<SystemTime>,
    pub supported_codecs: RawSupportedCodecs,
    pub attributes: HashMap<String, String>,
}

impl From<RawConsumer> for Consumer {
    fn from(value: RawConsumer) -> Self {
        let read_from = value.read_from.map(|value_read_from| {
            Timestamp {
                seconds: value_read_from
                    .duration_since(SystemTime::UNIX_EPOCH)
                    .unwrap()
                    .as_secs() as i64,
                nanos: value_read_from
                    .duration_since(SystemTime::UNIX_EPOCH)
                    .unwrap()
                    .as_nanos() as i32,
            }
        });

        Self {
            name: value.name,
            important: value.important,
            read_from,
            supported_codecs: Some(SupportedCodecs::from(value.supported_codecs)),
            attributes: value.attributes,
        }
    }
}