/
githubmirror
/
traefik
Обзор
Документация
Войти
/
githubmirror
/
traefik
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
pkg/udp/proxy.go
60 строк
1 KB
romain
Merge current v2.9 into v3.0
15 фев 2023, 13:29
15 фев 2023, 13:29
241fb50
Код
Авторство
О чём код?
package udp import ( "io" "net" "github.com/rs/zerolog/log" ) // Proxy is a reverse-proxy implementation of the Handler interface. type Proxy struct { // TODO: maybe optimize by pre-resolving it at proxy creation time target string } // NewProxy creates a new Proxy. func NewProxy(address string) (*Proxy, error) { return &Proxy{target: address}, nil } // ServeUDP implements the Handler interface. func (p *Proxy) ServeUDP(conn *Conn) { log.Debug().Msgf("Handling UDP stream from %s to %s", conn.rAddr, p.target) // needed because of e.g. server.trackedConnection defer conn.Close() connBackend, err := net.Dial("udp", p.target) if err != nil { log.Error().Err(err).Msg("Error while dialing backend") return } // maybe not needed, but just in case defer connBackend.Close() errChan := make(chan error) go connCopy(conn, connBackend, errChan) go connCopy(connBackend, conn, errChan) err = <-errChan if err != nil { log.Error().Err(err).Msg("Error while handling UDP stream") } <-errChan } func connCopy(dst io.WriteCloser, src io.Reader, errCh chan error) { // The buffer is initialized to the maximum UDP datagram size, // to make sure that the whole UDP datagram is read or written atomically (no data is discarded). buffer := make([]byte, maxDatagramSize) _, err := io.CopyBuffer(dst, src, buffer) errCh <- err if err := dst.Close(); err != nil { log.Debug().Err(err).Msg("Error while terminating UDP stream") } }