Ensure Zipkin shutdown correctness (#2765)

* Fix Zipkin shutdown

* Fix zipkin shutdown

* Remove nil check for http Serve on OC and OTLP

* Switch from channel to WaitGroup for Zipkin shutdown
This commit is contained in:
Paulo Janotti 2021-03-22 18:11:18 -07:00 committed by GitHub
parent 49ebadd104
commit c7e9b68f84
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 14 additions and 12 deletions

View File

@ -261,7 +261,7 @@ func (ocr *ocReceiver) startServer(host component.Host) error {
}
}()
go func() {
if errHTTP := ocr.httpServer().Serve(httpL); errHTTP != nil {
if errHTTP := ocr.httpServer().Serve(httpL); errHTTP != http.ErrServerClosed {
host.ReportFatalError(errHTTP)
}
}()

View File

@ -119,7 +119,7 @@ func (r *otlpReceiver) startHTTPServer(cfg *confighttp.HTTPServerSettings, host
go func() {
defer r.shutdownWG.Done()
if errHTTP := r.serverHTTP.Serve(hln); errHTTP != nil && errHTTP != http.ErrServerClosed {
if errHTTP := r.serverHTTP.Serve(hln); errHTTP != http.ErrServerClosed {
host.ReportFatalError(errHTTP)
}
}()

View File

@ -59,10 +59,11 @@ type ZipkinReceiver struct {
nextConsumer consumer.TracesConsumer
instanceName string
startOnce sync.Once
stopOnce sync.Once
server *http.Server
config *Config
startOnce sync.Once
stopOnce sync.Once
shutdownWG sync.WaitGroup
server *http.Server
config *Config
}
var _ http.Handler = (*ZipkinReceiver)(nil)
@ -99,13 +100,14 @@ func (zr *ZipkinReceiver) Start(ctx context.Context, host component.Host) error
var listener net.Listener
listener, err = zr.config.HTTPServerSettings.ToListener()
if err != nil {
host.ReportFatalError(err)
return
}
zr.shutdownWG.Add(1)
go func() {
err = zr.server.Serve(listener)
if err != nil {
host.ReportFatalError(err)
defer zr.shutdownWG.Done()
if errHTTP := zr.server.Serve(listener); errHTTP != http.ErrServerClosed {
host.ReportFatalError(errHTTP)
}
}()
})
@ -165,6 +167,7 @@ func (zr *ZipkinReceiver) Shutdown(context.Context) error {
var err = componenterror.ErrAlreadyStopped
zr.stopOnce.Do(func() {
err = zr.server.Close()
zr.shutdownWG.Wait()
})
return err
}

View File

@ -75,8 +75,7 @@ func TestDefaultReceivers(t *testing.T) {
},
},
{
receiver: "zipkin",
skipLifecyle: true, // TODO: Upcoming PR to fix zipkin lifecycle.
receiver: "zipkin",
},
}