use crate::client::AkShareClient;
use crate::error::{Error, Result};
use crate::types::value_ext::ValueExt;
use crate::types::{CandlePoint, EtfSpotItem, FundSnapshot};
use crate::util::{parse_f64_safe, today_iso};
fn lof_secid(symbol: &str) -> Result<String> {
let s = symbol.trim();
if s.contains('.') && s.len() >= 3 {
return Ok(s.to_string());
}
if s.len() == 6 && s.chars().all(|c| c.is_ascii_digit()) {
let prefix = if s.starts_with('1') || s.starts_with('0') {
"0"
} else if s.starts_with('5') {
"1"
} else {
"0"
};
return Ok(format!("{prefix}.{s}"));
}
Err(Error::invalid_input(format!(
"invalid LOF symbol: {symbol}"
)))
}
impl AkShareClient {
pub async fn fund_lof_list(&self, limit: usize) -> Result<Vec<FundSnapshot>> {
let pz = limit.max(1).to_string();
let response = crate::util::send_and_check(
self.get("https://push2.eastmoney.com/api/qt/clist/get")
.query(&crate::util::eastmoney_clist_params(
pz.as_str(),
&[("fs", "b:MK0025"), ("fields", "f12,f14,f2,f3")],
)),
)
.await?;
let payload: serde_json::Value = response.json().await.map_err(Error::from)?;
let items = payload
.get("data")
.and_then(|d| d.get("diff"))
.and_then(|d| d.as_array())
.cloned()
.unwrap_or_default();
let date = today_iso();
let snapshots: Vec<FundSnapshot> = items
.into_iter()
.take(limit)
.filter_map(|v| {
Some(FundSnapshot {
symbol: v.get("f12")?.as_str()?.to_string(),
name: v.str_or(&["f14"], ""),
date: date.clone(),
nav: v.f64_or(&["f2"], 0.0),
acc_nav: v.f64_or(&["f2"], 0.0),
change_pct: v.f64_or(&["f3"], 0.0),
fund_type: Some("lof".to_string()),
})
})
.collect();
if snapshots.is_empty() {
return Err(Error::not_found("eastmoney returned no LOF fund data"));
}
Ok(snapshots)
}
pub async fn fund_lof_hist(&self, symbol: &str, limit: usize) -> Result<Vec<CandlePoint>> {
let secid = lof_secid(symbol)?;
self.eastmoney_klines(&secid, "qfq", limit).await
}
pub async fn fund_lof_hist_em(
&self,
symbol: &str,
period: &str,
_start_date: &str,
_end_date: &str,
adjust: &str,
) -> Result<Vec<CandlePoint>> {
let period_map = match period {
"daily" => "101",
"weekly" => "102",
"monthly" => "103",
_ => {
return Err(Error::invalid_input(format!(
"unsupported period: {period}"
)));
}
};
let adjust_map = match adjust {
"qfq" => "1",
"hfq" => "2",
"" | "none" => "0",
_ => {
return Err(Error::invalid_input(format!(
"unsupported adjust: {adjust}"
)));
}
};
let secid = lof_secid(symbol)?;
let klines = self
.kline_fetch(&secid, period_map, adjust_map, usize::MAX, &[])
.await?;
let mut candles = Vec::new();
for s in &klines {
let fields: Vec<&str> = s.split(',').collect();
if fields.len() < 11 {
continue;
}
candles.push(CandlePoint {
trade_date: fields[0].to_string(),
open: parse_f64_safe(fields[1]),
close: parse_f64_safe(fields[2]),
high: parse_f64_safe(fields[3]),
low: parse_f64_safe(fields[4]),
volume: fields[5].parse().unwrap_or(0),
amount: parse_f64_safe(fields[6]),
amplitude_pct: parse_f64_safe(fields[7]),
change_pct: parse_f64_safe(fields[8]),
change_amount: parse_f64_safe(fields[9]),
turnover_pct: parse_f64_safe(fields[10]),
});
}
Ok(candles)
}
pub async fn fund_lof_hist_min(
&self,
symbol: &str,
period: &str,
start_date: &str,
end_date: &str,
adjust: &str,
) -> Result<Vec<CandlePoint>> {
let adjust_map = match adjust {
"qfq" => "1",
"hfq" => "2",
"" | "none" => "0",
_ => {
return Err(Error::invalid_input(format!(
"unsupported adjust: {adjust}"
)));
}
};
let secid = lof_secid(symbol)?;
if period == "1" {
let response = crate::util::send_and_check(
self.get("https://push2his.eastmoney.com/api/qt/stock/trends2/get")
.query(&[
("fields1", "f1,f2,f3,f4,f5,f6,f7,f8,f9,f10,f11,f12,f13"),
("fields2", "f51,f52,f53,f54,f55,f56,f57,f58"),
("ut", "7eea3edcaed734bea9cbfc24409ed989"),
("ndays", "5"),
("iscr", "0"),
("secid", secid.as_str()),
]),
)
.await?;
let payload: serde_json::Value = response.json().await.map_err(Error::from)?;
let trends = payload
.get("data")
.and_then(|d| d.get("trends"))
.and_then(|t| t.as_array())
.ok_or_else(|| Error::not_found(format!("no trend data for LOF {symbol}")))?;
let mut candles = Vec::new();
for trend in trends {
let s = trend.as_str().unwrap_or("");
let fields: Vec<&str> = s.split(',').collect();
if fields.len() < 8 {
continue;
}
let dt = fields[0];
if dt < start_date || dt > end_date {
continue;
}
candles.push(CandlePoint {
trade_date: dt.to_string(),
open: parse_f64_safe(fields[1]),
close: parse_f64_safe(fields[2]),
high: parse_f64_safe(fields[3]),
low: parse_f64_safe(fields[4]),
volume: fields[5].parse().unwrap_or(0),
amount: parse_f64_safe(fields[6]),
amplitude_pct: 0.0,
change_pct: 0.0,
change_amount: 0.0,
turnover_pct: 0.0,
});
}
Ok(candles)
} else {
let klines = self
.kline_fetch(&secid, period, adjust_map, usize::MAX, &[])
.await?;
let mut candles = Vec::new();
for s in &klines {
let fields: Vec<&str> = s.split(',').collect();
if fields.len() < 11 {
continue;
}
let dt = fields[0];
if dt < start_date || dt > end_date {
continue;
}
candles.push(CandlePoint {
trade_date: dt.to_string(),
open: parse_f64_safe(fields[1]),
close: parse_f64_safe(fields[2]),
high: parse_f64_safe(fields[3]),
low: parse_f64_safe(fields[4]),
volume: fields[5].parse().unwrap_or(0),
amount: parse_f64_safe(fields[6]),
amplitude_pct: parse_f64_safe(fields[7]),
change_pct: parse_f64_safe(fields[8]),
change_amount: parse_f64_safe(fields[9]),
turnover_pct: parse_f64_safe(fields[10]),
});
}
Ok(candles)
}
}
pub async fn fund_lof_spot(&self) -> Result<Vec<EtfSpotItem>> {
let response = crate::util::send_and_check(
self.get("https://88.push2.eastmoney.com/api/qt/clist/get")
.query(&crate::util::eastmoney_clist_params(
"10000",
&[
("fid", "f3"),
("fs", "b:MK0404,b:MK0405,b:MK0406,b:MK0407"),
(
"fields",
"f2,f3,f4,f5,f6,f7,f12,f14,f15,f16,f17,f18,f20,f21",
),
],
)),
)
.await?;
let payload: serde_json::Value = response.json().await.map_err(Error::from)?;
let items = payload
.get("data")
.and_then(|d| d.get("diff"))
.and_then(|d| d.as_array())
.cloned()
.unwrap_or_default();
let result: Vec<EtfSpotItem> = items
.into_iter()
.filter_map(|v| {
Some(EtfSpotItem {
code: v.get("f12")?.as_str()?.to_string(),
name: v.str_or(&["f14"], ""),
latest_price: v.f64_or(&["f2"], 0.0),
change_pct: v.f64_or(&["f3"], 0.0),
change_amount: v.f64_or(&["f4"], 0.0),
volume: v.f64_or(&["f5"], 0.0),
amount: v.f64_or(&["f6"], 0.0),
open: v.f64_or(&["f17"], 0.0),
high: v.f64_or(&["f15"], 0.0),
low: v.f64_or(&["f16"], 0.0),
prev_close: v.f64_or(&["f18"], 0.0),
amplitude: v.f64_or(&["f7"], 0.0),
turnover_rate: 0.0,
iopv: 0.0,
discount_rate: 0.0,
shares: 0.0,
circ_mv: v.f64_or(&["f21"], 0.0),
total_mv: v.f64_or(&["f20"], 0.0),
data_date: String::new(),
})
})
.collect();
if result.is_empty() {
return Err(Error::not_found("no LOF spot data"));
}
Ok(result)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_lof_secid_shenzhen() {
assert_eq!(lof_secid("160105").unwrap(), "0.160105");
}
#[test]
fn test_lof_secid_shanghai() {
assert_eq!(lof_secid("501000").unwrap(), "1.501000");
}
#[test]
fn test_lof_secid_raw() {
assert_eq!(lof_secid("0.160105").unwrap(), "0.160105");
}
#[test]
fn test_lof_secid_invalid() {
assert!(lof_secid("abc").is_err());
assert!(lof_secid("12345").is_err());
}
}