use std::collections::BTreeMap; use chrono::Utc; use reqwest::{Client, Proxy}; use reqwest::header::HeaderMap; use ring::hmac; use rust_decimal::Decimal; use rust_decimal::prelude::FromPrimitive; use rust_decimal_macros::dec; use serde_json::{json, Value}; use tracing::{error, info, trace}; use crate::http_tool::RestTool; use crate::proxy; use crate::response_base::ResponseData; #[derive(Clone, Debug)] pub struct PhemexSwapRest { pub tag: String, base_url: String, /*******参数*/ //是否需要登录 //登录所需参数 login_param: BTreeMap, delays: Vec, max_delay: i64, avg_delay: Decimal, } impl PhemexSwapRest { /*******************************************************************************************************/ /*****************************************获取一个对象****************************************************/ /*******************************************************************************************************/ pub fn new(is_colo: bool, login_param: BTreeMap) -> PhemexSwapRest { return PhemexSwapRest::new_with_tag("default-PhemexSwapRest".to_string(), is_colo, login_param); } pub fn new_with_tag(tag: String, is_colo: bool, login_param: BTreeMap) -> PhemexSwapRest { let base_url = if is_colo { let z = "https://api.phemex.com".to_string(); info!("开启高速(未配置,走普通:{})通道",z); z } else { let z = "https://api.phemex.com".to_string(); info!("走普通通道:{}",z); z }; /*****返回结构体*******/ PhemexSwapRest { tag, base_url, login_param, delays: vec![], max_delay: 0, avg_delay: dec!(0.0), } } /*******************************************************************************************************/ /*****************************************rest请求函数********************************************************/ /*******************************************************************************************************/ //服务器时间 pub async fn get_server(&mut self) -> ResponseData { let params = json!({}); let data = self.request("GET".to_string(), "".to_string(), "/public/time".to_string(), false, params, ).await; data } //查詢合約基礎信息 pub async fn get_market(&mut self, params: Value) -> ResponseData { let data = self.request("GET".to_string(), "".to_string(), "/public/products".to_string(), false, params, ).await; data } //查詢ticker(Query 24 ticker) pub async fn get_ticker(&mut self, params: Value) -> ResponseData { let data = self.request("GET".to_string(), "".to_string(), "/md/v3/ticker/24hr".to_string(), false, params, ).await; data } //持仓(查询交易账户和仓位) pub async fn get_account_and_positions(&mut self, params: Value) -> ResponseData { let data = self.request("GET".to_string(), "".to_string(), "/g-accounts/accountPositions".to_string(), true, params, ).await; data } //仓位设置(Switch Position Mode Synchronously) pub async fn set_target_pos_mode(&mut self, params: Value) -> ResponseData { let data = self.request("PUT".to_string(), "".to_string(), "/g-positions/switch-pos-mode-sync".to_string(), true, params, ).await; data } //设置杠杆(Set Leverage 设置杠杆) pub async fn set_leverage(&mut self, params: Value) -> ResponseData { let data = self.request("PUT".to_string(), "".to_string(), "/g-positions/leverage".to_string(), true, params, ).await; data } //下单 pub async fn orders(&mut self, params: Value) -> ResponseData { let data = self.request("PUT".to_string(), "".to_string(), "/g-orders/create".to_string(), true, params, ).await; data } //撤单 pub async fn cancel_order(&mut self, params: Value) -> ResponseData { let data = self.request("DELETE".to_string(), "".to_string(), "/g-orders/cancel".to_string(), true, params, ).await; data } //撤销所有 pub async fn cancel_order_all(&mut self, params: Value) -> ResponseData { let data = self.request("DELETE".to_string(), "".to_string(), "/g-orders/all".to_string(), true, params, ).await; data } //订单列表 pub async fn get_orders(&mut self, params: Value) -> ResponseData { let data = self.request("GET".to_string(), "".to_string(), "/api-data/g-futures/orders".to_string(), true, params, ).await; data } //根据id查询订单 pub async fn get_orders_by_id(&mut self, params: Value) -> ResponseData { let data = self.request("GET".to_string(), "".to_string(), "/api-data/g-futures/orders/by-order-id".to_string(), true, params, ).await; data } /*******************************************************************************************************/ /*****************************************工具函数********************************************************/ /*******************************************************************************************************/ pub fn get_delays(&self) -> Vec { self.delays.clone() } pub fn get_avg_delay(&self) -> Decimal { self.avg_delay.clone() } pub fn get_max_delay(&self) -> i64 { self.max_delay.clone() } fn get_delay_info(&mut self) { let last_100 = if self.delays.len() > 100 { self.delays[self.delays.len() - 100..].to_vec() } else { self.delays.clone() }; let max_value = last_100.iter().max().unwrap(); if max_value.clone().to_owned() > self.max_delay { self.max_delay = max_value.clone().to_owned(); } let sum: i64 = last_100.iter().sum(); let sum_v = Decimal::from_i64(sum).unwrap(); let len_v = Decimal::from_u64(last_100.len() as u64).unwrap(); self.avg_delay = (sum_v / len_v).round_dp(1); self.delays = last_100.clone().into_iter().collect(); } //调用请求 pub async fn request(&mut self, method: String, prefix_url: String, request_url: String, is_login: bool, params_json: Value) -> ResponseData { trace!("login_param:{:?}", self.login_param); //解析账号信息 let mut access_key = "".to_string(); let mut secret_key = "".to_string(); if self.login_param.contains_key("access_key") { access_key = self.login_param.get("access_key").unwrap().to_string(); } if self.login_param.contains_key("secret_key") { secret_key = self.login_param.get("secret_key").unwrap().to_string(); } let mut is_login_param = true; if access_key == "" || secret_key == "" { is_login_param = false } //每个接口都有的参数 let timestamp = (Utc::now().timestamp_millis() + (60 * 1000)) / 1000; //请求类型不同,可能请求头body 不同 let mut body = "".to_string(); let mut params = "".to_string(); let mut headers = HeaderMap::new(); if method == "POST" { body = params_json.to_string(); } if method == "GET" || method == "PUT" || method == "DELETE" { let z = params_json.to_string(); params = RestTool::parse_params_to_str(z); } //是否需要登录-- 组装sing if is_login { if !is_login_param { let e = ResponseData::error(self.tag.clone(), "登录参数错误!".to_string()); return e; } else { //需要登录-且登录参数齐全 trace!("Path:{}{}", prefix_url.clone(),request_url.clone()); trace!("Query:{}", params); trace!("Body:{}", body); trace!("expire:{}", timestamp.to_string()); //组装sing let sing = Self::sign(secret_key.clone(), prefix_url.clone(), request_url.clone(), params.clone(), body.clone(), timestamp.to_string(), ); trace!("Signature:{}", sing); //组装header headers.extend(Self::headers(sing, timestamp.to_string(), access_key)); } } let start_time = chrono::Utc::now().timestamp_millis(); let response = self.http_tool( format!("{}{}", prefix_url.clone(), request_url.clone()), method.to_string(), params.clone(), body.clone(), headers, is_login, ).await; let time_array = chrono::Utc::now().timestamp_millis() - start_time; self.delays.push(time_array); self.get_delay_info(); response } pub fn headers(sign: String, timestamp: String, access_key: String) -> HeaderMap { let mut headers = HeaderMap::new(); headers.insert("x-phemex-access-token", access_key.parse().unwrap());// 这是 Phemex 网站的 API-KEY(id 字段) headers.insert("x-phemex-request-expiry", timestamp.parse().unwrap());// 描述请求过期的 Unix EPoch 秒数,通常应为 (Now() + 1 分钟) headers.insert("x-phemex-request-signature", sign.parse().unwrap());// 这是 http 请求的 HMAC SHA256 签名。Secret 是 API Secret,其公式为:HMacSha256(URL Path + QueryString + Expiry + body) // let tracing = format!("4l_{:?}", Utc::now().timestamp_millis().to_string()); // headers.insert("x-phemex-request-tracing", tracing.parse().unwrap()); headers } pub fn sign(secret_key: String, prefix_url: String, request_url: String, url_param_str: String, body: String, timestamp: String) -> String { /*签名生成*/ let base_url = format!("{}{}", prefix_url, request_url); // HMacSha256(URL Path + QueryString + Expiry + body) // 时间戳 + 请求类型+ 请求参数字符串 let message = format!("{}{}{}{}", base_url, url_param_str, timestamp, body); trace!("message:{}",message); // 做签名 let hmac_key = ring::hmac::Key::new(hmac::HMAC_SHA256, secret_key.as_bytes()); let sign = hex::encode(hmac::sign(&hmac_key, message.as_bytes()).as_ref()); sign } // async fn http_tool(&mut self, request_path: String, request_type: String, params: String, headers: HeaderMap) -> Result { async fn http_tool(&mut self, request_path: String, request_type: String, params: String, body: String, headers: HeaderMap, is_login: bool) -> ResponseData { /****请求接口与 地址*/ let url = format!("{}{}", self.base_url.to_string(), request_path); let request_type = request_type.clone().to_uppercase(); let addrs_url: String = if params == "" { url.clone() } else { format!("{}?{}", url.clone(), params) }; trace!("url-----:{}",url.clone()); trace!("addrs_url-----:{}",addrs_url.clone()); trace!("params-----:{}",params.clone()); trace!("body-----:{}",body.clone()); trace!("headers-----:{:?}",headers.clone()); trace!("request_type-----:{:?}",request_type.clone()); let client = if is_login { let params = proxy::ParsingDetail::http_enable_proxy(Some("phemex")); let client_re; if params { let proxy_address = "socks5://127.0.0.1:17890"; // 替换为你的 SOCKS5 代理地址 let proxy = Proxy::all(proxy_address).unwrap(); client_re = Client::builder().proxy(proxy).build().unwrap(); } else { client_re = Client::new() } client_re } else { proxy::ParsingDetail::http_enable_proxy(None); Client::new() }; let request_builder = match request_type.as_str() { "GET" => client.get(addrs_url.clone()).headers(headers), "POST" => client.post(url.clone()).body(body).headers(headers), "DELETE" => client.delete(addrs_url.clone()).headers(headers), "PUT" => client.put(addrs_url.clone()).headers(headers), _ => { panic!("{}", format!("错误的请求类型:{}", request_type.clone())) } }; // 读取响应的内容 let response = request_builder.send().await.unwrap(); let is_success = response.status().is_success(); // 先检查状态码 let text = response.text().await.unwrap(); // trace!("text:{:?}",text); return if is_success { self.on_success_data(&text) } else { self.on_error_data(&text, &addrs_url, ¶ms) }; } pub fn on_success_data(&mut self, text: &String) -> ResponseData { let json_value = serde_json::from_str::(&text).unwrap(); // return ResponseData::new(self.tag.clone(), 200, "success".to_string(), json_value); let id = json_value["id"].as_i64(); match id { None => {} Some(v) => { match v { 0 => { let result = json_value.get("result").unwrap(); return ResponseData::new(self.tag.clone(), 200, "success".to_string(), result.clone()); } _ => {} } } } let code = json_value["code"].as_i64(); match code { None => {} Some(v) => { match v { 0 => { //判断是否有code ,没有表示特殊接口,直接返回 if json_value.get("data").is_some() { let data = json_value.get("data").unwrap(); return ResponseData::new(self.tag.clone(), 200, "success".to_string(), data.clone()); } else { return ResponseData::new(self.tag.clone(), 200, "success".to_string(), json_value); } } _ => { if json_value.get("msg").is_some() { return ResponseData::new(self.tag.clone(), 400, format!("{:?}", json_value["msg"].as_str()), json_value); } else { return ResponseData::new(self.tag.clone(), 400, "error".to_string(), json_value); } } } } } return ResponseData::new(self.tag.clone(), 400, "error".to_string(), json_value); } pub fn on_error_data(&mut self, text: &String, _base_url: &String, _params: &String) -> ResponseData { let json_value = serde_json::from_str::(&text); match json_value { Ok(data) => { let code = data["code"].as_i64(); match code { None => {} Some(v) => { match v { _ => { if data.get("msg").is_some() { return ResponseData::new(self.tag.clone(), 400, format!("{:?}", data["msg"].as_str().unwrap()), data); } else { return ResponseData::new(self.tag.clone(), 400, "error".to_string(), data); } } } } } } Err(e) => { error!("解析错误:{:?}", e); return ResponseData::error("".to_string(), format!("json 解析失败:{},相关参数:{}", e, text)); } } return ResponseData::error("".to_string(), format!("请求失败:{:?}", text)); } }