strategy.rs 59 KB

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