new rule count

1 files changed, 22 insertions, 0 deletions

diff --git a/dtrain/hstreaming/rule_count/red.rb b/dtrain/hstreaming/rule_count/red.rb
new file mode 100644
index 00000000..8f9109cc
--- /dev/null
+++ b/dtrain/hstreaming/rule_count/red.rb
@@ -0,0 +1,22 @@
+STDIN.set_encoding 'utf-8'
+STDOUT.set_encoding 'utf-8'
+
+def output(key, val)
+  puts "#{key}\t#{val}"
+end
+
+prev_key = nil
+sum = 0
+while line = STDIN.gets
+   key, val = line.strip.split /\t/
+   if key != prev_key && sum > 0
+      output prev_key, sum
+      prev_key = key
+      sum = 0
+   elsif !prev_key
+      prev_key = key
+   end
+   sum += val.to_i
+end
+output prev_key, sum
+