Merge pull request elastic#1434 from shanekenney/develop

Add size parameter to reindex descriptor

Author: gmarz

Diff for: src/Nest/DSL/Reindex/ReindexDescriptor.cs

@@ -5,7 +5,8 @@ public class ReindexDescriptor<T> where T : class
 	{
 		internal string _ToIndexName { get; set; }
 		internal string _FromIndexName { get; set; }
-		internal string _Scroll { get; set; }
+		internal string _Scroll { get; set; }
+		internal int? _Size { get; set; }
 
 		internal Func<QueryDescriptor<T>, QueryContainer> _QuerySelector { get; set; }
 
@@ -43,6 +44,25 @@ public ReindexDescriptor<T> Scroll(string scrollTime)
 			return this;
 		}
 
+		/// <summary>
+		/// The number of hits to return. Defaults to 100. When using scroll search type,
+		/// size is actually multiplied by the number of shards!
+		/// </summary>
+		public ReindexDescriptor<T> Size(int size)
+		{
+			this._Size = size;
+			return this;
+		}
+
+		/// <summary>
+		/// The number of hits to return. Defaults to 100.
+		/// </summary>
+		public ReindexDescriptor<T> Take(int take)
+		{
+			this.Size(take);
+			return this;
+		}
+
 		/// <summary>
 		/// A query to optionally limit the documents to use for the reindex operation.  
 		/// </summary>

Diff for: src/Nest/Domain/Observers/ReindexObservable.cs

@@ -36,7 +36,8 @@ private void Reindex(IObserver<IReindexResponse<T>> observer)
 		{
 			var fromIndex = this._reindexDescriptor._FromIndexName;
 			var toIndex = this._reindexDescriptor._ToIndexName;
-			var scroll = this._reindexDescriptor._Scroll ?? "2m";
+			var scroll = this._reindexDescriptor._Scroll ?? "2m";
+			var size = this._reindexDescriptor._Size ?? 100;
 
 			fromIndex.ThrowIfNullOrEmpty("fromIndex");
 			toIndex.ThrowIfNullOrEmpty("toIndex");
@@ -56,7 +57,7 @@ private void Reindex(IObserver<IReindexResponse<T>> observer)
 					.Index(fromIndex)
 					.AllTypes()
 					.From(0)
-					.Take(100)
+					.Size(size)
 					.Query(this._reindexDescriptor._QuerySelector ?? (q=>q.MatchAll()))
 					.SearchType(SearchType.Scan)
 					.Scroll(scroll)

Diff for: src/Tests/Nest.Tests.Integration/Index/ReindexTests.cs

@@ -47,6 +47,7 @@ public void Reindex()
 				.ToIndex(toIndex)
 				.Query(q=>q.MatchAll())
 				.Scroll("10s")
+				.Size(100)
 				.CreateIndex(c=>c
 					.NumberOfReplicas(0)
 					.NumberOfShards(1)