dev 2

go-client/libs/market.go

@@ -0,0 +1,58 @@
+package libs
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+)
+
+// AllowOpen 每次开仓或补仓前取 60 分钟大盘信号，只有 UP 才放行。
+func AllowOpen(apiHost string, timeout time.Duration) bool {
+	rawURL := strings.TrimRight(apiHost, "/") + "/a/market"
+	payload, err := getJSON(rawURL, url.Values{"period": {"60m"}}, timeout)
+	if err != nil {
+		log.Printf("[ERROR] 获取60m大盘信号失败: %s %v", rawURL, err)
+		return false
+	}
+	status := Status(payload)
+	log.Printf("[INFO] 大盘信号: url=%s status=%s", rawURL, status)
+	return status == "UP"
+}
+
+// Status 兼容常见响应结构；无法识别的值统一按 UNKNOWN 处理。
+func Status(payload map[string]any) string {
+	var value any = payload
+	if m, ok := value.(map[string]any); ok {
+		if d, exists := m["data"]; exists {
+			value = d
+		}
+	}
+	if arr, ok := value.([]any); ok {
+		if len(arr) == 0 {
+			value = nil
+		} else {
+			value = arr[len(arr)-1]
+		}
+	}
+	if m, ok := value.(map[string]any); ok {
+		if v, exists := m["action"]; exists {
+			value = v
+		} else if v, exists := m["status"]; exists {
+			value = v
+		} else if v, exists := m["signal"]; exists {
+			value = v
+		}
+	}
+	s := strings.ToUpper(strings.TrimSpace(fmt.Sprint(value)))
+	switch s {
+	case "UP", "DOWN", "NEUTRAL":
+		return s
+	default:
+		return "UNKNOWN"
+	}
+}