init

client.go

@@ -0,0 +1,140 @@
+package gotdx
+
+import (
+	"bytes"
+	"compress/zlib"
+	"encoding/binary"
+	"gotdx/proto"
+	"io"
+	"log"
+	"net"
+	"strconv"
+	"strings"
+)
+
+type Client struct {
+	conn     net.Conn
+	opt      *Opt
+	complete chan bool
+	sending  chan bool
+}
+
+type Opt struct {
+	Host          string
+	Port          int
+	MaxRetryTimes int
+}
+
+func NewClient(opt *Opt) *Client {
+	client := &Client{}
+	if opt.MaxRetryTimes <= 0 {
+		opt.MaxRetryTimes = DefaultRetryTimes
+	}
+
+	client.opt = opt
+	client.sending = make(chan bool, 1)
+	client.complete = make(chan bool, 1)
+
+	return client
+}
+
+func (client *Client) connect() error {
+	addr := strings.Join([]string{client.opt.Host, strconv.Itoa(client.opt.Port)}, ":")
+	conn, err := net.Dial("tcp", addr)
+	if err != nil {
+		return err
+	}
+	client.conn = conn
+	return err
+}
+
+func (client *Client) do(msg proto.Msg) error {
+	sendData, err := msg.Serialize()
+	if err != nil {
+		return err
+	}
+
+	retryTimes := 0
+
+	for {
+		n, err := client.conn.Write(sendData)
+		if n < len(sendData) {
+			retryTimes++
+			if retryTimes <= client.opt.MaxRetryTimes {
+				log.Printf("第%d次重试\n", retryTimes)
+			} else {
+				return err
+			}
+		} else {
+			if err != nil {
+				return err
+			}
+			break
+		}
+	}
+
+	headerBytes := make([]byte, proto.MessageHeaderBytes)
+	_, err = io.ReadFull(client.conn, headerBytes)
+	if err != nil {
+		return err
+	}
+
+	headerBuf := bytes.NewReader(headerBytes)
+	var header proto.RespHeader
+	if err := binary.Read(headerBuf, binary.LittleEndian, &header); err != nil {
+		return err
+	}
+
+	if header.ZipSize > proto.MessageMaxBytes {
+		log.Printf("msgData has bytes(%d) beyond max %d\n", header.ZipSize, proto.MessageMaxBytes)
+		return ErrBadData
+	}
+
+	msgData := make([]byte, header.ZipSize)
+	_, err = io.ReadFull(client.conn, msgData)
+	if err != nil {
+		return err
+	}
+
+	var out bytes.Buffer
+	if header.ZipSize != header.UnZipSize {
+		b := bytes.NewReader(msgData)
+		r, _ := zlib.NewReader(b)
+		io.Copy(&out, r)
+		err = msg.UnSerialize(header, out.Bytes())
+	} else {
+		err = msg.UnSerialize(header, msgData)
+	}
+
+	return err
+}
+
+// Connect 连接券商行情服务器
+func (client *Client) Connect() (*proto.Hello1Reply, error) {
+	err := client.connect()
+	if err != nil {
+		return nil, err
+	}
+	obj := proto.NewHello1()
+	err = client.do(obj)
+	if err != nil {
+		return nil, err
+	}
+	return obj.Reply, err
+}
+
+// Disconnect 断开服务器
+func (client *Client) Disconnect() error {
+	return client.conn.Close()
+}
+
+// GetSecurityCount 获取指定市场内的证券数目
+func (client *Client) GetSecurityCount(market uint16) (*proto.SecurityCountReply, error) {
+	obj := proto.NewSecurityCount()
+	obj.SetParams(market)
+	err := client.do(obj)
+	if err != nil {
+		return nil, err
+	}
+	return obj.Reply, err
+}