mirror of
https://github.com/pomerium/pomerium.git
synced 2025-06-15 17:22:56 +02:00
new tracing system
This commit is contained in:
parent
8f36870650
commit
b9065b6a55
130 changed files with 7928 additions and 1836 deletions
40
internal/testenv/snippets/shutdown.go
Normal file
40
internal/testenv/snippets/shutdown.go
Normal file
|
@ -0,0 +1,40 @@
|
|||
package snippets
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"net/http"
|
||||
|
||||
"github.com/pomerium/pomerium/internal/log"
|
||||
"github.com/pomerium/pomerium/internal/testenv"
|
||||
)
|
||||
|
||||
func RunWithDelayedShutdown(ctx context.Context, serve func() error, stop func()) func() error {
|
||||
env := testenv.EnvFromContext(ctx)
|
||||
|
||||
stopping := make(chan struct{})
|
||||
serveExited := make(chan error, 1)
|
||||
env.OnStateChanged(testenv.Stopping, func() {
|
||||
close(stopping)
|
||||
})
|
||||
cancel := env.OnStateChanged(testenv.Stopped, func() {
|
||||
stop()
|
||||
if err := <-serveExited; err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||||
log.Ctx(ctx).Err(err).Msg("error stopping server")
|
||||
}
|
||||
})
|
||||
go func() {
|
||||
serveExited <- serve()
|
||||
close(serveExited)
|
||||
}()
|
||||
|
||||
return func() error {
|
||||
select {
|
||||
case <-stopping:
|
||||
return nil
|
||||
case err := <-serveExited:
|
||||
cancel()
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue