go-gost/gost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update ssu

Browse Source
This commit is contained in:
ginuerzh
2021-11-08 22:06:42 +08:00
parent 10f53e18a0
commit 92dc87830f
13 changed files with 328 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/service/service.go
View File

@ -7,11 +7,13 @@ import (
"github.com/go-gost/gost/pkg/handler"
"github.com/go-gost/gost/pkg/listener"
"github.com/go-gost/gost/pkg/logger"
)
type Service struct {
listener listener.Listener
handler handler.Handler
logger logger.Logger
}
func (s *Service) WithListener(ln listener.Listener) *Service {
@ -24,6 +26,11 @@ func (s *Service) WithHandler(h handler.Handler) *Service {
return s
}
func (s *Service) WithLogger(logger logger.Logger) *Service {
s.logger = logger
return s
}
func (s *Service) Addr() net.Addr {
return s.listener.Addr()
}
@ -50,7 +57,7 @@ func (s *Service) serve() error {
if max := 1 * time.Second; tempDelay > max {
tempDelay = max
}
// log.Logf("server: Accept error: %v; retrying in %v", e, tempDelay)
s.logger.Warnf("accept: %v, retrying in %v", e, tempDelay)
time.Sleep(tempDelay)
continue
}