95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package tcp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"log/slog"
|
|
"net"
|
|
)
|
|
|
|
type TcpClient struct {
|
|
conn *net.TCPConn
|
|
handler func(n int, data []byte)
|
|
ctx context.CancelFunc
|
|
conning bool
|
|
}
|
|
|
|
func StartTcpClient(rAddr string, handler func(n int, data []byte), readErr func(err error)) (*TcpClient, error) {
|
|
raddr, addErr := net.ResolveTCPAddr("tcp", rAddr)
|
|
if addErr != nil {
|
|
return nil, addErr
|
|
}
|
|
conn, err := net.DialTCP("tcp", nil, raddr)
|
|
ctx, ctxFun := context.WithCancel(context.Background())
|
|
client := &TcpClient{conn: conn, ctx: ctxFun}
|
|
if err != nil {
|
|
client.ctx = ctxFun
|
|
client.conning = false
|
|
return nil, err
|
|
}
|
|
|
|
go func() {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
default:
|
|
}
|
|
data := make([]byte, 1024)
|
|
l, readDataErr := conn.Read(data)
|
|
if readDataErr != nil {
|
|
if opErr, ok := readDataErr.(*net.OpError); ok {
|
|
slog.Error(fmt.Sprintf("TCP客户端[rAddr:%s]读取数据异常连接可能断开：", rAddr), opErr)
|
|
client.conning = false
|
|
readErr(readDataErr)
|
|
} else if readDataErr == io.EOF {
|
|
slog.Warn(fmt.Sprintf("TCP客户端[rAddr:%s]断开连接：", rAddr))
|
|
client.conning = false
|
|
readErr(readDataErr)
|
|
} else {
|
|
slog.Error(fmt.Sprintf("TCP客户端[rAddr:%s]读数据出错：%s", raddr, readDataErr))
|
|
client.conning = false
|
|
readErr(readDataErr)
|
|
}
|
|
return
|
|
}
|
|
client.conning = true
|
|
handler(l, data)
|
|
}
|
|
}()
|
|
client.conning = true
|
|
return client, nil
|
|
}
|
|
func (c *TcpClient) Close() {
|
|
if c != nil && c.conn != nil {
|
|
slog.Info(fmt.Sprintf("TCP客户端[rAddr:%s]关闭连接", c.conn.RemoteAddr().String()))
|
|
c.ctx()
|
|
c.conn.Close()
|
|
c.conn = nil
|
|
}
|
|
}
|
|
func (c *TcpClient) IsConning() bool {
|
|
if c != nil && c.conn != nil {
|
|
return c.conning
|
|
}
|
|
return false
|
|
}
|
|
func (c *TcpClient) Send(data []byte) error {
|
|
|
|
if c == nil || c.conn == nil {
|
|
//slog.Error("tcp client send error,conn is nil")
|
|
return fmt.Errorf("tcp client send error,conn is nil")
|
|
}
|
|
_, err := c.conn.Write(data)
|
|
if err != nil {
|
|
//slog.Error("tcp client send error", "error", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *TcpClient) RemoteInfo() net.Addr {
|
|
return c.conn.RemoteAddr()
|
|
}
|