lib/mongo/operation/parallel_scan/op_msg.rb in mongo-2.5.3 vs lib/mongo/operation/parallel_scan/op_msg.rb in mongo-2.6.0

@@ -39,9 +39,12 @@
         # @since 2.5.2
         def execute(server)
           result = Result.new(dispatch_message(server))
           process_result(result, server)
           result.validate!
+        rescue Mongo::Error::SocketError => e
+          e.send(:add_label, Mongo::Error::TRANSIENT_TRANSACTION_ERROR_LABEL) if session.in_transaction?
+          raise e
         end
 
         private
 
         def selector(server)