package tcp import ( "net" "github.com/containous/traefik/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) }