major refactoring :/

tcp-cart-client.go

@@ -0,0 +1,104 @@
+package main
+
+import (
+	"encoding/binary"
+	"io"
+	"net"
+	"time"
+)
+
+type CartClient struct {
+	*CartTCPClient
+}
+
+func CartDial(address string) (*CartClient, error) {
+
+	mux, err := NewCartTCPClient(address)
+	if err != nil {
+		return nil, err
+	}
+	client := &CartClient{
+		CartTCPClient: mux,
+	}
+	return client, nil
+}
+
+func (c *Client) Close() {
+	c.Conn.Close()
+}
+
+type CartTCPClient struct {
+	net.Conn
+	Errors     chan error
+	ErrorCount int
+	address    string
+	*PacketQueue
+}
+
+func NewCartTCPClient(address string) (*CartTCPClient, error) {
+	connection, err := net.Dial("tcp", address)
+	if err != nil {
+		return nil, err
+	}
+	return &CartTCPClient{
+		Errors:      make(chan error),
+		ErrorCount:  0,
+		Conn:        connection,
+		address:     address,
+		PacketQueue: NewPacketQueue(connection),
+	}, nil
+}
+
+func (m *CartTCPClient) Connect() error {
+	if m.Conn == nil {
+		connection, err := net.Dial("tcp", m.address)
+		if err != nil {
+
+			m.Errors <- err
+			m.ErrorCount++
+
+			return err
+		}
+		m.ErrorCount = 0
+		m.Conn = connection
+	}
+	return nil
+}
+
+func (m *CartTCPClient) SendPacket(messageType uint16, id CartId, data []byte) error {
+	err := m.Connect()
+	if err != nil {
+		return err
+	}
+	err = binary.Write(m.Conn, binary.LittleEndian, CartPacket{
+		Version:     1,
+		MessageType: messageType,
+		DataLength:  uint16(len(data)),
+		Id:          id,
+	})
+	if err != nil {
+		return err
+	}
+	_, err = m.Conn.Write(data)
+	return err
+}
+
+func (m *CartTCPClient) SendPacketFn(messageType uint16, id CartId, datafn func(w io.Writer) error) error {
+	data, err := GetData(datafn)
+	if err != nil {
+		return err
+	}
+	return m.SendPacket(messageType, id, data)
+}
+
+func (m *CartTCPClient) Call(messageType uint16, id CartId, responseType uint16, data []byte) ([]byte, error) {
+	err := m.SendPacket(messageType, id, data)
+	if err != nil {
+		return nil, err
+	}
+	packet, err := m.Expect(responseType, time.Second)
+	if err != nil {
+		return nil, err
+	}
+	return packet.Data, nil
+}