traefik/pkg/tcp/switcher.go
Ludovic Fernandez f1b085fa36 Move code to pkg
2019-03-15 09:42:03 +01:00

28 lines
546 B
Go

package tcp
import (
"net"
"github.com/containous/traefik/pkg/safe"
)
// HandlerSwitcher is a TCP handler switcher
type HandlerSwitcher struct {
router safe.Safe
}
// ServeTCP forwards the TCP connection to the current active handler
func (s *HandlerSwitcher) ServeTCP(conn net.Conn) {
handler := s.router.Get()
h, ok := handler.(Handler)
if ok {
h.ServeTCP(conn)
} else {
conn.Close()
}
}
// Switch sets the new TCP handler to use for new connections
func (s *HandlerSwitcher) Switch(handler Handler) {
s.router.Set(handler)
}