diff options
Diffstat (limited to 'scalding/src/main/scala/sandcrawler/HBaseCountJob.scala')
-rw-r--r-- | scalding/src/main/scala/sandcrawler/HBaseCountJob.scala | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/scalding/src/main/scala/sandcrawler/HBaseCountJob.scala b/scalding/src/main/scala/sandcrawler/HBaseCountJob.scala new file mode 100644 index 0000000..250f175 --- /dev/null +++ b/scalding/src/main/scala/sandcrawler/HBaseCountJob.scala @@ -0,0 +1,30 @@ +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 HBaseCountJob(args: Args, colSpec: String) extends JobBase(args) with HBasePipeConversions { + val output = args("output") + // TODO: Add error checking. + val Col: String = colSpec.split(":")(1) + + HBaseCountJob.getHBaseSource(colSpec) + .read + .fromBytesWritable(Symbol(Col)) + .debug + .groupBy(Col){group => group.size('count)} + .write(Tsv(output)) +} + +object HBaseCountJob { + def getHBaseSource(colSpec: String) = 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(colSpec), + SourceMode.SCAN_ALL) +} |