Altair-sh
61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
package server
|
|
|
|
import (
|
|
"errors"
|
|
"net"
|
|
"serveg/internal/log"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type Server struct {
|
|
Conf *Config
|
|
logger log.LoggerContext
|
|
// used in sockets to sed read/write deadline
|
|
timeout time.Duration
|
|
listener net.Listener
|
|
wg sync.WaitGroup
|
|
}
|
|
|
|
const defaultTimeout = 30 * time.Second
|
|
|
|
func NewServer(conf *Config) *Server {
|
|
return &Server{
|
|
Conf: conf,
|
|
logger: log.LoggerContext{Name: "server"},
|
|
timeout: defaultTimeout,
|
|
}
|
|
}
|
|
|
|
func (srv *Server) StartGoroutine() (err error) {
|
|
srv.logger.Info("listening at ", srv.Conf.Addr)
|
|
srv.listener, err = net.Listen("tcp", srv.Conf.Addr)
|
|
if err == nil {
|
|
srv.wg.Go(srv.serve)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (srv *Server) serve() {
|
|
srv.logger.Info("server started")
|
|
for i := int64(0); ; i++ {
|
|
accepted, err := srv.listener.Accept()
|
|
if err != nil {
|
|
if errors.Is(err, net.ErrClosed) {
|
|
srv.logger.Info("listener socket closed")
|
|
return
|
|
}
|
|
srv.logger.Warn(err)
|
|
continue
|
|
}
|
|
conn := newConnection(srv, i, accepted.(*net.TCPConn))
|
|
go conn.handle()
|
|
}
|
|
}
|
|
|
|
func (srv *Server) Stop() {
|
|
srv.logger.Info("stopping server")
|
|
_ = srv.listener.Close()
|
|
srv.wg.Wait()
|
|
}
|