95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"log"
|
||
"os"
|
||
"os/signal"
|
||
"syscall"
|
||
"time"
|
||
|
||
"big-qmt/go-client/apps/trend/logic"
|
||
"big-qmt/go-client/config"
|
||
"big-qmt/go-client/libs"
|
||
"big-qmt/go-client/sdk"
|
||
|
||
"github.com/robfig/cron/v3"
|
||
)
|
||
|
||
var (
|
||
StrategyName = "trend"
|
||
)
|
||
|
||
func main() {
|
||
log.SetFlags(log.LstdFlags | log.Lmicroseconds)
|
||
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
|
||
defer stop()
|
||
|
||
// 第一步：只从 YAML 文件加载系统配置和本机账户配置。
|
||
err := config.Load("etc")
|
||
if err != nil {
|
||
log.Fatalf("[ERROR] 加载配置失败: %v", err)
|
||
}
|
||
client := sdk.New(config.Global.QMTBaseURL, config.Global.QMTToken, config.HttpTimeOut)
|
||
|
||
// 第二步：QMT 未就绪时持续重试，退出信号仍可立即终止等待。
|
||
assets, positions, ok := waitForQMT(ctx, client)
|
||
if !ok {
|
||
return
|
||
}
|
||
|
||
// 第三步 初始化
|
||
logic.NewOrderBook()
|
||
logic.InitWatch()
|
||
if err := logic.InitState(StrategyName); err != nil {
|
||
log.Panicln("ERROR", err.Error())
|
||
}
|
||
if err := logic.SyncPositions(positions); err != nil {
|
||
log.Panicln("ERROR", err.Error())
|
||
}
|
||
signals, err := libs.InitSignals()
|
||
if err != nil {
|
||
log.Panicln("ERROR", err.Error())
|
||
}
|
||
|
||
// 打印启动信息
|
||
logic.Overview(assets, positions)
|
||
log.Printf("[INFO] 已加载 %d 个开仓信号", len(signals))
|
||
|
||
// 第四步：工作日每 30 秒触发，交易时段由 RunOnce 统一判断。
|
||
scheduler := cron.New(
|
||
cron.WithSeconds(),
|
||
cron.WithChain(cron.SkipIfStillRunning(cron.DefaultLogger)),
|
||
)
|
||
if _, err := scheduler.AddFunc("0,30 * 9-15 * * 1-5", func() {
|
||
logic.RunOnce(ctx, client, signals)
|
||
}); err != nil {
|
||
log.Fatalf("[ERROR] 创建计划任务失败: %v", err)
|
||
}
|
||
scheduler.Start()
|
||
log.Printf("[INFO] 计划任务已启动")
|
||
|
||
<-ctx.Done()
|
||
<-scheduler.Stop().Done()
|
||
log.Printf("[INFO] 停止")
|
||
}
|
||
|
||
func waitForQMT(ctx context.Context, client *sdk.Client) (*sdk.Assets, []sdk.Position, bool) {
|
||
for {
|
||
attempt, cancel := context.WithTimeout(ctx, config.HttpTimeOut)
|
||
assets, assetsErr := client.Assets(attempt)
|
||
_, positions, positionsErr := client.Positions(attempt)
|
||
cancel()
|
||
if assetsErr == nil && positionsErr == nil {
|
||
log.Printf("[INFO] QMT连接成功: %s", config.Global.QMTBaseURL)
|
||
return assets, positions, true
|
||
}
|
||
log.Printf("[WARNING] QMT未就绪，5秒后重试: assets=%v positions=%v", assetsErr, positionsErr)
|
||
select {
|
||
case <-ctx.Done():
|
||
return nil, nil, false
|
||
case <-time.After(5 * time.Second):
|
||
}
|
||
}
|
||
}
|