LBRYCommunity/tracker
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

Merge pull request #442 from cenkalti/split-http

Browse source 
split listen&serve for http
This commit is contained in:
mrd0ll4r 2019-01-13 20:07:11 +03:00 committed by GitHub
commit 5f99a7e778
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 26 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
frontend/http/frontend.go
View file

@ -146,9 +146,27 @@ func NewFrontend(logic frontend.TrackerLogic, provided Config) (*Frontend, error
return nil, errors.New("must specify https_addr when using tls_cert_path and tls_key_path")
}
var listenerHTTP, listenerHTTPS net.Listener
var err error
if cfg.Addr != "" {
listenerHTTP, err = net.Listen("tcp", f.Addr)
if err != nil {
return nil, err
}
}
if cfg.HTTPSAddr != "" {
listenerHTTPS, err = net.Listen("tcp", f.HTTPSAddr)
if err != nil {
if listenerHTTP != nil {
listenerHTTP.Close()
}
return nil, err
}
}
if cfg.Addr != "" {
go func() {
if err := f.listenAndServe(); err != nil {
if err := f.serveHTTP(listenerHTTP); err != nil {
log.Fatal("failed while serving http", log.Err(err))
}
}()
@ -156,7 +174,7 @@ func NewFrontend(logic frontend.TrackerLogic, provided Config) (*Frontend, error
if cfg.HTTPSAddr != "" {
go func() {
if err := f.listenAndServeTLS(); err != nil {
if err := f.serveHTTPS(listenerHTTPS); err != nil {
log.Fatal("failed while serving https", log.Err(err))
}
}()
@ -203,9 +221,9 @@ func (f *Frontend) handler() http.Handler {
return router
}
// listenAndServe blocks while listening and serving non-TLS HTTP BitTorrent
// serveHTTP blocks while listening and serving non-TLS HTTP BitTorrent
// requests until Stop() is called or an error is returned.
func (f *Frontend) listenAndServe() error {
func (f *Frontend) serveHTTP(l net.Listener) error {
f.srv = &http.Server{
Addr: f.Addr,
Handler: f.handler(),
@ -217,15 +235,15 @@ func (f *Frontend) listenAndServe() error {
f.srv.SetKeepAlivesEnabled(f.EnableKeepAlive)
// Start the HTTP server.
if err := f.srv.ListenAndServe(); err != http.ErrServerClosed {
if err := f.srv.Serve(l); err != http.ErrServerClosed {
return err
}
return nil
}
// listenAndServeTLS blocks while listening and serving TLS HTTP BitTorrent
// serveHTTPS blocks while listening and serving TLS HTTP BitTorrent
// requests until Stop() is called or an error is returned.
func (f *Frontend) listenAndServeTLS() error {
func (f *Frontend) serveHTTPS(l net.Listener) error {
f.tlsSrv = &http.Server{
Addr: f.HTTPSAddr,
TLSConfig: f.tlsCfg,
@ -234,11 +252,10 @@ func (f *Frontend) listenAndServeTLS() error {
WriteTimeout: f.WriteTimeout,
}
// Disable KeepAlives.
f.tlsSrv.SetKeepAlivesEnabled(f.EnableKeepAlive)
// Start the HTTP server.
if err := f.tlsSrv.ListenAndServeTLS("", ""); err != http.ErrServerClosed {
if err := f.tlsSrv.ServeTLS(l, "", ""); err != http.ErrServerClosed {
return err
}
return nil