Created
December 15, 2019 13:31
-
-
Save nhoclove/d57f6582c02b12eac651184f8044b7ff to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
func (srv *Server) Serve(l net.Listener) error { | |
if fn := testHookServerServe; fn != nil { | |
fn(srv, l) // call hook with unwrapped listener | |
} | |
origListener := l | |
l = &onceCloseListener{Listener: l} | |
defer l.Close() | |
if err := srv.setupHTTP2_Serve(); err != nil { | |
return err | |
} | |
if !srv.trackListener(&l, true) { | |
return ErrServerClosed | |
} | |
defer srv.trackListener(&l, false) | |
var tempDelay time.Duration // how long to sleep on accept failure | |
baseCtx := context.Background() | |
if srv.BaseContext != nil { | |
baseCtx = srv.BaseContext(origListener) | |
if baseCtx == nil { | |
panic("BaseContext returned a nil context") | |
} | |
} | |
ctx := context.WithValue(baseCtx, ServerContextKey, srv) | |
for { | |
rw, e := l.Accept() | |
if e != nil { | |
select { | |
case <-srv.getDoneChan(): | |
return ErrServerClosed | |
default: | |
} | |
if ne, ok := e.(net.Error); ok && ne.Temporary() { | |
if tempDelay == 0 { | |
tempDelay = 5 * time.Millisecond | |
} else { | |
tempDelay *= 2 | |
} | |
if max := 1 * time.Second; tempDelay > max { | |
tempDelay = max | |
} | |
srv.logf("http: Accept error: %v; retrying in %v", e, tempDelay) | |
time.Sleep(tempDelay) | |
continue | |
} | |
return e | |
} | |
if cc := srv.ConnContext; cc != nil { | |
ctx = cc(ctx, rw) | |
if ctx == nil { | |
panic("ConnContext returned nil") | |
} | |
} | |
tempDelay = 0 | |
c := srv.newConn(rw) | |
c.setState(c.rwc, StateNew) // before Serve can return | |
go c.serve(ctx) | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment