跳转到内容

空白策略

先从精简的策略实现入手,了解运行时钩子,暂不引入 Alpha 模型或信号处理器。

use aq_engine::core::broker::data_feeds::yahoo::YahooFinanceDataFeed;
use aq_engine::core::broker::paper_broker::PaperBroker;
use aq_engine::core::broker::types::{AccountType, Asset, BarData};
use aq_engine::core::broker::UnifiedBroker;
use aq_engine::core::insight::Insight;
use aq_engine::core::strategy::{
EventStreamType, Strategy, StrategyContext, StrategyMode, StrategyState,
};
use aq_engine::core::utils::timeframe::{TimeFrame, TimeFrameUnit};
use chrono::{Duration, Utc};
use std::collections::HashSet;
pub struct BlankStrategy;
impl Strategy for BlankStrategy {
fn on_start(&mut self, ctx: &mut dyn StrategyContext) {
// The main strategy timeframe is registered automatically.
// This adds a feature bar stream for every symbol in the universe.
ctx.add_events(
EventStreamType::Bar,
Some(TimeFrame::new(15, TimeFrameUnit::Minute)),
);
}
fn init(&mut self, _ctx: &mut dyn StrategyContext, _asset: &Asset) {}
fn universe(&self, _ctx: &mut dyn StrategyContext) -> HashSet<String> {
HashSet::from([String::from("AAPL")])
}
fn on_bar(&mut self, _ctx: &mut dyn StrategyContext, _symbol: &str, _bar: &BarData) {}
fn generate_insights(&mut self, _ctx: &mut dyn StrategyContext, _symbol: &str) {}
fn insight_pipeline(&mut self, _ctx: &mut dyn StrategyContext, _insight: &Insight) {}
fn on_teardown(&mut self, _ctx: &mut dyn StrategyContext) {}
}
let execution = PaperBroker::new(AccountType::Paper, 100_000.0, 1);
let data = YahooFinanceDataFeed::new();
let broker = UnifiedBroker::new_backtest(execution, data);
let timeframe = TimeFrame::new(1, TimeFrameUnit::Day);
let strategy = BlankStrategy;
let mut state = StrategyState::new(
"blank-strategy".to_string(),
"1.0.0".to_string(),
strategy,
broker,
StrategyMode::Backtest,
timeframe.clone(),
);
let start = Utc::now() - Duration::days(30);
let end = Utc::now();
let results = state.run_backtest(start, end, timeframe).await?;
results.print_metrics();

这段入门代码展示了一个精简而完整的流程:定义策略、连接模拟执行适配器和数据源、运行回测,再查看结果指标。add_events 调用会注册一个额外的特征周期;策略主周期仍会自动运行。

通常在 Alpha 模型或 generate_insights() 中创建交易信号,再交给信号处理流水线计算仓位、添加风险控制并提交。

use aq_engine::core::broker::types::OrderSide;
use aq_engine::core::insight::{types::StrategyType, Insight};
fn generate_insights(&mut self, ctx: &mut dyn StrategyContext, symbol: &str) {
let mut insight = Insight::new(
OrderSide::Buy,
symbol.to_string(),
StrategyType::Testing,
ctx.timeframe().clone(),
80,
None,
);
insight
.set_limit_price(Some(200.0))
.set_take_profit_levels(Some(vec![206.0]))
.set_stop_loss(Some(197.5))
.set_period_unfilled(Some(5))
.set_period_till_tp(Some(12));
ctx.add_insight(insight);
}

有关完整的交易信号模型,请参阅交易信号。有关仓位计算、验证和提交,请参阅信号处理器。