package tcpx

import (
	"net"
)

/* tcp 连接定义 */

// TCPType tcp连接类型
type TCPType = int

const (
	// TCPSocketType tcp连接
	TCPSocketType = iota
	// WebSocketType WebSocket连接
	WebSocketType
)

// Conn 自定义连接对象结构体,可以存储tcp或webSocket连接对象
type Conn struct {
	net.Conn
	connType TCPType // 连接对象类型
	clientID string  // 客户端id
}

// SendMessage 发送消息，参数为自己报文结构体
func (c *Conn) SendMessage(data []byte) (int, error) {
	return c.Write(data)
}

// GetClientID 获取当前连接id
func (c *Conn) GetClientID() string {
	return c.clientID
}

// GetConnType 获取连接类型
func (c *Conn) GetConnType() TCPType {
	return c.connType
}