42 lines
946 B
Go
42 lines
946 B
Go
|
package statsd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"net"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// udpWriter is an internal class wrapping around management of UDP connection
|
||
|
|
type udpWriter struct {
|
||
|
|
conn net.Conn
|
||
|
|
}
|
||
|
|
|
||
|
|
// New returns a pointer to a new udpWriter given an addr in the format "hostname:port".
|
||
|
|
func newUdpWriter(addr string) (*udpWriter, error) {
|
||
|
|
udpAddr, err := net.ResolveUDPAddr("udp", addr)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
conn, err := net.DialUDP("udp", nil, udpAddr)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
writer := &udpWriter{conn: conn}
|
||
|
|
return writer, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetWriteTimeout is not needed for UDP, returns error
|
||
|
|
func (w *udpWriter) SetWriteTimeout(d time.Duration) error {
|
||
|
|
return errors.New("SetWriteTimeout: not supported for UDP connections")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Write data to the UDP connection with no error handling
|
||
|
|
func (w *udpWriter) Write(data []byte) error {
|
||
|
|
_, e := w.conn.Write(data)
|
||
|
|
return e
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *udpWriter) Close() error {
|
||
|
|
return w.conn.Close()
|
||
|
|
}
|