From 6a4bbe378144e4d7291b21969ac0b022ab76d27b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BD=B1=E8=88=9E=E8=80=85?= Date: Tue, 20 Jul 2021 11:30:49 +0800 Subject: [PATCH] Update mongodb.go --- Plugins/mongodb.go | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/Plugins/mongodb.go b/Plugins/mongodb.go index 8484036..03aa443 100644 --- a/Plugins/mongodb.go +++ b/Plugins/mongodb.go @@ -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