1 Star 0 Fork 0

cgutech-golang/core

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
stream_udp.go 2.34 KB
一键复制 编辑 原始数据 按行查看 历史
// 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])
}
}
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/cgutech-golang/core.git
git@gitee.com:cgutech-golang/core.git
cgutech-golang
core
core
v1.0.5

搜索帮助