98 lines
1.8 KiB
Go
98 lines
1.8 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/go-gost/gost/pkg/admission"
|
|
"github.com/go-gost/gost/pkg/handler"
|
|
"github.com/go-gost/gost/pkg/listener"
|
|
"github.com/go-gost/gost/pkg/logger"
|
|
)
|
|
|
|
type options struct {
|
|
admission admission.Admission
|
|
logger logger.Logger
|
|
}
|
|
|
|
type Option func(opts *options)
|
|
|
|
func AdmissionOption(admission admission.Admission) Option {
|
|
return func(opts *options) {
|
|
opts.admission = admission
|
|
}
|
|
}
|
|
|
|
func LoggerOption(logger logger.Logger) Option {
|
|
return func(opts *options) {
|
|
opts.logger = logger
|
|
}
|
|
}
|
|
|
|
type Service interface {
|
|
Serve() error
|
|
Addr() net.Addr
|
|
Close() error
|
|
}
|
|
|
|
type service struct {
|
|
listener listener.Listener
|
|
handler handler.Handler
|
|
options options
|
|
}
|
|
|
|
func NewService(ln listener.Listener, h handler.Handler, opts ...Option) Service {
|
|
var options options
|
|
for _, opt := range opts {
|
|
opt(&options)
|
|
}
|
|
return &service{
|
|
listener: ln,
|
|
handler: h,
|
|
options: options,
|
|
}
|
|
}
|
|
|
|
func (s *service) Addr() net.Addr {
|
|
return s.listener.Addr()
|
|
}
|
|
|
|
func (s *service) Close() error {
|
|
return s.listener.Close()
|
|
}
|
|
|
|
func (s *service) Serve() error {
|
|
var tempDelay time.Duration
|
|
for {
|
|
conn, e := s.listener.Accept()
|
|
if e != nil {
|
|
if ne, ok := e.(net.Error); ok && ne.Temporary() {
|
|
if tempDelay == 0 {
|
|
tempDelay = 1 * time.Second
|
|
} else {
|
|
tempDelay *= 2
|
|
}
|
|
if max := 5 * time.Second; tempDelay > max {
|
|
tempDelay = max
|
|
}
|
|
s.options.logger.Warnf("accept: %v, retrying in %v", e, tempDelay)
|
|
time.Sleep(tempDelay)
|
|
continue
|
|
}
|
|
s.options.logger.Errorf("accept: %v", e)
|
|
return e
|
|
}
|
|
tempDelay = 0
|
|
|
|
if s.options.admission != nil &&
|
|
!s.options.admission.Admit(conn.RemoteAddr().String()) {
|
|
s.options.logger.Infof("admission: %s is denied", conn.RemoteAddr())
|
|
conn.Close()
|
|
continue
|
|
}
|
|
|
|
go s.handler.Handle(context.Background(), conn)
|
|
}
|
|
}
|