1use futu_core::error::{FutuError, Result};
2use futu_core::proto_id;
3use futu_net::client::FutuClient;
4
5use crate::types::Security;
6
7#[derive(Debug, Clone)]
9pub struct BrokerEntry {
10 pub id: i64,
11 pub name: String,
12 pub pos: i32,
13}
14
15impl BrokerEntry {
16 pub fn from_proto(b: &futu_proto::qot_common::Broker) -> Self {
17 Self {
18 id: b.id,
19 name: b.name.clone(),
20 pos: b.pos,
21 }
22 }
23}
24
25#[derive(Debug, Clone)]
27pub struct BrokerData {
28 pub security: Security,
29 pub ask_list: Vec<BrokerEntry>,
30 pub bid_list: Vec<BrokerEntry>,
31}
32
33pub async fn get_broker(client: &FutuClient, security: &Security) -> Result<BrokerData> {
37 let req = futu_proto::qot_get_broker::Request {
38 c2s: futu_proto::qot_get_broker::C2s {
39 security: security.to_proto(),
40 },
41 };
42
43 let body = prost::Message::encode_to_vec(&req);
44 let resp_frame = client.request(proto_id::QOT_GET_BROKER, body).await?;
45
46 let resp: futu_proto::qot_get_broker::Response =
47 prost::Message::decode(resp_frame.body.as_ref()).map_err(FutuError::Proto)?;
48
49 if resp.ret_type != 0 {
50 return Err(FutuError::ServerError {
51 ret_type: resp.ret_type,
52 msg: resp.ret_msg.unwrap_or_default(),
53 });
54 }
55
56 let s2c = resp
57 .s2c
58 .ok_or(FutuError::Codec("missing s2c in GetBroker".into()))?;
59
60 Ok(BrokerData {
61 security: Security::from_proto(&s2c.security),
62 ask_list: s2c
63 .broker_ask_list
64 .iter()
65 .map(BrokerEntry::from_proto)
66 .collect(),
67 bid_list: s2c
68 .broker_bid_list
69 .iter()
70 .map(BrokerEntry::from_proto)
71 .collect(),
72 })
73}