1 Star 0 Fork 0

cgutech-golang/core

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
stream_tcp.go 2.02 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"
)
// TCPStream TCP数据写入读取底层实现
type TCPStream struct {
conn *net.TCPConn
isOpened bool
receiver func(data []byte)
eventReporter func(err error)
}
// NewTCPStream 实例化
func NewTCPStream(conn *net.TCPConn, eventReporter func(err error)) *TCPStream {
return &TCPStream{conn: conn, eventReporter: eventReporter}
}
// Open 打开端口,并开启数据接收线程
func (t *TCPStream) Open() error {
if t.isOpened {
return errors.New("连接已打开")
}
go t.doWork()
return nil
}
// Close 关闭
func (t *TCPStream) Close() {
if !t.isOpened {
return
}
t.isOpened = false
t.conn.Close()
}
// Write 写入数据
func (t *TCPStream) Write(data []byte) error {
if !t.isOpened {
return errors.New("连接未打开")
}
_, err := t.conn.Write(data)
return err
}
// IsOpened 判断是否已打开
func (t *TCPStream) IsOpened() bool {
return t.isOpened
}
// SetReceiver 设置数据接收器
func (t *TCPStream) SetReceiver(receiver func(data []byte)) {
t.receiver = receiver
}
func (t *TCPStream) doWork() {
for t.isOpened {
buffer := make([]byte, 256)
l, err := t.conn.Read(buffer)
if err != nil {
if t.eventReporter != nil {
t.eventReporter(errors.New("读取数据失败:" + err.Error()))
}
break
}
if t.receiver != nil {
t.receiver(buffer[:l])
}
}
t.Close()
}
// TCPClientStream TCP客户端实现
type TCPClientStream struct {
stream *TCPStream
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/cgutech-golang/core.git
git@gitee.com:cgutech-golang/core.git
cgutech-golang
core
core
v1.0.5

搜索帮助