From f64b185e6d4d991fa4e0811abafd66153ff774f6 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:25:25 +0800 Subject: [PATCH] Update mongodb.go --- Plugins/mongodb.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/Plugins/mongodb.go b/Plugins/mongodb.go index a691457..8484036 100644 --- a/Plugins/mongodb.go +++ b/Plugins/mongodb.go @@ -34,6 +34,9 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) { } 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 @@ -45,6 +48,9 @@ func MongodbUnauth(info *common.HostInfo) (flag bool, err error) { 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