ave-core 0.8.0

Averiun Ledger core runtime and node API
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

use std::sync::Arc;

use async_trait::async_trait;
use ave_actors::{
    Actor, ActorContext, ActorError, ActorPath, Handler, NotPersistentActor,
};
use ave_common::identity::{PublicKey, TimeStamp};
use borsh::{BorshDeserialize, BorshSerialize};
use serde::{Deserialize, Serialize};
use tracing::{Span, error, info_span};

use crate::{NetworkMessage, helpers::network::service::NetworkSender};

use super::common::emit_fail;

#[derive(
    Debug,
    Clone,
    Serialize,
    Deserialize,
    PartialEq,
    Eq,
    Hash,
    BorshSerialize,
    BorshDeserialize,
    Ord,
    PartialOrd,
)]
pub struct TimeOut {
    pub who: PublicKey,
    pub re_trys: u32,
    pub timestamp: TimeStamp,
}

#[derive(Clone, Debug)]
pub struct RetryNetwork {
    network: Arc<NetworkSender>,
}

impl RetryNetwork {
    pub const fn new(network: Arc<NetworkSender>) -> Self {
        Self { network }
    }
}

#[async_trait]
impl Actor for RetryNetwork {
    type Event = ();
    type Message = NetworkMessage;
    type Response = ();

    fn get_span(_id: &str, parent_span: Option<Span>) -> tracing::Span {
        parent_span.map_or_else(
            || info_span!("RetryNetwork"),
            |parent_span| info_span!(parent: parent_span, "RetryNetwork"),
        )
    }
}

impl NotPersistentActor for RetryNetwork {}

#[async_trait]
impl Handler<Self> for RetryNetwork {
    async fn handle_message(
        &mut self,
        _sender: ActorPath,
        msg: NetworkMessage,
        ctx: &mut ActorContext<Self>,
    ) -> Result<(), ActorError> {
        if let Err(e) = self
            .network
            .send_command(network::CommandHelper::SendMessage { message: msg })
            .await
        {
            error!(
                error = %e,
                "Failed to send message to network helper"
            );
            return Err(emit_fail(ctx, e).await);
        };
        Ok(())
    }
}