代码拉取完成,页面将自动刷新
package main
import (
"fmt"
"net"
log "github.com/sirupsen/logrus"
)
// UDPConfig udp config
type UDPConfig struct {
IP string `json:"ip,omitempty"`
Port uint16 `json:"port,omitempty"`
AuxProperty map[string]interface{} `json:"auxProperty,omitempty"`
}
// UDPReceiver udp receiver handle
type UDPReceiver struct {
config UDPConfig
conn *net.UDPConn
ch chan rdata
}
// UDPData udp data
type UDPData struct {
properties map[string]interface{}
data []byte
}
// Data implements Source interface
func (d UDPData) Data() []byte {
return d.data
}
// GetProperty implements Source interface
func (d UDPData) GetProperty(prop string) (interface{}, bool) {
p, ok := d.properties[prop]
return p, ok
}
// Open open receiver
func (d *UDPReceiver) Open() error {
bind := fmt.Sprintf("%s:%d", d.config.IP, d.config.Port)
s4001addr, err := net.ResolveUDPAddr("udp", bind)
if err != nil {
return err
}
d.conn, err = net.ListenUDP("udp", s4001addr)
if err != nil {
return err
}
log.WithFields(log.Fields{"config": d.config}).Info("udp listen open")
// defer s4001.Close() // nolint
return nil
}
// Start start the receiver
func (d *UDPReceiver) Start() error {
log.Info("udp listen start")
go func() {
for {
var buf [4096]byte
num, peer, err := d.conn.ReadFromUDP(buf[0:])
if err != nil {
log.WithFields(log.Fields{"err": err}).Error("error receive data")
} else {
log.WithFields(log.Fields{"peer": peer, "bytes": num}).Info("received data")
}
props := map[string]interface{}{
"peer": peer,
"config": d.config,
}
for k, v := range d.config.AuxProperty {
props[k] = v
}
d.ch <- UDPData{
properties: props,
data: buf[0:num],
}
}
}()
return nil
}
// NewUDPReceiver create new receiver instance
func NewUDPReceiver(params UDPConfig, ch chan rdata) (UDPReceiver, error) {
d := UDPReceiver{
config: params,
ch: ch,
}
return d, nil
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。