dev 1
This commit is contained in:
193
go-client/sdk/account.go
Normal file
193
go-client/sdk/account.go
Normal file
@@ -0,0 +1,193 @@
|
||||
package sdk
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// Position 对应 HoldingHandler 封装后的持仓。
|
||||
type Position struct {
|
||||
StockCode string `json:"StockCode"`
|
||||
StockName string `json:"StockName"`
|
||||
Direction any `json:"Direction"`
|
||||
Volume int `json:"Volume"`
|
||||
OpenPrice float64 `json:"OpenPrice"`
|
||||
FloatProfit float64 `json:"FloatProfit"`
|
||||
MarketValue float64 `json:"MarketValue"`
|
||||
StockHolder string `json:"StockHolder"`
|
||||
FrozenVolume int `json:"FrozenVolume"`
|
||||
CanUseVolume int `json:"CanUseVolume"`
|
||||
OnRoadVolume int `json:"OnRoadVolume"`
|
||||
YesterdayVolume int `json:"YesterdayVolume"`
|
||||
LastPrice float64 `json:"LastPrice"`
|
||||
ProfitRate float64 `json:"ProfitRate"`
|
||||
FutureTradeType any `json:"FutureTradeType"`
|
||||
ExpireDate string `json:"ExpireDate"`
|
||||
}
|
||||
|
||||
// Assets 对应 /api/v2/assets。
|
||||
type Assets struct {
|
||||
Total float64 `json:"total"`
|
||||
Available float64 `json:"available"`
|
||||
}
|
||||
|
||||
type accountBody struct {
|
||||
Account string `json:"account"`
|
||||
}
|
||||
|
||||
func (c *Client) Positions(ctx context.Context, account string) ([]Position, error) {
|
||||
raw := map[string]json.RawMessage{}
|
||||
if err := c.post(ctx, "/api/v2/positions", accountBody{Account: c.Account(account)}, &raw); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out := make([]Position, 0, len(raw))
|
||||
for code, blob := range raw {
|
||||
var p Position
|
||||
if err := json.Unmarshal(blob, &p); err != nil {
|
||||
return nil, fmt.Errorf("position %s: %w", code, err)
|
||||
}
|
||||
if p.StockCode == "" {
|
||||
p.StockCode = code
|
||||
}
|
||||
out = append(out, p)
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *Client) Holding(ctx context.Context, account string) ([]Position, error) {
|
||||
raw := map[string]json.RawMessage{}
|
||||
if err := c.post(ctx, "/api/holding", accountBody{Account: c.Account(account)}, &raw); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out := make([]Position, 0, len(raw))
|
||||
for code, blob := range raw {
|
||||
var p Position
|
||||
if err := json.Unmarshal(blob, &p); err != nil {
|
||||
return nil, fmt.Errorf("holding %s: %w", code, err)
|
||||
}
|
||||
if p.StockCode == "" {
|
||||
p.StockCode = code
|
||||
}
|
||||
out = append(out, p)
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (c *Client) Assets(ctx context.Context, account string) (*Assets, error) {
|
||||
var out Assets
|
||||
if err := c.post(ctx, "/api/v2/assets", accountBody{Account: c.Account(account)}, &out); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
func (c *Client) TotalMoney(ctx context.Context, account string) (float64, error) {
|
||||
var out struct {
|
||||
TotalMoney float64 `json:"total_money"`
|
||||
}
|
||||
if err := c.post(ctx, "/api/money/total", accountBody{Account: c.Account(account)}, &out); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return out.TotalMoney, nil
|
||||
}
|
||||
|
||||
func (c *Client) AvailableMoney(ctx context.Context, account string) (float64, error) {
|
||||
var out struct {
|
||||
AvailableMoney float64 `json:"available_money"`
|
||||
}
|
||||
if err := c.post(ctx, "/api/money/available", accountBody{Account: c.Account(account)}, &out); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return out.AvailableMoney, nil
|
||||
}
|
||||
|
||||
type OrderRefResult struct {
|
||||
Status string `json:"status"`
|
||||
Action string `json:"action"`
|
||||
Stock string `json:"stock"`
|
||||
OpType int `json:"opType"`
|
||||
OrderRef string `json:"order_ref"`
|
||||
}
|
||||
|
||||
func (c *Client) Buy(ctx context.Context, stock string, price float64, volume int, prType int) (*OrderRefResult, error) {
|
||||
body := map[string]any{"stock": stock, "price": price, "volume": volume}
|
||||
if prType != 0 {
|
||||
body["prType"] = prType
|
||||
}
|
||||
var out OrderRefResult
|
||||
if err := c.post(ctx, "/api/order/buy", body, &out); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
func (c *Client) Sell(ctx context.Context, stock string, price float64, volume int, prType int) (*OrderRefResult, error) {
|
||||
body := map[string]any{"stock": stock, "price": price, "volume": volume}
|
||||
if prType != 0 {
|
||||
body["prType"] = prType
|
||||
}
|
||||
var out OrderRefResult
|
||||
if err := c.post(ctx, "/api/order/sell", body, &out); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
type OrderStatus struct {
|
||||
OrderSysID string `json:"order_sys_id"`
|
||||
Status int `json:"status"`
|
||||
VolumeLeft int `json:"volume_left"`
|
||||
VolumeTraded int `json:"volume_traded"`
|
||||
}
|
||||
|
||||
func (c *Client) OrderStatusList(ctx context.Context, account string) ([]OrderStatus, error) {
|
||||
var out struct {
|
||||
Orders []OrderStatus `json:"orders"`
|
||||
}
|
||||
if err := c.post(ctx, "/api/order/status", accountBody{Account: c.Account(account)}, &out); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out.Orders, nil
|
||||
}
|
||||
|
||||
type CanceledOrder struct {
|
||||
OrderSysID string `json:"order_sys_id"`
|
||||
Stock string `json:"stock"`
|
||||
VolumeLeft int `json:"volume_left"`
|
||||
}
|
||||
|
||||
type CancelAllResult struct {
|
||||
Status string `json:"status"`
|
||||
Message string `json:"message"`
|
||||
CanceledOrders []CanceledOrder `json:"canceled_orders"`
|
||||
CanceledSysIDs []string `json:"canceled_sys_ids"`
|
||||
}
|
||||
|
||||
func (c *Client) CancelAll(ctx context.Context, account string) (*CancelAllResult, error) {
|
||||
var out CancelAllResult
|
||||
if err := c.post(ctx, "/api/order/cancel_all", accountBody{Account: c.Account(account)}, &out); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
// CancelByRule 按「代码.市场 + (剩余+已成)」匹配撤单。HTTP 没有按委托号撤单，ZT 用它代替 cancel(orderId)。
|
||||
func (c *Client) CancelByRule(ctx context.Context, stock string, volume int, account string) (*CancelAllResult, error) {
|
||||
var out CancelAllResult
|
||||
body := map[string]any{"stock": stock, "volume": volume, "account": c.Account(account)}
|
||||
if err := c.post(ctx, "/api/order/cancel_order", body, &out); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &out, nil
|
||||
}
|
||||
|
||||
func (c *Client) Deals(ctx context.Context, account string) ([]map[string]string, error) {
|
||||
var out struct {
|
||||
Deals []map[string]string `json:"deals"`
|
||||
}
|
||||
if err := c.post(ctx, "/api/order/deal", accountBody{Account: c.Account(account)}, &out); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out.Deals, nil
|
||||
}
|
||||
Reference in New Issue
Block a user