kstone-client 0.1.0

gRPC client library for KeystoneDB remote access
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

/// Remote PartiQL operations
use crate::convert::*;
use crate::error::{ClientError, Result};
use bytes::Bytes;
use kstone_core::Item;
use kstone_proto as proto;

/// Execute statement response (mirrors kstone-api ExecuteStatementResponse)
#[derive(Debug)]
pub enum RemoteExecuteStatementResponse {
    /// SELECT statement result
    Select {
        items: Vec<Item>,
        count: usize,
        scanned_count: usize,
        last_key: Option<(Bytes, Option<Bytes>)>,
    },
    /// INSERT statement result
    Insert { success: bool },
    /// UPDATE statement result
    Update { item: Item },
    /// DELETE statement result
    Delete { success: bool },
}

/// Parse ExecuteStatementResponse from protobuf
pub(crate) fn parse_execute_statement_response(
    response: proto::ExecuteStatementResponse,
) -> Result<RemoteExecuteStatementResponse> {
    use proto::execute_statement_response::Response as ProtoResponse;

    let proto_response = response
        .response
        .ok_or_else(|| ClientError::InternalError("Empty response from server".to_string()))?;

    match proto_response {
        ProtoResponse::Select(select_result) => {
            let items: Vec<Item> = select_result
                .items
                .into_iter()
                .map(|proto_item| proto_item_to_ks(proto_item).expect("Invalid item from server"))
                .collect();

            let last_key = select_result
                .last_key
                .map(proto_last_key_to_ks);

            Ok(RemoteExecuteStatementResponse::Select {
                items,
                count: select_result.count as usize,
                scanned_count: select_result.scanned_count as usize,
                last_key,
            })
        }
        ProtoResponse::Insert(insert_result) => {
            Ok(RemoteExecuteStatementResponse::Insert {
                success: insert_result.success,
            })
        }
        ProtoResponse::Update(update_result) => {
            let item = proto_item_to_ks(
                update_result.item.expect("Server should return updated item")
            )?;

            Ok(RemoteExecuteStatementResponse::Update { item })
        }
        ProtoResponse::Delete(delete_result) => {
            Ok(RemoteExecuteStatementResponse::Delete {
                success: delete_result.success,
            })
        }
    }
}