diff --git a/Plugins/mongodb.go b/Plugins/mongodb.go index 8ca84e2..03aa443 100644 --- a/Plugins/mongodb.go +++ b/Plugins/mongodb.go @@ -28,6 +28,10 @@ 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 diff --git a/Plugins/redis.go b/Plugins/redis.go index 6bbeecd..c9752ff 100644 --- a/Plugins/redis.go +++ b/Plugins/redis.go @@ -44,6 +44,10 @@ func RedisConn(info *common.HostInfo, pass string) (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([]byte(fmt.Sprintf("auth %s\r\n", pass))) if err != nil { return flag, err @@ -69,6 +73,10 @@ func RedisUnauth(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([]byte("info\r\n")) if err != nil { return flag, err