utils.go

最新推荐文章于 2024-12-06 09:46:35 发布

原创最新推荐文章于 2024-12-06 09:46:35 发布 · 392 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#utils.go

项目练习专栏收录该内容

25 篇文章

订阅专栏

这是一个Go语言编写的utils包，包含了一个`Transfer`结构体，用于处理网络连接和数据传输。`Transfer`结构体包含一个网络连接`Conn`和一个缓冲区`Buf`。`readPkg`函数用于读取客户端发送的数据包头，解析出数据包的长度。`writePkg`函数则负责发送数据包，首先发送数据长度，然后发送实际数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package utils

import (
“net”
“fmt”
“errors”
“encoding/binary”
“encoding/json”
)
//这里将这些方法关联到结构体中
type Transfer struct {
//分析它应该有哪些字段
Conn net.Conn
Buf[8096]byte//这时传输时，使用缓冲
}

func (this *Transfer)readPkg() (mes message.Message, err error) {
//buf := make([]byte, 8096)
fmt.Println(“读取客户端发送的消息”)
//conn.Read在conn没有被关闭的情况下，才会阻塞
//如果客户端关闭了conn则，就不会阻塞
n, err := this.Conn.Read(this.Buf[:4])
if n != 4 || err != nil {
fmt.Println(“conn.Read err=”, err)
err = errors.New(“read pkg header error”)
return
}
//根据buf[：4]转成一个uint32类型
var pkgLen uint32
pkgLen = binary.BigEndian.Uint32(this.Buf[0:4])

//根据pkglen读取消息内容
n,err = this.Conn.Read(this.Buf[:pkgLen])
if n != int(pkgLen) || err != nil {
	//err = errors.New("read pkg body error")
	//fmt.Println("conn.Read fail err=", err)
	return
}
//把pkgLen反序列化成->message.Message
err = json.Unmarshal(this.Buf[:pkgLen], &mes)
if err != nil {
	fmt.Println("json.Unmarsha err=", err)
	return
}
return

}
func (this *Transfer)writePkg(data[]byte)(err error){
//先发送一个长度给对方
var pkgLen uint32
pkgLen = uint32(len(data))
var buf[4]byte
binary.BigEndian.PutUint32(this.Buf[0:4],pkgLen)
//发送长度
n,err:=this.Conn.Write(this.Buf[:4])
if n!=4||err!=nil{
fmt.Println(“conn.Write(bytes)fail”,err)
return
}
//发送data本身
n,err=this.Conn.Write(data)
if n !=int(pkgLen)||err!=nil {
fmt.Println(“conn.Write(bytes)fail”, err)
return
}
return

}