代码拉取完成,页面将自动刷新
// Copyright 2019 Yang ZhongXi
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
// Unless required by applicable law or agreed to in writing, software
// distributed under the License 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 core
import (
"errors"
"net"
"github.com/sirupsen/logrus"
)
// UDPStream UDP数据收发流
type UDPStream struct {
raddr *net.UDPAddr
conn *net.UDPConn
isOpened bool
receiver func(data []byte)
laddr *net.UDPAddr
}
// NewUDPStream 实例化, address: 对端地址, laddr: 本端地址
func NewUDPStream(address string, laddr string) *UDPStream {
stream := &UDPStream{}
var err error
stream.raddr, err = net.ResolveUDPAddr("udp", address)
if err != nil {
return nil
}
stream.laddr, err = net.ResolveUDPAddr("udp", laddr)
if err != nil {
return nil
}
return stream
}
// Open 打开UDP
func (stream *UDPStream) Open() error {
logrus.Debug("打开UDP")
if stream.isOpened {
return errors.New("UDP Socket已打开")
}
var err error
stream.conn, err = net.ListenUDP("udp", stream.laddr)
if err != nil {
return err
}
stream.isOpened = true
go stream.doWork()
return err
}
// Close 关闭UDP socket
func (stream *UDPStream) Close() error {
if !stream.isOpened {
return errors.New("UDP Socket已关闭")
}
stream.isOpened = false
_ = stream.conn.Close()
return nil
}
// Write 写入数据
func (stream *UDPStream) Write(data []byte) error {
if !stream.isOpened {
return errors.New("未连接")
}
_, err := stream.conn.WriteToUDP(data, stream.raddr)
return err
}
// IsOpened 判断是否已打开
func (stream *UDPStream) IsOpened() bool {
return stream.isOpened
}
// SetReceiver 设置数据监听器
func (stream *UDPStream) SetReceiver(receiver func(data []byte)) {
stream.receiver = receiver
}
func (stream *UDPStream) doWork() {
buffer := make([]byte, 1024)
for stream.isOpened {
rLen, _, err := stream.conn.ReadFromUDP(buffer)
if err != nil {
break
}
if stream.receiver != nil {
stream.receiver(buffer[:rLen])
}
}
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。