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

use model::*;
use client::*;
use errors::*;
use serde_json::from_str;

static USER_DATA_STREAM: &'static str = "/api/v1/userDataStream";

#[derive(Clone)]
pub struct UserStream {
    pub client: Client,
    pub recv_window: u64,
}

impl UserStream {
    // User Stream
    pub fn start(&self) -> Result<(UserDataStream)> {
        let data = self.client.post(USER_DATA_STREAM)?;
        let user_data_stream: UserDataStream = from_str(data.as_str()).unwrap();

        Ok(user_data_stream)
    }

    // Current open orders on a symbol
    pub fn keep_alive(&self, listen_key: &str) -> Result<(Success)> {
        let data = self.client.put(USER_DATA_STREAM, listen_key)?;

        let success: Success = from_str(data.as_str()).unwrap();

        Ok(success)
    }

    pub fn close(&self, listen_key: &str) -> Result<(Success)> {
        let data = self.client.put(USER_DATA_STREAM, listen_key)?;

        let success: Success = from_str(data.as_str()).unwrap();

        Ok(success)
    }
}