dev 5

go-client/apps/trend/main.go

@@ -0,0 +1,94 @@
+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):
+		}
+	}
+}