Drop Connection On Connection Errors in Sca/Query Commands

aerospike · Jul 16, 2015 · 0f1e4cd · 0f1e4cd · ericdmann · Jul 16, 2015
1 parent c392b5e
commit 0f1e4cd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/query_record_command.go b/query_record_command.go
@@ -15,7 +15,6 @@
 package aerospike
 
 import (
-
 	// . "github.com/aerospike/aerospike-client-go/logger"
 
 	. "github.com/aerospike/aerospike-client-go/types"
@@ -47,7 +46,9 @@ func (cmd *queryRecordCommand) parseRecordResults(ifc command, receiveSize int)
 			if resultCode == KEY_NOT_FOUND_ERROR {
 				// consume the rest of the input buffer from the socket
 				if cmd.dataOffset < receiveSize {
-					cmd.readBytes(receiveSize - cmd.dataOffset)
+					if err := cmd.readBytes(receiveSize - cmd.dataOffset); err != nil {
+						return false, err
+					}
 				}
 				return false, nil
 			}

diff --git a/scan_command.go b/scan_command.go
@@ -68,7 +68,9 @@ func (cmd *scanCommand) parseRecordResults(ifc command, receiveSize int) (bool,
 			if resultCode == KEY_NOT_FOUND_ERROR {
 				// consume the rest of the input buffer from the socket
 				if cmd.dataOffset < receiveSize {
-					cmd.readBytes(receiveSize - cmd.dataOffset)
+					if err := cmd.readBytes(receiveSize - cmd.dataOffset); err != nil {
+						return false, err
+					}
 				}
 				return false, nil
 			}