strategy.rs 59 KB

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