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

use exc_core::error::ExchangeError;
use thiserror::Error;

use crate::{
    key::SignError,
    websocket::types::{
        messages::{request::WsRequest, Args},
        response::StatusKind,
    },
};

/// All OKX errors.
#[derive(Debug, Error)]
pub enum OkxError {
    /// Stream dropped.
    #[error("stream is dropped")]
    StreamDropped,
    /// Websocket error.
    #[error("weboscket: {0}")]
    Websocket(#[from] tokio_tungstenite::tungstenite::Error),
    /// Remote closed.
    #[error("remote closed")]
    RemoteClosed,
    /// Connection error.
    #[error("connection error: {0}")]
    Connection(Box<dyn std::error::Error + Send + Sync>),
    /// Websocket disconnected.
    #[error("websocket disconnected")]
    WebsocketDisconnected,
    /// Ping error.
    #[error("ping error: {0}")]
    Ping(anyhow::Error),
    /// Ping timeout.
    #[error("ping timeout")]
    PingTimeout,
    /// Json error.
    #[error("json: {0}")]
    Json(#[from] serde_json::Error),
    /// Request sender dropped.
    #[error("request sender dropped")]
    RequestSenderDropped,
    /// Dispatch error.
    #[error("dispatch error: req={0:?}")]
    Dispatch(WsRequest),
    /// Callback error.
    #[error("responser error: {0}")]
    Callback(#[from] tokio::sync::oneshot::error::RecvError),
    /// Already subscribed or unsubscribing.
    #[error("already subscribed or unsubscribping {0:?}")]
    SubscribedOrUnsubscribing(Args),
    /// Subscribing or unsubscribing.
    #[error("subscribing or unsubscribing {0:?}")]
    SubscribingOrUnsubscribing(Args),
    /// Websocket closed.
    #[error("websocket closed")]
    WebsocketClosed,
    /// API Error.
    #[error("api error: {0}")]
    Api(StatusKind),
    /// Protocol Error.
    #[error("protocol: {0}")]
    Protocol(anyhow::Error),
    /// Layers error.
    #[error(transparent)]
    Layer(Box<dyn std::error::Error + Send + Sync>),
    /// Buffer Layer Error.
    #[error(transparent)]
    Buffer(Box<dyn std::error::Error + Send + Sync>),
    /// Unexpected data type.
    #[error("unpexpected data type: {0}")]
    UnexpectedDataType(anyhow::Error),
    /// Sign error.
    #[error("okx key sign error: {0}")]
    SignError(#[from] SignError),
    /// Login error.
    #[error("login error: {0}")]
    LoginError(anyhow::Error),
    /// Parsing order error.
    #[error("parsing order: {0}")]
    ParsingOrder(String),
}

impl OkxError {
    /// Parsing order error.
    pub fn parsing_order(msg: impl ToString) -> Self {
        Self::ParsingOrder(msg.to_string())
    }
}

impl From<OkxError> for ExchangeError {
    fn from(err: OkxError) -> Self {
        Self::Other(err.into())
    }
}