This commit is contained in:
lanyi 2021-07-20 10:20:41 +08:00
parent 1499c7253a
commit 71024954e2
2 changed files with 3 additions and 0 deletions

View File

@ -33,6 +33,7 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) {
return flag, err
}
buf := make([]byte, 1024)
err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second))
count, err := conn.Read(buf)
if err != nil {
return flag, err
@ -43,6 +44,7 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) {
if err != nil {
return flag, err
}
err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second))
count, err := conn.Read(buf)
if err != nil {
return flag, err

View File

@ -262,6 +262,7 @@ func Readfile(filename string) (string, error) {
func readreply(conn net.Conn) (result string, err error) {
buf := make([]byte, 4096)
for {
conn.SetReadDeadline(time.Now().Add(10 * time.Second))
count, err := conn.Read(buf)
if err != nil {
break