Added job and test for counting mime types.

1 files changed, 28 insertions, 0 deletions

diff --git a/scalding/src/main/scala/sandcrawler/HBaseMimeCountJob.scala b/scalding/src/main/scala/sandcrawler/HBaseMimeCountJob.scala
new file mode 100644
index 0000000..819a652
--- /dev/null
+++ b/scalding/src/main/scala/sandcrawler/HBaseMimeCountJob.scala
@@ -0,0 +1,28 @@
+package sandcrawler
+
+import cascading.property.AppProps
+import cascading.tuple.Fields
+import com.twitter.scalding._
+import java.util.Properties
+import parallelai.spyglass.base.JobBase
+import parallelai.spyglass.hbase.{HBaseSource, HBasePipeConversions}
+import parallelai.spyglass.hbase.HBaseConstants.SourceMode
+
+class HBaseMimeCountJob(args: Args) extends JobBase(args) with HBasePipeConversions {
+  val output = args("output")
+
+  HBaseMimeCountJob.getHBaseSource
+    .read
+    .fromBytesWritable(List('mime))
+    .debug
+    .groupBy('mime){group => group.size('count)}
+    .write(Tsv(output))
+}
+
+object HBaseMimeCountJob {
+  def getHBaseSource = HBaseBuilder.build(
+    "wbgrp-journal-extract-0-qa",     // HBase Table Name
+    "mtrcs-zk1.us.archive.org:2181",  // HBase Zookeeper server (to get runtime config info; can be array?)
+    List("file:mime"),
+    SourceMode.SCAN_ALL)
+}