dev 1

go-client/apps/cmd/main.go

@@ -0,0 +1,117 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"sort"
+	"strings"
+	"time"
+
+	"big-qmt/go-client/sdk"
+)
+
+var (
+	BaseURL = "http://127.0.0.1:10086"
+	Token   = "QMTbyYanweidong"
+	AccountType = "stock"
+	PassCodes = []string{}
+	Timeout = 15 * time.Second
+)
+
+func main() {
+	client := sdk.New(BaseURL, Token, AccountType, Timeout)
+	ctx, cancel := context.WithTimeout(context.Background(), Timeout)
+	defer cancel()
+
+	assets, err := client.Assets(ctx, AccountType)
+	if err != nil {
+		fatal("获取资产失败: %v", err)
+	}
+	positions, err := client.Positions(ctx, AccountType)
+	if err != nil {
+		fatal("获取持仓失败: %v", err)
+	}
+
+	fmt.Println(strings.Repeat("=", 80))
+	fmt.Printf("【时间】%s\n", time.Now().Format("2006-01-02 15:04:05"))
+	fmt.Printf("【服务】%s  accountType=%s\n", BaseURL, AccountType)
+	fmt.Printf("【资金】总资产：%.2f元，可用资金：%.2f元\n", assets.Total, assets.Available)
+	fmt.Printf("【持仓】%d只\n", len(positions))
+	fmt.Println(strings.Repeat("=", 80))
+
+	sort.Slice(positions, func(i, j int) bool {
+		return positions[i].StockCode < positions[j].StockCode
+	})
+	for _, p := range positions {
+		if p.Volume <= 0 {
+			continue
+		}
+		fmt.Printf(
+			"【持仓】%s %s 持仓=%d 可用=%d 冻结=%d 在途=%d 昨仓=%d 成本=%.3f 现价=%.3f 市值=%.2f 浮盈=%.2f 盈亏比例=%.2f%%\n",
+			p.StockCode, p.StockName, p.Volume, p.CanUseVolume, p.FrozenVolume, p.OnRoadVolume, p.YesterdayVolume,
+			p.OpenPrice, p.LastPrice, p.MarketValue, p.FloatProfit, p.ProfitRate*100,
+		)
+	}
+
+	printTicks(client, Timeout, PassCodes)
+}
+
+func printTicks(client *sdk.Client, timeout time.Duration, codes []string) {
+	fmt.Println(strings.Repeat("-", 80))
+	ctx, cancel := context.WithTimeout(context.Background(), timeout)
+	defer cancel()
+	ticks, err := client.FullTick(ctx, codes)
+	if err != nil {
+		fatal("获取行情失败: %v", err)
+	}
+	fmt.Printf("【行情】请求 %d 只，返回 %d 只\n", len(codes), len(ticks))
+	keys := make([]string, 0, len(ticks))
+	for code := range ticks {
+		keys = append(keys, code)
+	}
+	sort.Strings(keys)
+	for _, code := range keys {
+		t := ticks[code]
+		fmt.Printf("【Tick】%s last=%.3f close=%.3f open=%s high=%s low=%s volume=%s\n",
+			code, t.LastPrice, t.LastClose,
+			rawStr(t.Raw, "open", "lastOpen", "Open"),
+			rawStr(t.Raw, "high", "High"),
+			rawStr(t.Raw, "low", "Low"),
+			rawStr(t.Raw, "volume", "Volume"),
+		)
+	}
+}
+
+func splitCSV(s string) []string {
+	parts := strings.Split(s, ",")
+	out := make([]string, 0, len(parts))
+	for _, p := range parts {
+		p = strings.TrimSpace(p)
+		if p != "" {
+			out = append(out, p)
+		}
+	}
+	return out
+}
+
+func rawStr(m map[string]any, names ...string) string {
+	for _, name := range names {
+		if v, ok := m[name]; ok && v != nil {
+			return fmt.Sprint(v)
+		}
+	}
+	return "-"
+}
+
+func envOr(key, fallback string) string {
+	if v := strings.TrimSpace(os.Getenv(key)); v != "" {
+		return v
+	}
+	return fallback
+}
+
+func fatal(format string, args ...any) {
+	fmt.Fprintf(os.Stderr, format+"\n", args...)
+	os.Exit(1)
+}