jruby-druid 1.0.0.pre.rc2

data/lib/druid/writer/tranquilizer/dimensions.rb

@@ -0,0 +1,16 @@
+module Druid
+  module Writer
+    module Tranquilizer
+      class Dimensions
+        class << self
+          java_import com.metamx.tranquility.druid.DruidDimensions
+
+          def build(dimensions)
+            dimensions_list = ImmutableList.of(*dimensions.keys)
+            DruidDimensions.specific(dimensions_list)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/druid/writer/tranquilizer/druid_beam_config.rb

@@ -0,0 +1,40 @@
+# Defaults based on https://github.com/druid-io/tranquility/blob/master/core/src/main/scala/com/metamx/tranquility/druid/DruidBeamConfig.scala
+module Druid
+  module Writer
+    module Tranquilizer
+      module DruidBeamConfig
+        class << self
+          java_import com.metamx.tranquility.druid.DruidBeamConfig
+          java_import com.metamx.tranquility.druid.OverlordLocator
+          java_import com.metamx.tranquility.druid.TaskLocator
+
+          def build(randomize_task_id)
+            firehoseGracePeriod = org.joda.time.Period.new('PT5M')
+            firehoseQuietPeriod = org.joda.time.Period.new('PT1M')
+            firehoseRetryPeriod = org.joda.time.Period.new('PT1M')
+            firehoseChunkSize = 1000
+            randomizeTaskId = randomize_task_id
+            indexRetryPeriod = org.joda.time.Period.new('PT1M')
+            firehoseBufferSize = 100000
+            overlordLocator = OverlordLocator.Curator
+            taskLocator = TaskLocator.Curator
+            overlordPollPeriod = org.joda.time.Period.new('PT20S')
+
+            DruidBeamConfig.new(
+              firehoseGracePeriod,
+              firehoseQuietPeriod,
+              firehoseRetryPeriod,
+              firehoseChunkSize,
+              randomizeTaskId,
+              indexRetryPeriod,
+              firehoseBufferSize,
+              overlordLocator,
+              taskLocator,
+              overlordPollPeriod
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/druid/writer/tranquilizer/druid_beams.rb

@@ -0,0 +1,12 @@
+module Druid
+  module Writer
+    module Tranquilizer
+      java_import com.metamx.tranquility.druid.DruidBeams
+
+      class DruidBeams
+        # Ensures the correct override is used.
+        java_alias :timestamper_builder, :builder, [com.metamx.tranquility.typeclass.Timestamper.java_class]
+      end
+    end
+  end
+end

data/lib/druid/writer/tranquilizer/event_listener.rb

@@ -0,0 +1,17 @@
+module Druid
+  module Writer
+    module Tranquilizer
+      class EventListener
+        include_package com.twitter.util.FutureEventListener
+
+        def onSuccess(data)
+          # puts "success: #{data.to_s}" #TODO: Log this (trace)
+        end
+
+        def onFailure(error)
+          puts "failure: #{error.to_s}" #TODO: Log this (debug)
+        end
+      end
+    end
+  end
+end

data/lib/druid/writer/tranquilizer/rollup.rb

@@ -0,0 +1,23 @@
+module Druid
+  module Writer
+    module Tranquilizer
+      module Rollup
+        class << self
+          java_import com.metamx.tranquility.druid.DruidRollup
+
+          def build(config, datapoint)
+            dimensions = Druid::Writer::Tranquilizer::Dimensions.build(datapoint.dimensions)
+            aggregators = Druid::Writer::Tranquilizer::Aggregators.build(datapoint.metrics)
+            DruidRollup.create(dimensions, aggregators, get_granularity(config.rollup_granularity))
+          end
+
+          private
+
+          def get_granularity(granularity)
+            "Java::IoDruidGranularity::QueryGranularity::#{granularity}".constantize
+          end
+        end
+      end
+    end
+  end
+end

data/lib/druid/writer/tranquilizer/timestamper.rb

@@ -0,0 +1,13 @@
+module Druid
+  module Writer
+    module Tranquilizer
+      class Timestamper
+        include_package com.metamx.tranquility.typeclass.Timestamper
+
+        def timestamp(theMap)
+          org.joda.time.DateTime.new(theMap.get("timestamp"))
+        end
+      end
+    end
+  end
+end

data/lib/druid/writer/tranquilizer/tuning.rb

@@ -0,0 +1,25 @@
+module Druid
+  module Writer
+    module Tranquilizer
+      class Tuning
+        class << self
+          def build(config)
+            ClusteredBeamTuning.
+              builder.
+              segmentGranularity(get_granularity(config.tuning_granularity)).
+              windowPeriod(org.joda.time.Period.new(config.tuning_window)).
+              partitions(1).
+              replicants(1).
+              build
+          end
+
+          private
+
+          def get_granularity(granularity)
+            "Java::ComMetamxCommon::Granularity::#{granularity}".constantize
+          end
+        end
+      end
+    end
+  end
+end

data/lib/jruby-druid.rb

@@ -0,0 +1,35 @@
+require "java"
+Dir["#{File.dirname(__FILE__)}/../vendor/tranquility/*.jar"].each { |file| require file }
+
+require "active_support/all"
+require "json"
+
+require "druid/configuration"
+require "druid/connection"
+require "druid/errors"
+require "druid/version"
+
+require "druid/node/broker"
+require "druid/node/coordinator"
+require "druid/node/overlord"
+
+require "druid/query/core"
+require "druid/query/datasource"
+require "druid/query/task"
+
+require "druid/writer/base"
+
+require "druid/writer/tranquilizer"
+require "druid/writer/tranquilizer/aggregators"
+require "druid/writer/tranquilizer/base"
+require "druid/writer/tranquilizer/curator"
+require "druid/writer/tranquilizer/datapoint"
+require "druid/writer/tranquilizer/dimensions"
+require "druid/writer/tranquilizer/druid_beams"
+require "druid/writer/tranquilizer/druid_beam_config"
+require "druid/writer/tranquilizer/event_listener"
+require "druid/writer/tranquilizer/rollup"
+require "druid/writer/tranquilizer/timestamper"
+require "druid/writer/tranquilizer/tuning"
+
+require "druid/client"