/
utils.go
70 lines (62 loc) · 1.64 KB
/
utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package utils
import (
"fmt"
"net"
"go_code/chatroom/common/message"
"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 则,就不会阻塞
_, err = this.Conn.Read(this.Buf[:4])
if err != nil {
//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")
return
}
//把pkgLen 反序列化成 -> message.Message
// 技术就是一层窗户纸 &mes!!
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
}