strategy.rs 56 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338
  1. use std::cmp::{max, min};
  2. use std::collections::HashMap;
  3. use std::ops::{Div, Mul};
  4. use std::str::FromStr;
  5. use chrono::Utc;
  6. use rust_decimal::Decimal;
  7. use rust_decimal::prelude::{FromPrimitive, ToPrimitive};
  8. use rust_decimal_macros::dec;
  9. use crate::model::{DealRecord, LocalPosition, OrderInfo, TraderMsg};
  10. use crate::utils;
  11. use tracing::{info, error, warn, instrument};
  12. use reqwest::{Client};
  13. use tokio::spawn;
  14. use global::params::Params;
  15. use standard::{OrderCommand};
  16. #[derive(Debug)]
  17. pub struct Strategy {
  18. // 各类时间戳和时延,我们都改成了毫秒级
  19. pub _print_time: i64, // 上次打印时间
  20. pub _print_interval: i64, // 打印时延
  21. pub _start_time: i64, // 开始时间
  22. pub local_time: i64, // 本地时间
  23. pub local_start_time: i64, // 本地开始时间
  24. pub post_open_time: i64, // 上次提交订单的时间戳
  25. pub post_open_interval: i64, // 提交订单时延
  26. pub _check_local_orders_time: i64, // 上次查单时间
  27. pub _check_local_orders_interval: i64, // 查单间距,原文是秒级,这里改成毫秒级
  28. pub in_cancel: HashMap<String, i64>, // 撤单队列
  29. pub cancel_wait_interval: i64, // 取消等待时延
  30. pub in_check: HashMap<String, i64>, // 查单队列
  31. pub check_wait_interval: i64, // 检测时延
  32. pub request_limit_check_time: i64, // 上次检查订单的时间
  33. pub request_limit_check_interval: i64, // 原文是秒级,这里改成毫秒级
  34. pub request_count: i64, // 记录请求次数,原文的request_num
  35. pub request_order_count: i64, // 记录下单次数,原文的request_order_num
  36. pub request_over_log_interval: i64, // 两次超时打印之间的间隔
  37. pub request_over_log_time: i64, // 上次打印时间
  38. pub limit_requests_num: i64, // 单位(时延)时间内请求次数上限
  39. pub limit_order_requests_num: i64, // 单位(时延)时间内下单次数上限
  40. pub _req_num_per_window: i64, // 单位(时延)时间内请求上限窗口
  41. pub params: Params, //
  42. pub exchange: String, //
  43. pub broker_id: String, //
  44. pub trade_name: String, //
  45. pub ref_exchange_length: usize, //
  46. pub ref_name: Vec<String>, //
  47. pub maker_mode: String, //
  48. pub local_orders: HashMap<String, OrderInfo>, // 本地订单
  49. pub pos: LocalPosition, //
  50. pub long_hold_value: Decimal, //
  51. pub short_hold_value: Decimal, //
  52. pub equity: Decimal, //
  53. pub coin: Decimal, //
  54. pub cash: Decimal, //
  55. pub start_equity: Decimal, //
  56. pub start_coin: Decimal, //
  57. pub start_cash: Decimal, //
  58. pub max_equity: Decimal, //
  59. pub local_profit: Decimal, //
  60. pub total_amount: Decimal, //
  61. pub is_ready: bool, // 程序是否已经准备好,ready
  62. pub _is_print: bool, //
  63. pub _min_amount_value: Decimal, //
  64. pub _max_amount_value: Decimal, //
  65. pub mp_ema: Decimal, // 原文的mp_ewma
  66. pub mp: Decimal, //
  67. pub bp: Decimal, //
  68. pub ap: Decimal, //
  69. pub ref_price: Decimal, //
  70. pub ref_bp: Decimal, //
  71. pub ref_ap: Decimal, //
  72. pub step_size: Decimal, // 原文的stepSize
  73. pub tick_size: Decimal, // 原文的tickSize
  74. pub max_pos_rate: Decimal, // 原文的maxPos,其实是最大持仓比例
  75. pub profit: Decimal, //
  76. pub daily_return: Decimal, //
  77. pub adjust_lever_rate: Decimal, // 原文的adjust_leverrate
  78. pub lever_rate: Decimal, // 原文的leverrate
  79. pub long_pos_bias: Decimal, //
  80. pub short_pos_bias: Decimal, //
  81. pub long_hold_rate: Decimal, //
  82. pub short_hold_rate: Decimal, //
  83. pub max_long_value: Decimal, // 最大做多持仓
  84. pub max_short_value: Decimal, // 最大做空持仓
  85. pub open_dist: Vec<Decimal>, // 开仓相关价格
  86. pub close_dist: Vec<Decimal>, // 平仓相关价格
  87. pub trade_close_dist: Decimal, //
  88. pub trade_open_dist: Decimal, //
  89. pub ref_index: usize, //
  90. pub predict: Decimal, //
  91. pub predict_alpha: Decimal, //
  92. pub post_side: i64, // 交易方向
  93. pub trade_vol_24h_w: Decimal, // 24小时成交额(单位:万)
  94. pub grid: Decimal, // 网格数量
  95. }
  96. impl Strategy {
  97. pub fn new(params: &Params, is_print: bool) -> Self {
  98. if params.ref_exchange.len() != params.ref_pair.len(){
  99. error!("参考盘口数不等于参考品种数,退出,请检查配置!");
  100. panic!("参考盘口数不等于参考品种数,退出,请检查配置!");
  101. }
  102. // strategy的初始化,里面已经有一些参数初始化了
  103. let mut strategy = Self {
  104. _print_time: 0,
  105. _start_time: 0,
  106. local_time: 0,
  107. local_start_time: 0,
  108. request_count: 0,
  109. request_order_count: 0,
  110. request_over_log_interval: 60 * 1000,
  111. request_over_log_time: 0,
  112. _print_interval: 5 * 1000,
  113. in_cancel: Default::default(),
  114. cancel_wait_interval: (0.2 * 1000f64).to_i64().unwrap(),
  115. in_check: Default::default(),
  116. check_wait_interval: 10 * 1000,
  117. _check_local_orders_time: 0,
  118. _check_local_orders_interval: 0,
  119. request_limit_check_time: 0,
  120. request_limit_check_interval: 0,
  121. limit_requests_num: 0,
  122. limit_order_requests_num: 0,
  123. _req_num_per_window: 0,
  124. post_open_time: 0,
  125. post_open_interval: 0,
  126. params: params.clone(),
  127. exchange: params.exchange.clone(),
  128. broker_id: params.broker_id.clone(),
  129. trade_name: "".to_string(),
  130. ref_exchange_length: params.ref_exchange.len(),
  131. ref_name: vec![],
  132. maker_mode: "free".to_string(),
  133. local_orders: Default::default(),
  134. pos: LocalPosition {
  135. long_pos: Default::default(),
  136. short_pos: Default::default(),
  137. long_avg: Default::default(),
  138. short_avg: Default::default(),
  139. },
  140. long_hold_value: Default::default(),
  141. short_hold_value: Default::default(),
  142. equity: Default::default(),
  143. coin: Default::default(),
  144. cash: Default::default(),
  145. start_equity: Default::default(),
  146. start_coin: Default::default(),
  147. start_cash: Default::default(),
  148. max_equity: Default::default(),
  149. local_profit: Default::default(),
  150. total_amount: Default::default(),
  151. is_ready: false,
  152. _is_print: is_print,
  153. _min_amount_value: dec!(5.0),
  154. _max_amount_value: dec!(10000.0),
  155. mp_ema: Default::default(),
  156. mp: Default::default(),
  157. bp: Default::default(),
  158. ap: Default::default(),
  159. ref_price: Default::default(),
  160. ref_bp: Default::default(),
  161. ref_ap: Default::default(),
  162. step_size: dec!(1e-10),
  163. tick_size: dec!(1e-10),
  164. max_pos_rate: Default::default(),
  165. profit: Default::default(),
  166. daily_return: Default::default(),
  167. adjust_lever_rate: Decimal::ONE,
  168. lever_rate: Default::default(),
  169. long_pos_bias: Default::default(),
  170. short_pos_bias: Default::default(),
  171. long_hold_rate: Default::default(),
  172. short_hold_rate: Default::default(),
  173. max_long_value: Default::default(),
  174. max_short_value: Default::default(),
  175. open_dist: vec![],
  176. close_dist: vec![],
  177. trade_close_dist: params.close,
  178. trade_open_dist: params.open,
  179. ref_index: 0,
  180. predict: Default::default(),
  181. predict_alpha: Default::default(),
  182. post_side: 0,
  183. trade_vol_24h_w: Default::default(),
  184. grid: Decimal::from(params.grid),
  185. };
  186. // 交易名字
  187. strategy.trade_name = format!("{}@{}", params.exchange.clone(), params.pair.clone());
  188. // 参考交易所的trade_name
  189. for index in 0..strategy.ref_exchange_length {
  190. strategy.ref_name.push(format!("{}@{}", params.ref_exchange[index], params.ref_pair[index]));
  191. }
  192. // 杠杆比例处理
  193. strategy.lever_rate = params.lever_rate;
  194. if strategy.exchange.contains("spot") {
  195. strategy.lever_rate = min(params.lever_rate, Decimal::ONE);
  196. }
  197. // 各类时间戳
  198. let now = Utc::now();
  199. strategy.local_time = now.timestamp_millis();
  200. strategy.local_start_time = now.timestamp_millis();
  201. strategy._print_time = now.timestamp_millis();
  202. strategy._start_time = now.timestamp_millis();
  203. // 检查订单的时间戳
  204. strategy._check_local_orders_time = now.timestamp_millis();
  205. strategy._check_local_orders_interval = 10 * 1000;
  206. // 下单的相关限制处理
  207. strategy.request_limit_check_time = now.timestamp_millis();
  208. strategy.request_limit_check_interval = 10 * 1000;
  209. // 求得正常请求数量和下单请求数量(interval时间内)
  210. let request_limit_check_interval_per_second = strategy.request_limit_check_interval / 1000;
  211. strategy.limit_requests_num = utils::get_limit_requests_num_per_second(params.exchange.clone()) * (request_limit_check_interval_per_second);
  212. strategy.limit_order_requests_num = utils::get_limit_order_requests_num_per_second(params.exchange.clone()) * (request_limit_check_interval_per_second);
  213. // 开仓下单间隔 均匀下单机会
  214. strategy.post_open_time = now.timestamp_millis();
  215. let post_open_interval_per_second = Decimal::ONE.div(Decimal::from_i64(utils::get_limit_order_requests_num_per_second(params.exchange.clone())).unwrap());
  216. strategy.post_open_interval = dec!(1000).mul(post_open_interval_per_second).to_i64().unwrap();
  217. info!("策略模块初始化完成!");
  218. return strategy;
  219. }
  220. // 更新当前strategy的各类信息
  221. #[instrument(skip(self, trader_msg), level="TRACE")]
  222. pub fn _update_data(&mut self, trader_msg: &TraderMsg) -> bool {
  223. // debug!(?self);
  224. // debug!(?trader_msg);
  225. self.local_orders.clear();
  226. self.local_orders = trader_msg.orders.clone();
  227. // position信息更新
  228. if self.pos.long_pos != trader_msg.position.long_pos {
  229. self.pos.long_pos = trader_msg.position.long_pos;
  230. self.pos.long_avg = trader_msg.position.long_avg;
  231. }
  232. if self.pos.short_pos != trader_msg.position.short_pos {
  233. self.pos.short_pos = trader_msg.position.short_pos;
  234. self.pos.short_avg = trader_msg.position.short_avg;
  235. }
  236. // debug!(?self.pos);
  237. // 价格值处理
  238. self.bp = trader_msg.market[global::public_params::BID_PRICE_INDEX];
  239. self.ap = trader_msg.market[global::public_params::ASK_PRICE_INDEX];
  240. self.mp = (self.bp + self.ap) * dec!(0.5);
  241. // 中间价的ema值处理
  242. if self.mp_ema.eq(&Decimal::ZERO) {
  243. self.mp_ema = self.mp;
  244. } else {
  245. self.mp_ema = self.mp_ema * dec!(0.999) + self.mp * dec!(0.001);
  246. }
  247. // debug!(?self.bp, ?self.ap, ?self.mp, ?self.mp_ema);
  248. // 动态杠杆调节
  249. if self.mp > self.mp_ema {
  250. self.adjust_lever_rate = Decimal::ONE;
  251. } else {
  252. self.adjust_lever_rate = dec!(0.8);
  253. }
  254. // debug!(?self.adjust_lever_rate);
  255. // 当前持仓价值处理
  256. self.long_hold_value = self.pos.long_pos * self.mp;
  257. self.short_hold_value = self.pos.short_pos * self.mp;
  258. // debug!(?self.long_hold_value, ?self.short_hold_value);
  259. // 分现货或合约计算最大开仓价值
  260. if self.exchange.contains("spot") {
  261. self.max_long_value = trader_msg.cash * self.lever_rate * self.adjust_lever_rate;
  262. self.max_short_value = trader_msg.coin * self.lever_rate * self.adjust_lever_rate * self.mp;
  263. } else {
  264. self.max_long_value = self.equity * self.lever_rate * self.adjust_lever_rate;
  265. self.max_short_value = self.max_long_value;
  266. }
  267. // debug!(?self.max_long_value, ?self.max_short_value, ?self.equity, ?self.lever_rate, ?self.adjust_lever_rate);
  268. // 做市模式识别
  269. if self.ref_name[self.ref_index].eq(&self.trade_name) {
  270. self.maker_mode = "free".to_string();
  271. } else {
  272. self.maker_mode = "follow".to_string();
  273. }
  274. // debug!(?self.maker_mode);
  275. // 参考价格
  276. if trader_msg.ref_price.len() == 0 {
  277. self.ref_bp = self.bp;
  278. self.ref_ap = self.ap;
  279. self.ref_price = self.mp;
  280. } else {
  281. self.ref_bp = trader_msg.ref_price[self.ref_index][0];
  282. self.ref_ap = trader_msg.ref_price[self.ref_index][1];
  283. self.ref_price = (self.ref_bp + self.ref_ap) * dec!(0.5);
  284. }
  285. // debug!(?self.ref_bp, ?self.ref_ap, %self.ref_price);
  286. // spread
  287. let temp_predict = trader_msg.predict * self.predict_alpha;
  288. self.predict = utils::clip(temp_predict, -self.trade_open_dist, self.trade_open_dist);
  289. // debug!(?self.predict);
  290. // 计算当前账户cash和coin
  291. self.coin = trader_msg.coin;
  292. self.cash = trader_msg.cash;
  293. self.equity = trader_msg.cash + trader_msg.coin * self.mp;
  294. if self.equity > self.max_equity {
  295. self.max_equity = self.equity;
  296. }
  297. // debug!(?self.coin, ?self.cash, ?self.equity, ?self.max_equity);
  298. // 总可开数量
  299. self.total_amount = self.equity * self.lever_rate * self.adjust_lever_rate / self.mp;
  300. self.total_amount = utils::fix_amount(self.total_amount, self.step_size);
  301. // debug!(?self.total_amount);
  302. if self.total_amount.eq(&Decimal::ZERO) {
  303. error!("总可开数量低于一张,请尝试加大杠杆倍数或资金!equity={}, lever_rate={}, adjust_lever_rate={}, mp={}, step_size={}",
  304. self.equity, self.lever_rate, self.adjust_lever_rate, self.mp, self.step_size);
  305. return false;
  306. }
  307. // 求最大pos
  308. if self.equity > Decimal::ZERO {
  309. let max_pos_rate = max(self.pos.long_pos, self.pos.short_pos) * self.mp / self.equity;
  310. if max_pos_rate > self.max_pos_rate {
  311. self.max_pos_rate = max_pos_rate;
  312. }
  313. // debug!(?max_pos_rate, ?self.max_pos_rate);
  314. }
  315. return true;
  316. }
  317. // 打印状态信息
  318. // 耗时700微秒
  319. #[instrument(skip(self), level="TRACE")]
  320. pub fn _print_summary(&mut self) {
  321. self.mp.rescale(6);
  322. self.ref_price.rescale(6);
  323. self.equity.rescale(3);
  324. self.cash.rescale(3);
  325. let mut value = self.coin * self.mp;
  326. value.rescale(3);
  327. let mut price_bias = Decimal::ONE_HUNDRED * (self.ref_price - self.mp) / self.mp;
  328. price_bias.rescale(2);
  329. // 盈亏计算
  330. self.profit = if self.start_equity.gt(&Decimal::ZERO) {
  331. ((self.equity - self.start_equity) / self.start_equity) * Decimal::ONE_HUNDRED
  332. } else {
  333. Decimal::ZERO
  334. };
  335. self.profit.rescale(2);
  336. // 多仓杠杆计算
  337. let mut long_pos_leverage = if self.equity.gt(&Decimal::ZERO) {
  338. self.pos.long_pos * self.mp / self.equity
  339. } else {
  340. Decimal::ZERO
  341. };
  342. long_pos_leverage.rescale(3);
  343. // 多仓浮盈计算
  344. self.long_pos_bias = if self.pos.long_pos.gt(&Decimal::ZERO) {
  345. Decimal::ONE_HUNDRED - Decimal::ONE_HUNDRED * self.pos.long_avg / self.mp
  346. } else {
  347. Decimal::ZERO
  348. };
  349. self.long_pos_bias.rescale(2);
  350. // 空仓杠杆计算
  351. let mut short_pos_leverage = if self.equity.gt(&Decimal::ZERO) {
  352. self.pos.short_pos * self.mp / self.equity
  353. } else {
  354. Decimal::ZERO
  355. };
  356. short_pos_leverage.rescale(3);
  357. // 多仓浮盈计算
  358. self.short_pos_bias = if self.pos.short_pos.gt(&Decimal::ZERO) {
  359. Decimal::ONE_HUNDRED - Decimal::ONE_HUNDRED * self.pos.short_avg / self.mp
  360. } else {
  361. Decimal::ZERO
  362. };
  363. let run_time = Utc::now().timestamp_millis() - self._start_time;
  364. let run_time_day = Decimal::from(run_time) / (dec!(86400000));
  365. self.daily_return = self.profit / run_time_day;
  366. self.daily_return.rescale(2);
  367. self.short_pos_bias.rescale(2);
  368. self.trade_open_dist.rescale(6);
  369. self.trade_close_dist.rescale(6);
  370. self.predict.rescale(5);
  371. // 挂单列表长度
  372. let o_num = self.local_orders.len();
  373. let mut msg = String::new();
  374. msg.push_str("当前状态: ");
  375. msg.push_str(format!("[品种 {}, 现价 {:?}, 定价 {:?}, 偏差 {:?}%, 杠杆 {:?}, 动态{:?}, 最大{:?}, 预测 {:?}, 预估24H成交额 {:?}万], ",
  376. self.params.pair, self.mp, self.ref_price, price_bias, self.lever_rate,
  377. self.adjust_lever_rate, self.max_pos_rate, self.predict, self.trade_vol_24h_w).as_str());
  378. msg.push_str(format!("[净值 {:?}, Cash {:?}, Coin(价值) {:?}, 日化 {:?}%], ", self.equity, self.cash, value, self.daily_return).as_str());
  379. msg.push_str(format!("[推算利润 {:?}, 盈亏 {:?}%, 做多杠杆 {:?}%, 做多浮盈 {:?}%, 做空杠杆 {:?}%, 做空浮盈 {:?}%], ",
  380. self.local_profit, self.profit, long_pos_leverage, self.long_pos_bias, short_pos_leverage, self.short_pos_bias).as_str());
  381. msg.push_str(format!("[请求 {:?}, 上限{:?}次/10秒], ", self._req_num_per_window, self.limit_order_requests_num).as_str());
  382. msg.push_str(format!("[当前参数, 开仓 {:?}, 平仓 {:?}, 方向 {:?}, 参考 {:?}, 模式 {:?}], ",
  383. self.trade_open_dist, self.trade_close_dist, self.post_side, self.ref_name[self.ref_index], self.maker_mode).as_str());
  384. msg.push_str(format!("[挂单列表,共{:?}单, ", o_num).as_str());
  385. for (_, order) in &self.local_orders {
  386. let mut order_value = order.amount * self.mp;
  387. let mut order_lever_rate = if self.equity.gt(&Decimal::ZERO) {
  388. order.amount * self.mp / self.equity
  389. } else {
  390. Decimal::ZERO
  391. };
  392. let mut order_bias = Decimal::ONE_HUNDRED * (order.price - self.mp) / self.mp;
  393. order_value.rescale(2);
  394. order_lever_rate.rescale(3);
  395. order_bias.rescale(3);
  396. msg.push_str(format!("[{:?} {:?} {:?}, 杠杆{:?}X 价值{:?}U 价格{:?} 偏离{:?}%]",
  397. order.symbol, order.client_id, order.side, order_lever_rate, order_value, order.price, order_bias).as_str());
  398. }
  399. msg.push_str("]");
  400. info!("{}", msg);
  401. }
  402. // 取消目标方向订单,原文是_cancel_targit_side_orders
  403. #[instrument(skip(self, command), level="TRACE")]
  404. pub fn _cancel_target_side_orders(&self, command: &mut OrderCommand) {
  405. // 要取消的目标方向
  406. let target_side = vec![
  407. "kd".to_string(),
  408. "kk".to_string(),
  409. "pd".to_string(),
  410. "pk".to_string()
  411. ];
  412. // debug!(?self.local_orders);
  413. for client_id in self.local_orders.keys() {
  414. let order = self.local_orders.get(client_id).unwrap();
  415. // 如果不属于目标方向,则不需要取消
  416. if !target_side.contains(&order.side.clone()) {
  417. continue;
  418. }
  419. // 属于目标方向,则取消
  420. let key = format!("Cancel{}", client_id);
  421. let value = vec![order.client_id.clone(), order.order_id.clone()];
  422. command.cancel.insert(key, value);
  423. }
  424. // debug!(?command);
  425. }
  426. // 生成各类挂单价格,原文是gen_dist
  427. #[instrument(skip(self), level="TRACE")]
  428. pub fn generate_dist(&mut self) {
  429. let open = self.trade_open_dist;
  430. let close = self.trade_close_dist;
  431. let pos_rate = vec![self.long_hold_rate, self.short_hold_rate];
  432. let ref_bp = self.ref_bp;
  433. let ref_ap = self.ref_ap;
  434. let predict = self.predict;
  435. let grid = self.grid;
  436. let mode = self.maker_mode.clone();
  437. // 平仓相关
  438. let mut mp = (ref_bp + ref_ap) * dec!(0.5);
  439. let mut buy_start = mp;
  440. let mut sell_start = mp;
  441. let mut avoid = min(dec!(0.0005), close * dec!(0.5));
  442. let mut close_dist = vec![
  443. buy_start * (Decimal::ONE + predict - close + avoid), // buy upper
  444. buy_start * (Decimal::ONE + predict - close - avoid), // buy lower
  445. sell_start * (Decimal::ONE + predict + close - avoid), // sell lower
  446. sell_start * (Decimal::ONE + predict + close + avoid), // sell upper
  447. ];
  448. // debug!(?mp, ?buy_start, ?sell_start, ?avoid, ?close_dist);
  449. // 自由做市模式
  450. if mode == "free".to_string() {
  451. buy_start = ref_bp;
  452. sell_start = ref_ap;
  453. }
  454. // 跟随做市模式
  455. else if mode == "follow".to_string() {
  456. mp = (ref_bp + ref_ap) * dec!(0.5);
  457. buy_start = mp;
  458. sell_start = mp;
  459. } else {
  460. error!("未知做市类型:mode={}", mode);
  461. panic!("未知做市类型:mode={}", mode);
  462. }
  463. // debug!(?mode, ?buy_start, ?sell_start, ?mp);
  464. // 开仓相关
  465. avoid = min(dec!(0.001), open * dec!(0.05));
  466. // 持仓偏移
  467. let buy_shift = Decimal::ONE + pos_rate[0] * grid;
  468. let sell_shift = Decimal::ONE + pos_rate[1] * grid;
  469. let mut open_dist = vec![
  470. buy_start * (Decimal::ONE + predict - open * buy_shift + avoid), // buy upper
  471. buy_start * (Decimal::ONE + predict - open * buy_shift - avoid), // buy lower
  472. sell_start * (Decimal::ONE + predict + open * sell_shift - avoid), // sell lower
  473. sell_start * (Decimal::ONE + predict + open * sell_shift + avoid), // sell upper
  474. ];
  475. // debug!(?avoid, ?buy_shift, ?sell_shift, ?avoid, ?open_dist);
  476. // 修复价格
  477. for open_price in &mut open_dist {
  478. *open_price = utils::fix_price(*open_price, self.tick_size);
  479. }
  480. for close_price in &mut close_dist {
  481. *close_price = utils::fix_price(*close_price, self.tick_size);
  482. }
  483. self.open_dist = open_dist.clone();
  484. self.close_dist = close_dist.clone();
  485. // debug!(?open_dist);
  486. // debug!(?close_dist);
  487. }
  488. // 统计请求次数
  489. #[instrument(skip(self, command), level="TRACE")]
  490. pub fn _update_request_num(&mut self, command: &OrderCommand) {
  491. // debug!(?command);
  492. // debug!(?self.request_order_count, ?self.request_count);
  493. let order_count = (command.limits_open.len() + command.limits_close.len()).to_i64().unwrap();
  494. let request_count = order_count + (command.cancel.len() + command.check.len()).to_i64().unwrap();
  495. self.request_order_count += order_count;
  496. self.request_count += request_count;
  497. // debug!(?self.request_order_count, ?self.request_count);
  498. }
  499. // 根据平均请求次数限制开仓下单
  500. #[instrument(skip(self, command), level="TRACE")]
  501. pub fn _check_request_limit(&mut self, command: &mut OrderCommand) {
  502. let mut msg = String::new();
  503. // 如果当前请求数超过限制
  504. if self.request_count > self.limit_requests_num {
  505. command.cancel.clear();
  506. command.check.clear();
  507. command.limits_open.clear();
  508. command.limits_close.clear();
  509. msg = format!("请求频率溢出,程序禁止任何操作!({}/{})", self.request_count, self.limit_requests_num);
  510. } else if self.request_order_count >= self.limit_order_requests_num { // 100%超过下单频率,则不再进行平仓挂单
  511. command.limits_close.clear();
  512. command.limits_open.clear();
  513. msg = format!("超过100%下单频率!程序禁止开平仓!({}/{})", self.request_order_count, self.limit_order_requests_num);
  514. } else if self.request_count > self.limit_requests_num * 5 / 10
  515. && self.request_order_count > self.limit_order_requests_num * 8 / 10 { // 超过80%,直接取消limits_open的下单指令
  516. command.limits_open.clear();
  517. msg = format!("超过80%下单频率,程序禁止开仓!({}/{})", self.request_order_count, self.limit_order_requests_num);
  518. }
  519. // 检查是否需要打印msg
  520. if !msg.is_empty() && self.local_time - self.request_over_log_time > self.request_over_log_interval {
  521. warn!("{}", msg);
  522. self.request_over_log_time = self.local_time;
  523. }
  524. }
  525. // 新增正在撤单、检查撤单队列,释放过时限制
  526. #[instrument(skip(self), level="TRACE")]
  527. pub fn _update_in_cancel(&mut self, command: &mut OrderCommand) {
  528. let mut new_cancel: HashMap<String, Vec<String>> = HashMap::new();
  529. for cancel_name in command.cancel.keys() {
  530. let cancel = command.cancel.get(cancel_name).unwrap();
  531. let client_id = cancel[0].clone();
  532. let mut need_limit_cancel = true;
  533. let order_some = self.local_orders.get(&client_id);
  534. // 判断是否在本地挂单表中
  535. if let Some(order) = order_some {
  536. // 如果订单创建时间大于100ms,才能有撤单操作
  537. if self.local_time - order.create_time < 100 {
  538. need_limit_cancel = false;
  539. }
  540. }
  541. if need_limit_cancel {
  542. // 如果已经不在撤销队列里,增加到撤销队列
  543. if self.in_cancel.get(&client_id).is_none() {
  544. self.in_cancel.insert(client_id.clone(), self.local_time);
  545. new_cancel.insert(cancel_name.clone(), cancel.clone());
  546. }
  547. }
  548. }
  549. // debug!(?command);
  550. command.cancel = new_cancel;
  551. // debug!(?command);
  552. // 释放撤单限制
  553. self._release_in_cancel();
  554. }
  555. // 维护查单队列,检查是否在撤单
  556. #[instrument(skip(self), level="TRACE")]
  557. pub fn _release_in_check(&mut self) {
  558. // debug!(?self.in_check);
  559. // 为什么要移出来:Rust不允许边循环边修改map
  560. let mut to_remove = Vec::new();
  561. for client_id in self.in_check.keys() {
  562. let time = self.in_check.get(client_id).unwrap();
  563. // 等待不超时,就不移除
  564. if self.local_time - time <= self.check_wait_interval {
  565. continue;
  566. }
  567. // 等待超时,就移除正在撤单队列
  568. // debug!("移除查单队列:{}", client_id.clone());
  569. to_remove.push(client_id.clone());
  570. }
  571. // 在后面的循环中去单独处理map的更新
  572. for client_id in to_remove {
  573. self.in_check.remove(&client_id);
  574. }
  575. // debug!(?self.in_check);
  576. }
  577. // 检查是否正在撤单
  578. #[instrument(skip(self), level="TRACE")]
  579. pub fn _release_in_cancel(&mut self) {
  580. // debug!(?self.in_cancel);
  581. // 为什么要移出来:Rust不允许边循环边修改map
  582. let mut to_remove = Vec::new();
  583. for client_id in self.in_cancel.keys() {
  584. let time = self.in_cancel.get(client_id).unwrap();
  585. // 等待不超时,就不移除
  586. if self.local_time - time <= self.cancel_wait_interval {
  587. continue;
  588. }
  589. // 等待超时,就移除正在撤单队列
  590. // debug!("等待超过后移除正在撤单队列:{}", client_id.clone());
  591. to_remove.push(client_id.clone());
  592. }
  593. // 在后面的循环中去单独处理map的更新
  594. for client_id in to_remove {
  595. self.in_cancel.remove(&client_id);
  596. }
  597. // debug!(?self.in_cancel);
  598. }
  599. // 刷新请求限制
  600. pub fn _refresh_request_limit(&mut self) {
  601. if self.local_time - self.request_limit_check_time < self.request_limit_check_interval {
  602. return;
  603. }
  604. self._req_num_per_window = self.request_count;
  605. self.request_count = 0;
  606. self.request_order_count = 0;
  607. self.request_limit_check_time = self.local_time;
  608. }
  609. // 刷新持仓比例
  610. #[instrument(skip(self), level="TRACE")]
  611. pub fn _pos_rate(&mut self) {
  612. // debug!(?self);
  613. if self.max_long_value > Decimal::ZERO {
  614. self.long_hold_rate = self.long_hold_value / self.max_long_value;
  615. // debug!(?self.long_hold_rate);
  616. }
  617. if self.max_short_value > Decimal::ZERO {
  618. self.short_hold_rate = self.short_hold_value / self.max_short_value;
  619. // debug!(?self.short_hold_rate);
  620. }
  621. }
  622. // 当退出时调用,全撤全平 准备退出
  623. pub fn on_exit(&mut self, trader_msg: &TraderMsg) -> OrderCommand {
  624. let mut command = OrderCommand::new();
  625. if self._update_data(trader_msg) {
  626. if !self.check_ready() {
  627. return command;
  628. }
  629. // 取消、平掉所有
  630. self._close_all(&mut command);
  631. // 更新撤单队列
  632. self._update_in_cancel(&mut command);
  633. // 检查限频
  634. self._check_request_limit(&mut command);
  635. // 统计请求频率
  636. self._update_request_num(&mut command);
  637. }
  638. // debug!(?command);
  639. return command;
  640. }
  641. // 休眠时调用,全撤 不再下新订单了 防止影响check_position执行
  642. pub fn on_sleep(&mut self, trader_msg: &TraderMsg) -> OrderCommand {
  643. let mut command = OrderCommand::new();
  644. if self._update_data(trader_msg) {
  645. if !self.check_ready() {
  646. return command;
  647. }
  648. // 只是取消掉目标侧订单
  649. self._cancel_target_side_orders(&mut command);
  650. // 更新撤单队列
  651. self._update_in_cancel(&mut command);
  652. // 检查限频
  653. self._check_request_limit(&mut command);
  654. // 统计请求频率
  655. self._update_request_num(&mut command);
  656. }
  657. // debug!(?command);
  658. return command;
  659. }
  660. // 清空所有挂单和仓位保持休眠状态
  661. #[instrument(skip(self, command), level="TRACE")]
  662. pub fn _close_all(&self, command: &mut OrderCommand) {
  663. // 撤掉全部挂单
  664. let mut pd_amount = Decimal::ZERO;
  665. let mut pk_amount = Decimal::ZERO;
  666. // debug!(?self.local_orders);
  667. for client_id in self.local_orders.keys() {
  668. let order = self.local_orders.get(client_id).unwrap();
  669. // 命令生成
  670. let key = format!("Cancel{}", client_id);
  671. let value = vec![order.client_id.clone(), order.order_id.clone()];
  672. command.cancel.insert(key, value);
  673. // 统计部分
  674. if order.side == "pk".to_string() {
  675. pk_amount += order.amount;
  676. } else if order.side == "pd".to_string() {
  677. pd_amount += order.amount;
  678. }
  679. }
  680. // debug!(?pd_amount, ?pk_amount);
  681. // 批量挂单
  682. let need_close_long = self.pos.long_pos - pd_amount;
  683. let need_close_short = self.pos.short_pos - pk_amount;
  684. // debug!(?need_close_long, ?need_close_short);
  685. // 做多仓位平仓
  686. if need_close_long * self.mp > self._min_amount_value {
  687. let mut amount = need_close_long;
  688. // 现货要对数量精度进行限定处理
  689. if self.exchange.contains("spot") {
  690. amount = utils::fix_amount(amount, self.step_size);
  691. }
  692. let price = utils::fix_price(self.mp, self.tick_size);
  693. let client_id = utils::generate_client_id(Some(self.broker_id.clone()));
  694. let value = vec![
  695. amount.to_string(),
  696. "pd".to_string(),
  697. price.to_string(),
  698. client_id.to_string()
  699. ];
  700. command.limits_close.insert(client_id.clone(), value);
  701. // debug!(?self.pos.long_pos, ?self.mp, ?need_close_long, ?command)
  702. }
  703. // 做空仓位平仓
  704. if need_close_short * self.mp > self._min_amount_value {
  705. let mut amount = need_close_short;
  706. if self.exchange.contains("spot") {
  707. amount = utils::fix_amount(amount, self.step_size);
  708. }
  709. let price = utils::fix_price(self.mp, self.tick_size);
  710. let client_id = utils::generate_client_id(Some(self.broker_id.clone()));
  711. let value = vec![
  712. amount.to_string(),
  713. "pk".to_string(),
  714. price.to_string(),
  715. client_id.to_string()
  716. ];
  717. command.limits_close.insert(client_id.clone(), value);
  718. // debug!(?self.pos.short_pos, ?self.mp, ?need_close_short, ?command)
  719. }
  720. }
  721. // 检查是否完成准备,注意:原文是未准备完成返回true!!!!!!!!!!!!!!!!!!!
  722. pub fn check_ready(&mut self) -> bool {
  723. if self.is_ready {
  724. return true;
  725. }
  726. let pre_hot:i64 = 10 * 1000;
  727. if !self.mp.eq(&Decimal::ZERO) && self.local_time - self.local_start_time > pre_hot {
  728. self.is_ready = true;
  729. // debug!(?self.mp, ?self.local_time, ?self.local_start_time, ?pre_hot);
  730. info!("策略预热完毕,可以执行后续逻辑!")
  731. }
  732. return false;
  733. }
  734. // 接近整点时刻 不允许报单 防止下单bug
  735. pub fn check_allow_post_open(&self) -> bool {
  736. let local_time_second = self.local_time / 1000;
  737. let diff_time = local_time_second % (60 * 60);
  738. return diff_time > 30 && diff_time < 3570;
  739. }
  740. // 平仓订单处理命令
  741. #[instrument(skip(self, command), level="TRACE")]
  742. pub fn _post_close(&self, command: &mut OrderCommand) {
  743. // debug!(?command);
  744. let mut pd_amount = Decimal::ZERO;
  745. let mut pd_order_num = 0;
  746. let mut pk_amount = Decimal::ZERO;
  747. let mut pk_order_num = 0;
  748. // 各类价格
  749. let long_upper = self.close_dist[0];
  750. let long_lower = self.close_dist[1];
  751. let short_lower = self.close_dist[2];
  752. let short_upper = self.close_dist[3];
  753. // 获取当前挂单,如果超过挂单范围则取消当前平仓单
  754. for order_client_id in self.local_orders.keys() {
  755. let order = self.local_orders.get(order_client_id).unwrap();
  756. let key = format!("Cancel{}", (*order_client_id).clone());
  757. let value = vec![order.client_id.clone(), order.order_id.clone()];
  758. if order.side == "pk".to_string() {
  759. if order.price > long_upper || order.price < long_lower {
  760. command.cancel.insert(key.clone(), value.clone());
  761. }
  762. pk_amount += order.amount;
  763. pk_order_num += 1;
  764. }
  765. if order.side == "pd".to_string() {
  766. if order.price < short_lower || order.price > short_upper {
  767. command.cancel.insert(key.clone(), value.clone());
  768. }
  769. pd_amount += order.amount;
  770. pd_order_num += 1;
  771. }
  772. }
  773. // debug!(?command);
  774. // 判断是否需要全平
  775. let is_need_cancel_all_close =
  776. (pd_amount - self.pos.long_pos).abs() * self.mp > self._min_amount_value
  777. || (pk_amount - self.pos.short_pos).abs() * self.mp > self._min_amount_value;
  778. if is_need_cancel_all_close {
  779. for order_client_id in self.local_orders.keys() {
  780. let order = self.local_orders.get(order_client_id).unwrap();
  781. if order.side == "pk".to_string() || order.side == "pd".to_string() {
  782. let key = format!("Cancel{}", (*order_client_id).clone());
  783. let value = vec![order.client_id.clone(), order.order_id.clone()];
  784. command.cancel.insert(key.clone(), value.clone());
  785. }
  786. }
  787. }
  788. // debug!(?command);
  789. // 区分现货和期货
  790. if self.exchange.contains("spot") {
  791. // 平多价值大于最小交易价值,执行平多逻辑
  792. if self.pos.long_pos * self.mp > self._min_amount_value {
  793. if pd_order_num == 0 {
  794. let mut price = (short_lower + short_upper) * dec!(0.5);
  795. price = utils::clip(price, self.bp * dec!(0.9995), self.ap * dec!(1.03));
  796. price = utils::fix_price(price, self.tick_size);
  797. let mut amount = self.pos.long_pos;
  798. amount = utils::fix_amount(amount, self.step_size);
  799. if amount * price > self._min_amount_value {
  800. let order_client_id = utils::generate_client_id(Some(self.broker_id.clone()));
  801. let order = vec![
  802. amount.to_string(),
  803. "pd".to_string(),
  804. price.to_string(),
  805. order_client_id.clone()
  806. ];
  807. command.limits_close.insert(order_client_id, order.clone());
  808. // debug!(?command);
  809. }
  810. }
  811. }
  812. // 平空价值大于最小交易价值,执行平空逻辑
  813. if self.pos.short_pos > self._min_amount_value {
  814. if pk_order_num == 0 {
  815. let mut price = (long_upper + long_lower) * dec!(0.5);
  816. price = utils::clip(price, self.bp * dec!(0.97), self.ap * dec!(1.0005));
  817. price = utils::fix_price(price, self.tick_size);
  818. let mut amount = self.pos.short_pos;
  819. amount = utils::fix_amount(amount, self.step_size);
  820. if amount * price > self._min_amount_value {
  821. let order_client_id = utils::generate_client_id(Some(self.broker_id.clone()));
  822. let order = vec![
  823. amount.to_string(),
  824. "pk".to_string(),
  825. price.to_string(),
  826. order_client_id.clone()
  827. ];
  828. command.limits_close.insert(order_client_id, order.clone());
  829. // debug!(?command);
  830. }
  831. }
  832. }
  833. } else {
  834. if self.pos.long_pos > Decimal::ZERO {
  835. if pd_order_num == 0 {
  836. let mut price = (short_lower + short_upper) * dec!(0.5);
  837. price = utils::clip(price, self.bp * dec!(0.9995), self.ap * dec!(1.03));
  838. price = utils::fix_price(price, self.tick_size);
  839. let order_client_id = utils::generate_client_id(Some(self.broker_id.clone()));
  840. let order = vec![
  841. self.pos.long_pos.to_string(),
  842. "pd".to_string(),
  843. price.to_string(),
  844. order_client_id.clone()
  845. ];
  846. command.limits_close.insert(order_client_id, order.clone());
  847. // debug!(?command);
  848. }
  849. }
  850. if self.pos.short_pos > Decimal::ZERO {
  851. if pk_order_num == 0 {
  852. let mut price = (long_upper + long_lower) * dec!(0.5);
  853. price = utils::clip(price, self.bp * dec!(0.97), self.ap * dec!(1.0005));
  854. price = utils::fix_price(price, self.tick_size);
  855. let order_client_id = utils::generate_client_id(Some(self.broker_id.clone()));
  856. let order = vec![
  857. self.pos.short_pos.to_string(),
  858. "pk".to_string(),
  859. price.to_string(),
  860. order_client_id.clone()
  861. ];
  862. command.limits_close.insert(order_client_id, order.clone());
  863. // debug!(?command);
  864. }
  865. }
  866. }
  867. }
  868. // 生成取消订单的指令
  869. #[instrument(skip(self, command), level="TRACE")]
  870. pub fn _cancel_open(&self, command: &mut OrderCommand) {
  871. // debug!(?command);
  872. // 挂单范围
  873. let long_upper = self.open_dist[0];
  874. let long_lower = self.open_dist[1];
  875. let short_lower = self.open_dist[2];
  876. let short_upper = self.open_dist[3];
  877. for order_client_id in self.local_orders.keys() {
  878. let order = self.local_orders.get(order_client_id).unwrap();
  879. let key = format!("Cancel{}", (*order_client_id).clone());
  880. let value = vec![order.client_id.clone(), order.order_id.clone()];
  881. // 开多订单处理
  882. if order.side == "kd".to_string() {
  883. // 在价格范围内时不处理
  884. if order.price <= long_upper && order.price >= long_lower {
  885. continue
  886. }
  887. // debug!(?key, ?order.price, ?long_upper, ?long_lower);
  888. command.cancel.insert(key.clone(), value.clone());
  889. }
  890. // 开空订单处理
  891. if order.side == "kk".to_string() {
  892. // 在价格范围内时不处理
  893. if order.price >= short_lower && order.price <= short_upper {
  894. continue
  895. }
  896. // debug!(?key, ?order.price, ?short_lower, ?short_upper);
  897. command.cancel.insert(key.clone(), value.clone());
  898. }
  899. }
  900. }
  901. // 超时触发查单信号
  902. #[instrument(skip(self, command), level="TRACE")]
  903. pub fn _check_local_orders(&mut self, command: &mut OrderCommand) {
  904. // debug!(?command);
  905. // 超时检测
  906. if self.local_time - self._check_local_orders_time < self._check_local_orders_interval {
  907. return;
  908. }
  909. // 查单指令生成主逻辑
  910. for client_id in self.local_orders.keys() {
  911. let check_some = self.in_check.get(client_id);
  912. // 如果在查单队列中,不需要再添加
  913. if let Some(_) = check_some {
  914. continue;
  915. }
  916. let order = self.local_orders.get(client_id).unwrap();
  917. // 没有超过10s的订单,不需要检查
  918. if self.local_time - order.local_time < self._check_local_orders_interval {
  919. continue;
  920. }
  921. let key = format!("Check{}", client_id.clone());
  922. let value = vec![
  923. client_id.clone(),
  924. order.order_id.clone(),
  925. ];
  926. command.check.insert(key, value);
  927. self.in_check.insert(client_id.clone(), self.local_time);
  928. // debug!("查询订单:{:?}", client_id.clone());
  929. // debug!(?command);
  930. }
  931. // 维护查单队列
  932. self._release_in_check();
  933. // 更新查单时间
  934. self._check_local_orders_time = self.local_time;
  935. }
  936. // 开单指令生成逻辑
  937. #[instrument(skip(self, command), level="TRACE")]
  938. pub fn _post_open(&mut self, command: &mut OrderCommand) {
  939. // debug!(?command);
  940. // 开仓逻辑检测,主要是检测整点开仓逻辑
  941. if !self.check_allow_post_open() {
  942. return;
  943. }
  944. // 报单时延检测
  945. if self.local_time - self.post_open_time < self.post_open_interval {
  946. return;
  947. }
  948. // 报单时间更新
  949. self.post_open_time = self.local_time;
  950. // 挂单范围获取
  951. let long_upper = self.open_dist[0];
  952. let long_lower = self.open_dist[1];
  953. let short_lower = self.open_dist[2];
  954. let short_upper = self.open_dist[3];
  955. // 获取当前挂单价值
  956. let mut buy_price_list: Vec<Decimal> = vec![];
  957. let mut sell_price_list: Vec<Decimal> = vec![];
  958. let mut buy_value = Decimal::ZERO;
  959. let mut sell_value = Decimal::ZERO;
  960. for client_id in self.local_orders.keys() {
  961. let order = self.local_orders.get(client_id).unwrap();
  962. if order.side == "kd".to_string() {
  963. buy_price_list.push(order.price);
  964. buy_value += order.amount * order.price;
  965. }
  966. if order.side == "kk".to_string() {
  967. sell_price_list.push(order.price);
  968. sell_value += order.amount * order.price;
  969. }
  970. }
  971. // 计算可开价值
  972. let mut long_free_value = self.max_long_value - self.long_hold_value - buy_value;
  973. let mut short_free_value = self.max_short_value - self.short_hold_value - sell_value;
  974. // debug!(?long_free_value, ?short_free_value);
  975. // 现货要特殊处理
  976. if self.exchange.contains("spot") {
  977. let coin_value = self.coin * self.mp * self.lever_rate * self.adjust_lever_rate;
  978. let cash_value = self.cash * self.lever_rate * self.adjust_lever_rate;
  979. long_free_value = min(cash_value, self.max_long_value) - buy_value;
  980. short_free_value = min(coin_value, self.max_short_value) - sell_value;
  981. }
  982. // 一手开单价值计算
  983. let one_hand_long_value = dec!(0.99) * (self.max_long_value / self.grid);
  984. let one_hand_short_value = dec!(0.99) * (self.max_short_value / self.grid);
  985. // debug!(?self.post_side);
  986. // 挂多单
  987. if self.post_side >= 0 {
  988. // debug!(?buy_price_list);
  989. if buy_price_list.len() == 0 {
  990. let mut target_buy_price = (long_upper + long_lower) * dec!(0.5);
  991. target_buy_price = utils::clip(target_buy_price, self.bp * dec!(0.97), self.ap * dec!(1.0005));
  992. target_buy_price = utils::fix_price(target_buy_price, self.tick_size);
  993. let value = min(one_hand_long_value, long_free_value);
  994. let amount = utils::fix_amount(value / self.mp, self.step_size);
  995. let amount_value = amount * self.mp;
  996. // debug!(?one_hand_long_value, ?long_free_value, ?amount);
  997. // 下单价值不能太大,也不能太小
  998. if amount_value >= self._min_amount_value && amount_value <= long_free_value {
  999. let client_id = utils::generate_client_id(Some(self.broker_id.clone()));
  1000. let order = vec![
  1001. amount.to_string(),
  1002. "kd".to_string(),
  1003. target_buy_price.to_string(),
  1004. client_id.clone(),
  1005. ];
  1006. // debug!(?order);
  1007. command.limits_open.insert(client_id.clone(), order);
  1008. }
  1009. }
  1010. }
  1011. // 挂空单
  1012. if self.post_side <= 0 {
  1013. // debug!(?sell_price_list);
  1014. if sell_price_list.len() == 0 {
  1015. let mut target_sell_price = (short_lower + short_upper) * dec!(0.5);
  1016. target_sell_price = utils::clip(target_sell_price, self.bp * dec!(0.9995), self.ap * dec!(1.03));
  1017. target_sell_price = utils::fix_price(target_sell_price, self.tick_size);
  1018. let value = min(one_hand_short_value, short_free_value);
  1019. let amount = utils::fix_amount(value / self.mp, self.step_size);
  1020. let amount_value = amount * self.mp;
  1021. // 下单价值不能太大,也不能太小
  1022. if amount_value >= self._min_amount_value && amount_value <= short_free_value {
  1023. let client_id = utils::generate_client_id(Some(self.broker_id.clone()));
  1024. let order = vec![
  1025. amount.to_string(),
  1026. "kk".to_string(),
  1027. target_sell_price.to_string(),
  1028. client_id.clone(),
  1029. ];
  1030. // debug!(?order);
  1031. command.limits_open.insert(client_id.clone(), order);
  1032. }
  1033. }
  1034. }
  1035. }
  1036. // 定时打印
  1037. pub fn on_time_print(&mut self) {
  1038. if self.local_time - self._print_time < self._print_interval {
  1039. return;
  1040. }
  1041. // 记录上次打印时间
  1042. self._print_time = self.local_time;
  1043. if !self._is_print {
  1044. return;
  1045. }
  1046. // 准备好了的话就不打印预热中了
  1047. if self.is_ready {
  1048. return;
  1049. }
  1050. info!("预热中");
  1051. }
  1052. // 在满足条件后,返回非空command,否则返回一个空的command。原文的onTime。
  1053. pub fn on_time(&mut self, trader_msg: &TraderMsg) -> OrderCommand {
  1054. self.on_time_print();
  1055. let mut command = OrderCommand::new();
  1056. // 更新逻辑数据出错时,不进行后面的逻辑处理
  1057. if !self._update_data(trader_msg) {
  1058. return command;
  1059. }
  1060. // 检查是否准备充分
  1061. if !self.check_ready() {
  1062. return command;
  1063. }
  1064. // 刷新多空双方持仓比例
  1065. self._pos_rate();
  1066. // 生成开仓、平仓等相关价格
  1067. self.generate_dist();
  1068. // 下单指令处理逻辑
  1069. self._cancel_open(&mut command); // 撤单命令处理
  1070. self._post_open(&mut command); // 限价单命令处理
  1071. self._post_close(&mut command); // 平仓单命令处理
  1072. self._check_local_orders(&mut command); // 固定时间检查超时订单
  1073. self._update_in_cancel(&mut command); // 更新撤单队列,是一个filter
  1074. self._check_request_limit(&mut command); // 限制频率,移除不合规则之订单,是一个filter
  1075. self._refresh_request_limit(); // 刷新频率限制
  1076. self._update_request_num(&mut command); // 统计刷新频率
  1077. if command.limits_open.len() > 0 || command.limits_close.len() > 0{
  1078. let time = chrono::Utc::now().timestamp_millis();
  1079. let name = self.params.account_name.clone();
  1080. // 参考卖价
  1081. let ref_ap = self.ref_ap;
  1082. // 参考买价
  1083. let ref_bp = self.ref_bp;
  1084. let limits_open = command.limits_open.clone();
  1085. let limits_close = command.limits_close.clone();
  1086. spawn(async move {
  1087. let param_list = paras_limit_command(name.clone(), time.clone(), ref_ap.clone(), ref_bp.clone(), limits_open, limits_close);
  1088. let param_json_obj = serde_json::to_string(&param_list).unwrap();
  1089. market_warehouse_request(param_json_obj).await;
  1090. });
  1091. }
  1092. return command;
  1093. }
  1094. }
  1095. async fn market_warehouse_request(body_params: String) {
  1096. /****请求接口与 地址*/
  1097. let url = "http://as.skyfffire.com:8848/basic/saveDealRecords";
  1098. let client = Client::new();
  1099. let req = client.post(url).header("auth", "43626546liangjiang")
  1100. .header("Content-Type", "application/json").body(body_params.clone());
  1101. match req.send().await {
  1102. Ok(_) => {}
  1103. Err(_) => {}
  1104. };
  1105. // if !response.status().is_success() {
  1106. // error!("行情数据------仓库挂单数据存储失败--------!{}", response.status());
  1107. // error!(body_params);
  1108. // }
  1109. }
  1110. fn paras_limit_command (robot_name: String, time: i64, ref_ap: Decimal, ref_bp: Decimal, limits_open: HashMap<String, Vec<String>>, limits_close: HashMap<String, Vec<String>>) -> Vec<DealRecord>{
  1111. let mut limits = HashMap::new();
  1112. limits.extend(limits_open);
  1113. limits.extend(limits_close);
  1114. let mut list: Vec<DealRecord> = Vec::with_capacity(limits.len());
  1115. for item in limits.keys() {
  1116. let item_clone = item.clone();
  1117. let value = limits[&item_clone].clone();
  1118. let amount = Decimal::from_str(value.get(0).unwrap_or(&"0".to_string())).unwrap();
  1119. let side = value.get(1).unwrap();
  1120. let price = Decimal::from_str(value.get(2).unwrap_or(&"0".to_string())).unwrap();
  1121. let mut ref_price = ref_ap;
  1122. if "kd" == side {
  1123. ref_price = ref_bp;
  1124. }
  1125. let deal_recode = DealRecord {
  1126. refPrice: ref_price.to_string(),
  1127. regPrice: price.to_string(),
  1128. num: amount.to_string(),
  1129. triggerTime: time,
  1130. robotName: robot_name.clone(),
  1131. side: side.to_string(),
  1132. };
  1133. list.push(deal_recode);
  1134. }
  1135. return list;
  1136. }
  1137. #[cfg(test)]
  1138. mod tests {
  1139. use rust_decimal::Decimal;
  1140. use rust_decimal_macros::dec;
  1141. use crate::model::{OrderInfo, TraderMsg};
  1142. use global::params::Params;
  1143. use crate::strategy::Strategy;
  1144. #[test]
  1145. fn on_time_test() {
  1146. global::log_utils::init_log_with_debug();
  1147. let params = Params::new("config.toml.gate").unwrap();
  1148. let mut strategy = Strategy::new(&params, true);
  1149. let mut trader_msg = TraderMsg::new();
  1150. trader_msg.market.append(&mut vec![dec!(0.92), dec!(1.0), dec!(0.89), dec!(0.79), dec!(0.99), dec!(1.0), dec!(0.89), dec!(0.79), dec!(0.89), dec!(0.79), dec!(0.99), dec!(1.0), dec!(0.89), dec!(0.79)]);
  1151. trader_msg.position.long_pos = dec!(100);
  1152. trader_msg.position.long_avg = dec!(0.9);
  1153. trader_msg.orders.insert("0001".to_string(), OrderInfo{
  1154. symbol: "".to_string(),
  1155. amount: Default::default(),
  1156. side: "pd".to_string(),
  1157. price: dec!(10),
  1158. client_id: "0001".to_string(),
  1159. filled_price: Default::default(),
  1160. filled: Decimal::ZERO,
  1161. order_id: "".to_string(),
  1162. local_time: 0,
  1163. create_time: 0,
  1164. status: "".to_string(),
  1165. fee: Default::default(),
  1166. trace_stack: Default::default(),
  1167. });
  1168. trader_msg.cash = dec!(1000);
  1169. trader_msg.coin = Decimal::ZERO;
  1170. strategy.is_ready = true;
  1171. strategy.equity = dec!(1000);
  1172. strategy.lever_rate = Decimal::ONE;
  1173. // debug!("{:?}", strategy.on_time(&trader_msg));
  1174. }
  1175. }