Merge pull request #24 from yajamon/trade_api/cancel_order

Trade api/cancel orderを実装

Author: yajamon

src/main.rs

@@ -1,4 +1,8 @@
 extern crate zaif_api;
+extern crate serde_json;
+
+use std::{thread, time};
+use serde_json::Value;
 
 use zaif_api::AccessKey;
 use zaif_api::public_api::*;
@@ -27,7 +31,21 @@ fn main() {
         .price(1.0)
         .amount(0.1)
         .finalize();
-    println!("{}", api.exec().unwrap());
+    match api.exec() {
+        Ok(res) => {
+            println!("{}", res);
+            let json:Value = serde_json::from_str(res.as_str()).unwrap();
+            let order_id = json["return"]["order_id"].as_u64().unwrap();
+            let api = CancelOrderBuilder::new(access_key.clone())
+                .order_id(order_id)
+                .currency_pair("zaif_jpy")
+                .finalize();
+            let wait_time = time::Duration::from_secs(5);
+            thread::sleep(wait_time);
+            println!("{}", api.exec().unwrap());
+        },
+        _ => return,
+    }
 
     let api = ActiveOrdersBuilder::new(access_key.clone()).currency_pair("zaif_jpy").finalize();
     println!("{}", api.exec().unwrap());

src/trade_api/cancel_order.rs

@@ -0,0 +1,62 @@
+extern crate reqwest;
+
+use std::collections::HashMap;
+
+use core::*;
+
+pub struct CancelOrder {
+    access_key: AccessKey,
+    order_id: u64,
+    currency_pair: Option<String>,
+}
+
+impl CancelOrder {
+    pub fn exec(&self) -> reqwest::Result<String> {
+        let param: &mut HashMap<String, String> = &mut HashMap::new();
+        param.insert("method".to_string(), "cancel_order".to_string());
+        param.insert("order_id".to_string(), format!("{}", self.order_id));
+        if let Some(ref currency_pair) = self.currency_pair {
+            param.insert("currency_pair".to_string(), format!("{}", currency_pair.clone()));
+        }
+
+        let api = ApiBuilder::new()
+            .access_key(self.access_key.clone())
+            .uri("https://api.zaif.jp/tapi")
+            .method(Method::Post)
+            .param(param.clone())
+            .finalize();
+
+        api.exec()
+    }
+}
+
+pub struct CancelOrderBuilder {
+    access_key: AccessKey,
+    order_id: Option<u64>,
+    currency_pair: Option<String>,
+}
+
+impl CancelOrderBuilder{
+    pub fn new(access_key: AccessKey) -> CancelOrderBuilder {
+        CancelOrderBuilder {
+            access_key: access_key,
+            order_id: None,
+            currency_pair: None,
+        }
+    }
+    pub fn order_id(&mut self, order_id: u64) -> &mut CancelOrderBuilder {
+        self.order_id = Some(order_id);
+        self
+    }
+    pub fn currency_pair(&mut self, currency_pair: &str) -> &mut CancelOrderBuilder {
+        self.currency_pair = Some(currency_pair.to_string());
+        self
+    }
+    pub fn finalize(&self) -> CancelOrder {
+        CancelOrder {
+            access_key: self.access_key.clone(),
+            order_id: self.order_id.unwrap(),
+            currency_pair: self.currency_pair.clone(),
+        }
+    }
+}

src/trade_api/mod.rs

@@ -1,7 +1,9 @@
 pub use self::get_info2::*;
 pub use self::trade::*;
 pub use self::active_orders::*;
+pub use self::cancel_order::*;
 
 mod get_info2;
 mod trade;
 mod active_orders;
+mod cancel_order;