94 lines
1.9 KiB
Go
94 lines
1.9 KiB
Go
package at_common
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
)
|
|
|
|
type ServerChannel struct {
|
|
ip string
|
|
port int
|
|
Listener *net.Listener
|
|
UDPListener *net.UDPConn
|
|
errAcceptHandler func(err error)
|
|
}
|
|
|
|
func NewServerChannel(ip string, port int) ServerChannel {
|
|
return ServerChannel{
|
|
ip: ip,
|
|
port: port,
|
|
errAcceptHandler: func(err error) {
|
|
fmt.Printf("accept error , ERR:%s", err)
|
|
},
|
|
}
|
|
}
|
|
func (sc *ServerChannel) SetErrAcceptHandler(fn func(err error)) {
|
|
sc.errAcceptHandler = fn
|
|
}
|
|
func (sc *ServerChannel) ListenTls(fn func(channel MessageChannel, conn net.Conn)) (err error) {
|
|
sc.Listener, err = ListenTls(sc.ip, sc.port)
|
|
if err == nil {
|
|
go func() {
|
|
for {
|
|
var conn net.Conn
|
|
conn, err = (*sc.Listener).Accept()
|
|
if err == nil {
|
|
go func() {
|
|
fn(NewMessageChannel(&conn), conn)
|
|
}()
|
|
} else {
|
|
sc.errAcceptHandler(err)
|
|
(*sc.Listener).Close()
|
|
break
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
return
|
|
}
|
|
|
|
func (sc *ServerChannel) ListenTCP(fn func(channel MessageChannel, conn net.Conn)) (err error) {
|
|
l, err := net.Listen("tcp", fmt.Sprintf("%s:%d", sc.ip, sc.port))
|
|
if err == nil {
|
|
sc.Listener = &l
|
|
go func() {
|
|
for {
|
|
var conn net.Conn
|
|
conn, err = (*sc.Listener).Accept()
|
|
if err == nil {
|
|
go func() {
|
|
fn(NewMessageChannel(&conn), conn)
|
|
}()
|
|
} else {
|
|
sc.errAcceptHandler(err)
|
|
break
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
return
|
|
}
|
|
func (sc *ServerChannel) ListenUDP(fn func(packet []byte, localAddr, srcAddr *net.UDPAddr)) (err error) {
|
|
addr := &net.UDPAddr{IP: net.ParseIP(sc.ip), Port: sc.port}
|
|
l, err := net.ListenUDP("udp", addr)
|
|
if err == nil {
|
|
sc.UDPListener = l
|
|
go func() {
|
|
for {
|
|
var buf = make([]byte, 2048)
|
|
n, srcAddr, err := (*sc.UDPListener).ReadFromUDP(buf)
|
|
if err == nil {
|
|
packet := buf[0:n]
|
|
go func() {
|
|
fn(packet, addr, srcAddr)
|
|
}()
|
|
} else {
|
|
sc.errAcceptHandler(err)
|
|
break
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
return
|
|
}
|