dev 3

go-client/libs/calc.go

@@ -0,0 +1,25 @@
+package libs
+
+import (
+	"math/rand"
+	"time"
+)
+
+var letters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
+
+func randStr(n int) string {
+	b := make([]rune, n)
+	for i := range b {
+		b[i] = letters[rand.Intn(len(letters))]
+	}
+	return string(b)
+}
+
+func TradingTime(t time.Time) bool {
+	if t.Weekday() == time.Saturday || t.Weekday() == time.Sunday {
+		return false
+	}
+	second := t.Hour()*3600 + t.Minute()*60 + t.Second()
+	return (second >= 9*3600+30*60 && second <= 11*3600+30*60) ||
+		(second >= 13*3600 && second <= 15*3600)
+}

go-client/libs/const.go

@@ -0,0 +1,8 @@
+package libs
+
+import "time"
+
+var (
+	API_HOST    = "http://139.224.247.176:13499"
+	HTTPTimeout = 5 * time.Second
+)

go-client/libs/http.go

@@ -1,17 +1,15 @@
 package libs
 
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+	"time"
+)
 
-func getJSON(rawURL string, params url.Values, timeout time.Duration) (map[string]any, error) {
-	if timeout <= 0 {
-		timeout = 5 * time.Second
-	}
-	if params != nil {
-		if strings.Contains(rawURL, "?") {
-			rawURL += "&" + params.Encode()
-		} else {
-			rawURL += "?" + params.Encode()
-		}
-	}
+// GetJSON 请求 JSON 接口并返回对象。
+func GetJSON(rawURL string, timeout time.Duration) ([]byte, error) {
 	req, err := http.NewRequest(http.MethodGet, rawURL, nil)
 	if err != nil {
 		return nil, err
@@ -31,9 +29,5 @@ func getJSON(rawURL string, params url.Values, timeout time.Duration) (map[strin
 	if resp.StatusCode >= 400 {
 		return nil, fmt.Errorf("http %d: %s", resp.StatusCode, strings.TrimSpace(string(body)))
 	}
-	out := map[string]any{}
-	if err := json.Unmarshal(body, &out); err != nil {
-		return nil, err
-	}
-	return out, nil
+	return body, nil
 }

go-client/libs/market.go

@@ -3,24 +3,33 @@ package libs
 import (
 	"encoding/json"
 	"fmt"
-	"io"
 	"log"
-	"net/http"
-	"net/url"
 	"strings"
-	"time"
+)
+
+var (
+	MarketUrl = "/a/market"
+	Period    = "60m"
 )
 
 // 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)
+func AllowOpen() bool {
+	// gen url.
+	fullUrl := fmt.Sprintf("%s%s?period=%s&t=%s", API_HOST, MarketUrl, Period, randStr(16))
+	payload, err := GetJSON(fullUrl, HTTPTimeout)
 	if err != nil {
-		log.Printf("[ERROR] 获取60m大盘信号失败: %s %v", rawURL, err)
+		log.Printf("[ERROR] 获取大盘指数失败: %s %v", fullUrl, err)
 		return false
 	}
-	status := Status(payload)
-	log.Printf("[INFO] 大盘信号: url=%s status=%s", rawURL, status)
+	var result map[string]any
+	err = json.Unmarshal(payload, &result)
+	if err != nil {
+		log.Printf("[ERROR] 获取大盘指数解析: %v", err)
+		return false
+	}
+
+	status := Status(result)
+	log.Printf("[INFO] 大盘信号: url=%s status=%s", fullUrl, status)
 	return status == "UP"
 }
 

go-client/libs/signal.go

@@ -0,0 +1,46 @@
+package libs
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+)
+
+var (
+	Dcm_Signal = "/a/dcm_signal"
+)
+
+type SignalResult struct {
+	Code    string                `json:"code"`
+	Total   int                   `json:"total"`
+	Updated string                `json:"updated"`
+	Data    map[string]SignalItem `json:"data"`
+	Message string                `json:"message"`
+}
+
+type SignalItem struct {
+	Code          string             `json:"code"`
+	Name          string             `json:"name"`
+	Desc          string             `json:"desc"`
+	LastClose     float64            `json:"last_close"`
+	TechIndicator map[string]float64 `json:"tech_indicator"`
+}
+
+// FetchSignals 启动时读取信号，运行期间直接使用内存数据。
+func FetchSignal(subUrl, host_key string) (*SignalResult, error) {
+	// gen url.
+	fullUrl := fmt.Sprintf("%s%s?host_key=%s&t=%s", API_HOST, subUrl, host_key, randStr(16))
+	// doing
+	payload, err := GetJSON(fullUrl, HTTPTimeout)
+	if err != nil {
+		log.Printf("[ERROR] 获取60m大盘信号失败: %s %v", fullUrl, err)
+		return nil, err
+	}
+
+	var result SignalResult
+	err = json.Unmarshal(payload, &result)
+	if err != nil {
+		return nil, err
+	}
+	return &result, nil
+}