67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
package server
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
|
|
"github.com/admin-else/strom/proto"
|
|
"github.com/admin-else/strom/proto_base"
|
|
)
|
|
|
|
func Servee(c net.Conn) (ret *proto.Conn) {
|
|
ret = &proto.Conn{}
|
|
_ = ret.SetVersion("1.21.8") // cant fail
|
|
ret.SetState(proto_base.Handshaking)
|
|
ret.CompressionThreshold = -1
|
|
ret.Actor = proto_base.Server
|
|
ret.Conn = c
|
|
ret.R = c
|
|
ret.W = c
|
|
return
|
|
}
|
|
|
|
func ServeClient(cNet net.Conn, factory func(c *proto.Conn) (h any, err error)) (err error) {
|
|
c := Servee(cNet)
|
|
h, err := factory(c)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = c.Start(h)
|
|
return
|
|
}
|
|
|
|
func StartServerWithFactory(listenAddr string, factory func(c *proto.Conn) (h any, err error)) (err error) {
|
|
l, err := net.Listen("tcp", listenAddr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
var cNet net.Conn
|
|
for {
|
|
cNet, err = l.Accept()
|
|
if err != nil {
|
|
return
|
|
}
|
|
go ServeClient(cNet, factory)
|
|
}
|
|
}
|
|
|
|
func StartServerWithOnConn(listenAddr string, onConn func(c *proto.Conn) (err error)) (err error) {
|
|
l, err := net.Listen("tcp", listenAddr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
for {
|
|
var cNet net.Conn
|
|
cNet, err = l.Accept()
|
|
if err != nil {
|
|
return
|
|
}
|
|
go func() {
|
|
connErr := onConn(Servee(cNet))
|
|
if connErr != nil {
|
|
log.Println(connErr)
|
|
}
|
|
}()
|
|
}
|
|
}
|