Skip to content

Commit c0431fa

Browse files
committed
feat: 最大读取200m文件
1 parent 1541fa1 commit c0431fa

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

module/worker/connect.go

+9-9
Original file line numberDiff line numberDiff line change
@@ -25,31 +25,31 @@ func Connect() error {
2525

2626
url := args.Worker.Remote
2727
hostname, _ := os.Hostname() // 获取主机名
28-
pod, err := socket.NewWsClient(url, "", "tdp://"+hostname)
29-
28+
conn, err := socket.NewWsClient(url, "", "tdp://"+hostname)
3029
if err != nil {
3130
return err
3231
}
3332

34-
defer pod.Close()
33+
conn.MaxPayloadBytes = 200 << 20 // 200M
34+
defer conn.Close()
3535

3636
// 注册节点
37-
send := &SendPod{pod}
37+
send := &SendPod{conn}
3838
go send.Register()
3939

4040
// 接收数据
41-
return Receiver(pod)
41+
return Receiver(conn)
4242

4343
}
4444

45-
func Receiver(pod *socket.WsConn) error {
45+
func Receiver(conn *socket.WsConn) error {
4646

47-
recv := &RecvPod{pod}
48-
resp := &RespPod{pod}
47+
recv := &RecvPod{conn}
48+
resp := &RespPod{conn}
4949

5050
for {
5151
var rq *socket.PlainData
52-
if err := pod.ReadJson(&rq); err != nil {
52+
if err := conn.ReadJson(&rq); err != nil {
5353
logman.Error("read json failed", "error", err)
5454
return err
5555
}

0 commit comments

Comments
 (0)