59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
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"
|
|
}
|
|
}
|