shkong/MiraiGo
1
0
Fork 0
mirror of https://github.com/Mrs4s/MiraiGo.git synced 2025-05-04 11:07:40 +08:00
Code Issues Projects Releases Wiki Activity

fix socket block (#236)

Browse Source
This commit is contained in:
Lin 2022-01-06 20:52:09 +08:00 committed by fumiama
parent 7bec167fa7
commit 4f3862fd87
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
client/internal/highway/highway.go
View File

@ -8,6 +8,7 @@ import (
"net"
"net/http"
"sync/atomic"
"time"
"github.com/pkg/errors"
@ -48,7 +49,7 @@ type Input struct {
func (s *Session) Upload(addr Addr, input Input) error {
fh, length := utils.ComputeMd5AndLength(input.Body)
_, _ = input.Body.Seek(0, io.SeekStart)
conn, err := net.DialTCP("tcp", nil, addr.asTcpAddr())
conn, err := net.DialTimeout("tcp", addr.String(), time.Second*3)
if err != nil {
return errors.Wrap(err, "connect error")
}