feature(hash_counter): method date when called with a block, now retu…

…rns the total number of lines

lib/redis_counters/hash_counter.rb

@@ -35,14 +35,19 @@ def partitions(parts = {})
     # Если блок, не передн, то аккумулирует данные,
     # из всех запрошенных партиций, и затем возвращает их.
     #
-    # Returns Array Of Hash.
+    # Returns Array Of Hash или общее кол-во строк данных, если передан блок.
     #
     def data(parts = {})
+      total_rows = 0
       parts = partitions(parts)
-      prepared_parts(parts).flat_map do |partition|
+
+      result = prepared_parts(parts).flat_map do |partition|
         rows = partition_data(partition)
+        total_rows += rows.size
         block_given? ? yield(rows) : rows
       end
+
+      block_given? ? total_rows : result
     end
 
     # Public: Транзакционно удаляет все данные счетчика.

spec/redis_counters/hash_counter_spec.rb

@@ -360,7 +360,9 @@
           ]
         end
 
+        it { expect(counter.data(partitions) {}).to eq 3 }
         it { expect { |b| counter.data(partitions, &b) }.to yield_control.twice }
+
         it do
           expect { |b| counter.data(partitions, &b) }.to(
             yield_successive_args(