lib/druid/client.rb in jruby-druid-1.0.0.pre.rc2 vs lib/druid/client.rb in jruby-druid-1.0.0.pre.rc3

@@ -1,10 +1,11 @@
 module Druid
   class Client
-    include Druid::Query::Core
-    include Druid::Query::Datasource
-    include Druid::Query::Task
+    include Druid::Logging
+    include Druid::Queries::Core
+    include Druid::Queries::Datasource
+    include Druid::Queries::Task
 
     attr_reader :broker,
                 :config,
                 :coordinator,
                 :overlord,
@@ -12,10 +13,17 @@
 
     def initialize(options = {})
       @config = Druid::Configuration.new(options)
       @broker = Druid::Node::Broker.new(config)
       @coordinator = Druid::Node::Coordinator.new(config)
+      setup_logger
       @overlord = Druid::Node::Overlord.new(config)
       @writer = Druid::Writer::Base.new(config)
+    end
+
+    private
+
+    def setup_logger
+      logger.set_level(config.log_level)
     end
   end
 end