ydb 0.11.2

Crate contains generated low-level grpc code from YDB API protobuf, used as base for ydb crate
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
104
105
106
107
108
109
110

use crate::grpc_wrapper::raw_errors::RawError;
use crate::grpc_wrapper::raw_table_service::query_stats::RawQueryStatMode;
use crate::grpc_wrapper::raw_table_service::transaction_control::RawTransactionControl;
use crate::grpc_wrapper::raw_table_service::value::r#type::RawType;
use crate::grpc_wrapper::raw_table_service::value::{RawResultSet, RawTypedValue};
use crate::grpc_wrapper::raw_ydb_operation::RawOperationParams;
use std::collections::HashMap;

#[derive(serde::Serialize)]
pub(crate) struct RawExecuteDataQueryRequest {
    pub session_id: String,
    pub tx_control: RawTransactionControl,
    pub yql_text: String,
    pub operation_params: RawOperationParams,
    pub params: HashMap<String, RawTypedValue>,
    pub keep_in_cache: bool,
    pub collect_stats: RawQueryStatMode,
}

impl From<RawExecuteDataQueryRequest> for ydb_grpc::ydb_proto::table::ExecuteDataQueryRequest {
    fn from(v: RawExecuteDataQueryRequest) -> Self {
        Self {
            session_id: v.session_id,
            tx_control: Some(v.tx_control.into()),
            query: Some(ydb_grpc::ydb_proto::table::Query {
                query: Some(ydb_grpc::ydb_proto::table::query::Query::YqlText(
                    v.yql_text,
                )),
            }),
            parameters: v.params.into_iter().map(|(k, v)| (k, v.into())).collect(),
            query_cache_policy: Some(ydb_grpc::ydb_proto::table::QueryCachePolicy {
                keep_in_cache: v.keep_in_cache,
            }),
            operation_params: Some(v.operation_params.into()),
            collect_stats: ydb_grpc::ydb_proto::table::query_stats_collection::Mode::from(
                v.collect_stats,
            ) as i32,
        }
    }
}

#[derive(serde::Serialize)]
pub(crate) struct RawExecuteDataQueryResult {
    pub result_sets: Vec<RawResultSet>,
    pub tx_meta: RawTransactionMeta,
    pub query_meta: Option<RawQueryMeta>,
    // query_stats: Option<RawQueryStats>, // todo
}

impl TryFrom<ydb_grpc::ydb_proto::table::ExecuteQueryResult> for RawExecuteDataQueryResult {
    type Error = RawError;

    fn try_from(
        value: ydb_grpc::ydb_proto::table::ExecuteQueryResult,
    ) -> Result<Self, Self::Error> {
        let result_sets_res: Result<_, RawError> = value
            .result_sets
            .into_iter()
            .map(|item| item.try_into())
            .collect();

        let query_meta = if let Some(proto_meta) = value.query_meta {
            Some(RawQueryMeta::try_from(proto_meta)?)
        } else {
            None
        };
        Ok(Self {
            result_sets: result_sets_res?,
            tx_meta: value
                .tx_meta
                .ok_or_else(|| RawError::custom("no tx_meta at ExecuteQueryResult"))?
                .into(),
            query_meta,
        })
    }
}

#[derive(serde::Serialize)]
pub(crate) struct RawTransactionMeta {
    pub id: String,
}

impl From<ydb_grpc::ydb_proto::table::TransactionMeta> for RawTransactionMeta {
    fn from(value: ydb_grpc::ydb_proto::table::TransactionMeta) -> Self {
        Self { id: value.id }
    }
}

#[derive(serde::Serialize)]
pub(crate) struct RawQueryMeta {
    pub id: String,
    pub parameter_types: HashMap<String, RawType>,
}

impl TryFrom<ydb_grpc::ydb_proto::table::QueryMeta> for RawQueryMeta {
    type Error = RawError;

    fn try_from(value: ydb_grpc::ydb_proto::table::QueryMeta) -> Result<Self, Self::Error> {
        let parameter_types_res: Result<HashMap<_, _>, RawError> = value
            .parameters_types
            .into_iter()
            .map(|(key, value)| Ok((key, value.try_into()?)))
            .collect();

        Ok(Self {
            id: value.id,
            parameter_types: parameter_types_res?,
        })
    }
}