84 lines
1.4 KiB
Go
84 lines
1.4 KiB
Go
package socket
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/panjf2000/gnet/v2"
|
|
"github.com/panjf2000/gnet/v2/pkg/logging"
|
|
)
|
|
|
|
func (s *Server) Boot() error {
|
|
err := gnet.Run(s, s.addr,
|
|
gnet.WithMulticore(true),
|
|
gnet.WithSocketRecvBuffer(s.bufferSize))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) Stop() error {
|
|
_ = s.eng.Stop(context.Background())
|
|
s.workerPool.Release()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) OnBoot(eng gnet.Engine) gnet.Action {
|
|
s.eng = eng
|
|
|
|
logging.Infof("syslog server is listening on %s\n", s.addr)
|
|
|
|
return gnet.None
|
|
}
|
|
|
|
func (s *Server) OnTraffic(conn gnet.Conn) (action gnet.Action) {
|
|
|
|
|
|
if s.network == "tcp" {
|
|
return s.handleTcp(conn)
|
|
}
|
|
|
|
|
|
|
|
return gnet.None
|
|
}
|
|
|
|
|
|
func (s *Server) handleTcp(conn gnet.Conn) (action gnet.Action) {
|
|
for {
|
|
data, err := s.codec.Decode(conn)
|
|
if err != nil {
|
|
break
|
|
}
|
|
|
|
//client := conn.RemoteAddr().String()
|
|
_ = s.workerPool.Submit(func() {
|
|
s.parser(conn, data)
|
|
})
|
|
|
|
return gnet.None
|
|
}
|
|
|
|
if conn.InboundBuffered() > 0 {
|
|
if err := conn.Wake(nil); err != nil { // wake up the connection manually to avoid missing the leftover data
|
|
logging.Errorf("failed to wake up the connection, %v", err)
|
|
return gnet.Close
|
|
}
|
|
}
|
|
|
|
return gnet.None
|
|
|
|
}
|
|
|
|
func (s *Server) parser(c gnet.Conn, line []byte) {
|
|
|
|
s.handler.Handle(line)
|
|
}
|
|
func (s *Server) Start() {
|
|
|
|
err := gnet.Run(s, s.network+"://"+s.addr, gnet.WithMulticore(s.multicore))
|
|
logging.Infof("server exits with error: %v", err)
|
|
}
|