Add flag to exit with only a single batch of search results

lib/netsuite-rest-client.rb

@@ -100,7 +100,7 @@ def search_records(record_type, search_filters, return_columns, options={})
         results_segment, payload['start_id'] = *parse_json_result_from_rest(:post, params, :payload=>payload)
         results += results_segment unless results_segment.empty?
         puts "Fetched #{results.count} records so far, querying from #{payload['start_id']}..." if options[:verbose]
-      end while (results_segment.length == payload['batch_size'].to_i)
+      end while (results_segment.length == payload['batch_size'].to_i) && !options[:exit_after_first_batch]
 
       results
     end
@@ -177,7 +177,7 @@ def get_saved_search(record_type, search_id, options={})
         results_segment, params['start_id'] = *parse_json_result_from_rest(:get, params)
         results_segment.class == Array ? results += results_segment : raise("Search error: #{results_segment}")
         puts "Fetched #{results.count} records so far, querying from #{params['start_id']}..." if options[:verbose]
-      end while (results_segment.length == params['batch_size'].to_i)
+      end while (results_segment.length == params['batch_size'].to_i) && !options[:exit_after_first_batch]
 
       results
     end