2 Star 0 Fork 0

Admin/bd_sender

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
udp_receiver.go 1.95 KB
一键复制 编辑 原始数据 按行查看 历史
钟林峰 提交于 2019-02-26 13:38 +08:00 . deployed to hezhong1
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
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/shipnet_masterclock/bd_sender.git
git@gitee.com:shipnet_masterclock/bd_sender.git
shipnet_masterclock
bd_sender
bd_sender
master

搜索帮助