dev 1

go-client/apps/zt/config.go

@@ -0,0 +1,118 @@
+package main
+
+import (
+	"os"
+	"strconv"
+	"strings"
+	"time"
+)
+
+type Config struct {
+	QMTBaseURL       string
+	QMTToken         string
+	AccountType      string
+	AccountID        string
+	HostKey          string
+	APIHost          string
+	DataDir          string
+	HTTPTimeout      time.Duration
+	OrderTimeout     time.Duration
+	LoopInterval     time.Duration
+	OpenMoney        float64
+	MinCashRatio     float64
+	LossTriggerPct   float64
+	GridStepPct      float64
+	MinProfitPct     float64
+	AdoptExisting    bool
+	ReadyCacheStart  int
+	WatchTimeout     time.Duration
+	ReboundThreshold float64
+}
+
+func loadConfig() Config {
+	cfg := Config{
+		QMTBaseURL:       env("QMT_BASE_URL", "http://127.0.0.1:10086"),
+		QMTToken:         env("QMT_TOKEN", "QMTbyYanweidong"),
+		AccountType:      env("QMT_ACCOUNT", "stock"),
+		AccountID:        env("ACCOUNT_ID", ""),
+		HostKey:          env("HOST_KEY", ""),
+		APIHost:          strings.TrimRight(env("API_HOST", "http://139.224.247.176:13499"), "/"),
+		DataDir:          env("DATA_DIR", "D:/qmt_strategy_state"),
+		HTTPTimeout:      durationEnv("HTTP_TIMEOUT_SEC", 5) * time.Second,
+		OrderTimeout:     durationEnv("ORDER_TIMEOUT_SEC", 60) * time.Second,
+		LoopInterval:     durationEnv("LOOP_INTERVAL_SEC", 30) * time.Second,
+		OpenMoney:        floatEnv("OPEN_MONEY", 5000),
+		MinCashRatio:     floatEnv("MIN_CASH_RATIO", 0.1),
+		LossTriggerPct:   floatEnv("LOSS_TRIGGER_PCT", -30),
+		GridStepPct:      floatEnv("GRID_STEP_PCT", 1),
+		MinProfitPct:     floatEnv("MIN_PROFIT_PCT", 2),
+		AdoptExisting:    boolEnv("ADOPT_EXISTING_POSITIONS", true),
+		ReadyCacheStart:  intEnv("READY_CACHE_START", 925),
+		WatchTimeout:     durationEnv("WATCH_TIMEOUT_SEC", 300) * time.Second,
+		ReboundThreshold: floatEnv("REBOUND_THRESHOLD", 0.61),
+	}
+	if strings.TrimSpace(cfg.AccountID) == "" {
+		logf("ERROR", "ACCOUNT_ID 为空")
+		os.Exit(1)
+	}
+	if strings.TrimSpace(cfg.HostKey) == "" {
+		logf("ERROR", "HOST_KEY 为空")
+		os.Exit(1)
+	}
+	if cfg.MinCashRatio < 0 || cfg.MinCashRatio >= 1 {
+		logf("ERROR", "MIN_CASH_RATIO 必须在 [0, 1)")
+		os.Exit(1)
+	}
+	if cfg.OpenMoney <= 0 {
+		logf("ERROR", "OPEN_MONEY 必须大于 0")
+		os.Exit(1)
+	}
+	if err := os.MkdirAll(cfg.DataDir, 0o755); err != nil {
+		logf("ERROR", "创建 DATA_DIR 失败: %v", err)
+		os.Exit(1)
+	}
+	return cfg
+}
+
+func env(key, fallback string) string {
+	if v := strings.TrimSpace(os.Getenv(key)); v != "" {
+		return v
+	}
+	return fallback
+}
+
+func intEnv(key string, fallback int) int {
+	v := strings.TrimSpace(os.Getenv(key))
+	if v == "" {
+		return fallback
+	}
+	n, err := strconv.Atoi(v)
+	if err != nil {
+		return fallback
+	}
+	return n
+}
+
+func floatEnv(key string, fallback float64) float64 {
+	v := strings.TrimSpace(os.Getenv(key))
+	if v == "" {
+		return fallback
+	}
+	f, err := strconv.ParseFloat(v, 64)
+	if err != nil {
+		return fallback
+	}
+	return f
+}
+
+func durationEnv(key string, fallbackSec int) time.Duration {
+	return time.Duration(intEnv(key, fallbackSec))
+}
+
+func boolEnv(key string, fallback bool) bool {
+	v := strings.ToLower(strings.TrimSpace(os.Getenv(key)))
+	if v == "" {
+		return fallback
+	}
+	return v == "1" || v == "true" || v == "yes"
+}