unc_jsonrpc_primitives/types/
chunks.rs

1use serde_json::Value;
2
3#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, arbitrary::Arbitrary)]
4#[serde(untagged)]
5pub enum ChunkReference {
6    BlockShardId {
7        block_id: unc_primitives::types::BlockId,
8        shard_id: unc_primitives::types::ShardId,
9    },
10    ChunkHash {
11        chunk_id: unc_primitives::hash::CryptoHash,
12    },
13}
14
15#[derive(serde::Serialize, serde::Deserialize, Debug, arbitrary::Arbitrary)]
16pub struct RpcChunkRequest {
17    #[serde(flatten)]
18    pub chunk_reference: ChunkReference,
19}
20
21#[derive(serde::Serialize, serde::Deserialize, Debug)]
22pub struct RpcChunkResponse {
23    #[serde(flatten)]
24    pub chunk_view: unc_primitives::views::ChunkView,
25}
26
27#[derive(thiserror::Error, Debug, serde::Serialize, serde::Deserialize)]
28#[serde(tag = "name", content = "info", rename_all = "SCREAMING_SNAKE_CASE")]
29pub enum RpcChunkError {
30    #[error("The node reached its limits. Try again later. More details: {error_message}")]
31    InternalError { error_message: String },
32    #[error("Block either has never been observed on the node or has been garbage collected: {error_message}")]
33    UnknownBlock {
34        #[serde(skip_serializing)]
35        error_message: String,
36    },
37    #[error("Shard id {shard_id} does not exist")]
38    InvalidShardId { shard_id: u64 },
39    #[error("Chunk with hash {chunk_hash:?} has never been observed on this node")]
40    UnknownChunk { chunk_hash: unc_primitives::sharding::ChunkHash },
41}
42
43impl From<RpcChunkError> for crate::errors::RpcError {
44    fn from(error: RpcChunkError) -> Self {
45        let error_data = match &error {
46            RpcChunkError::InternalError { .. } => Some(Value::String(error.to_string())),
47            RpcChunkError::UnknownBlock { error_message } => Some(Value::String(format!(
48                "DB Not Found Error: {} \n Cause: Unknown",
49                error_message
50            ))),
51            RpcChunkError::InvalidShardId { .. } => Some(Value::String(error.to_string())),
52            RpcChunkError::UnknownChunk { chunk_hash } => Some(Value::String(format!(
53                "Chunk Missing (unavailable on the node): ChunkHash(`{}`) \n Cause: Unknown",
54                chunk_hash.0
55            ))),
56        };
57
58        let error_data_value = match serde_json::to_value(error) {
59            Ok(value) => value,
60            Err(err) => {
61                return Self::new_internal_error(
62                    None,
63                    format!("Failed to serialize RpcStateChangesError: {:?}", err),
64                )
65            }
66        };
67
68        Self::new_internal_or_handler_error(error_data, error_data_value)
69    }
70}
unc_jsonrpc_primitives/types/chunks.rs

unc_jsonrpc_primitives/types/
chunks.rs
