mirror of
https://github.com/traefik/traefik.git
synced 2026-06-17 19:09:29 +03:00
Julien Salleyron
30 lines
525 B
Go
30 lines
525 B
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
)
|
|
|
|
type connContextFunc func(context.Context, net.Conn) context.Context
|
|
|
|
type multipleConnContext struct {
|
|
fns []connContextFunc
|
|
}
|
|
|
|
func (m *multipleConnContext) AddConnContextFunc(fn connContextFunc) {
|
|
m.fns = append(m.fns, fn)
|
|
}
|
|
|
|
func (m *multipleConnContext) Build() connContextFunc {
|
|
if len(m.fns) == 0 {
|
|
return nil
|
|
}
|
|
|
|
return func(ctx context.Context, c net.Conn) context.Context {
|
|
for _, contextFunc := range m.fns {
|
|
ctx = contextFunc(ctx, c)
|
|
}
|
|
return ctx
|
|
}
|
|
}
|