|
|
package nnet
|
|
|
|
|
|
import (
|
|
|
"github.com/noahlann/nnet/internal/client"
|
|
|
clientpkg "github.com/noahlann/nnet/pkg/client"
|
|
|
)
|
|
|
|
|
|
// Client 客户端类型别名
|
|
|
type Client = clientpkg.Client
|
|
|
|
|
|
// ClientConfig 客户端配置类型别名
|
|
|
type ClientConfig = clientpkg.Config
|
|
|
|
|
|
// ClientTLSConfig 客户端TLS配置类型别名
|
|
|
type ClientTLSConfig = clientpkg.TLSConfig
|
|
|
|
|
|
// ClientPool 客户端连接池类型别名
|
|
|
type ClientPool = client.Pool
|
|
|
|
|
|
// ClientPoolConfig 客户端连接池配置类型别名
|
|
|
type ClientPoolConfig = client.PoolConfig
|
|
|
|
|
|
// NewClientPool 创建客户端连接池
|
|
|
func NewClientPool(config *ClientPoolConfig) (*ClientPool, error) {
|
|
|
return client.NewPool(config)
|
|
|
}
|
|
|
|
|
|
// DefaultClientPoolConfig 返回默认客户端连接池配置
|
|
|
func DefaultClientPoolConfig() *ClientPoolConfig {
|
|
|
return client.DefaultPoolConfig()
|
|
|
}
|
|
|
|
|
|
// NewClient 创建客户端（根据配置或地址自动选择传输层协议）
|
|
|
// 传输层协议优先级：
|
|
|
// 1. 如果配置中指定了TransportProtocol，使用指定的协议
|
|
|
// 2. 否则根据地址前缀判断（tcp://, udp://, ws://, wss://, unix://, serial://）
|
|
|
// 3. 如果无法识别，默认使用TCP
|
|
|
// 应用层协议通过 ApplicationProtocol 配置指定（如nnet）
|
|
|
func NewClient(config *ClientConfig) Client {
|
|
|
if config == nil {
|
|
|
config = clientpkg.DefaultConfig()
|
|
|
}
|
|
|
|
|
|
// 优先使用配置中指定的传输层协议
|
|
|
if config.TransportProtocol != "" {
|
|
|
return newClientByProtocol(config, config.TransportProtocol)
|
|
|
}
|
|
|
|
|
|
// 根据地址前缀判断传输层协议
|
|
|
return newClientByAddr(config)
|
|
|
}
|
|
|
|
|
|
// newClientByProtocol 根据指定的传输层协议创建客户端
|
|
|
func newClientByProtocol(config *ClientConfig, protocol string) Client {
|
|
|
switch protocol {
|
|
|
case "tcp":
|
|
|
return client.NewTCPClient(config)
|
|
|
case "udp":
|
|
|
return client.NewUDPClient(config)
|
|
|
case "websocket", "ws", "wss":
|
|
|
return client.NewWebSocketClient(config)
|
|
|
case "unix":
|
|
|
return client.NewUnixClient(config)
|
|
|
case "serial":
|
|
|
return client.NewSerialClient(config)
|
|
|
default:
|
|
|
// 默认使用TCP客户端
|
|
|
return client.NewTCPClient(config)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// newClientByAddr 根据地址前缀判断传输层协议
|
|
|
func newClientByAddr(config *ClientConfig) Client {
|
|
|
addr := config.Addr
|
|
|
if len(addr) >= 8 {
|
|
|
if addr[:8] == "serial://" {
|
|
|
return client.NewSerialClient(config)
|
|
|
}
|
|
|
if len(addr) >= 7 && addr[:7] == "serial:" {
|
|
|
return client.NewSerialClient(config)
|
|
|
}
|
|
|
}
|
|
|
if len(addr) >= 6 {
|
|
|
if addr[:6] == "wss://" {
|
|
|
return client.NewWebSocketClient(config)
|
|
|
}
|
|
|
}
|
|
|
if len(addr) >= 5 {
|
|
|
if addr[:5] == "ws://" {
|
|
|
return client.NewWebSocketClient(config)
|
|
|
}
|
|
|
if addr[:5] == "udp://" {
|
|
|
return client.NewUDPClient(config)
|
|
|
}
|
|
|
if addr[:5] == "unix:" {
|
|
|
return client.NewUnixClient(config)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 默认使用TCP客户端
|
|
|
return client.NewTCPClient(config)
|
|
|
}
|