/
udp.go
52 lines (45 loc) · 1.38 KB
/
udp.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
// Copyright 2018-2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance with the License. A copy of the License is located at
//
// http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and limitations under the License.
package udp
import (
"net"
"os"
"github.com/aws/aws-xray-daemon/daemon/socketconn"
log "github.com/cihub/seelog"
)
// UDP defines UDP socket connection.
type UDP struct {
socket *net.UDPConn
}
// New returns new instance of UDP.
func New(udpAddress string) socketconn.SocketConn {
log.Debugf("Listening on UDP %v", udpAddress)
addr, err := net.ResolveUDPAddr("udp", udpAddress)
if err != nil {
log.Errorf("%v", err)
os.Exit(1)
}
sock, err := net.ListenUDP("udp", addr)
if err != nil {
log.Errorf("%v", err)
os.Exit(1)
}
return UDP{
socket: sock,
}
}
// Read returns number of bytes read from the UDP connection.
func (conn UDP) Read(b []byte) (int, error) {
rlen, _, err := conn.socket.ReadFromUDP(b)
return rlen, err
}
// Close closes current UDP connection.
func (conn UDP) Close() {
conn.socket.Close()
}