111 lines
2.0 KiB
Go
111 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
type Client struct {
|
|
*TCPClient
|
|
}
|
|
|
|
func Dial(address string) (*Client, error) {
|
|
|
|
mux, err := NewTCPClient(address)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
client := &Client{
|
|
TCPClient: mux,
|
|
}
|
|
return client, nil
|
|
}
|
|
|
|
type TCPClient struct {
|
|
net.Conn
|
|
Errors chan error
|
|
ErrorCount int
|
|
address string
|
|
*PacketQueue
|
|
}
|
|
|
|
func NewTCPClient(address string) (*TCPClient, error) {
|
|
connection, err := net.Dial("tcp", address)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &TCPClient{
|
|
Errors: make(chan error),
|
|
ErrorCount: 0,
|
|
Conn: connection,
|
|
address: address,
|
|
PacketQueue: NewPacketQueue(connection),
|
|
}, nil
|
|
}
|
|
|
|
func (m *TCPClient) Connect() error {
|
|
if m.Conn == nil {
|
|
connection, err := net.Dial("tcp", m.address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
m.ErrorCount = 0
|
|
m.Conn = connection
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *TCPClient) HandleConnectionError(err error) error {
|
|
if err != nil {
|
|
m.Errors <- err
|
|
m.ErrorCount++
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (m *TCPClient) Close() {
|
|
m.Conn.Close()
|
|
}
|
|
|
|
func (m *TCPClient) SendPacket(messageType uint32, data []byte) error {
|
|
err := m.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = binary.Write(m.Conn, binary.LittleEndian, Packet{
|
|
Version: 1,
|
|
MessageType: messageType,
|
|
DataLength: uint64(len(data)),
|
|
})
|
|
if err != nil {
|
|
return m.HandleConnectionError(err)
|
|
}
|
|
_, err = m.Conn.Write(data)
|
|
return m.HandleConnectionError(err)
|
|
}
|
|
|
|
// func (m *TCPClient) SendPacketFn(messageType uint32, datafn func(w io.Writer) error) error {
|
|
// data, err := GetData(datafn)
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// return m.SendPacket(messageType, data)
|
|
// }
|
|
|
|
func (m *TCPClient) Call(messageType uint32, responseType uint32, data []byte) ([]byte, error) {
|
|
packetChan := m.Expect(responseType)
|
|
err := m.SendPacket(messageType, data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
select {
|
|
case ret := <-packetChan:
|
|
return ret, nil
|
|
case <-time.After(3 * time.Second):
|
|
return nil, fmt.Errorf("timeout")
|
|
}
|
|
}
|