Update mongodb.go

This commit is contained in:
影舞者 2021-07-20 11:30:49 +08:00 committed by GitHub
parent c322700c6d
commit 6a4bbe3781
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -28,15 +28,15 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) {
return flag, err
}
defer conn.Close()
err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second))
if err != nil {
return flag, err
}
_, err = conn.Write(senddata)
if err != nil {
return flag, err
}
buf := make([]byte, 1024)
err = conn.SetReadDeadline(time.Now().Add(time.Duration(info.Timeout)*time.Second))
if err != nil {
return flag, err
}
count, err := conn.Read(buf)
if err != nil {
return flag, err
@ -47,10 +47,6 @@ 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))
if err != nil {
return flag, err
}
count, err := conn.Read(buf)
if err != nil {
return flag, err