druid_config 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 328c50d9e2224494241f1f74880675c1c85c08ca
-  data.tar.gz: acd65bf238e1ea8485935bb5dd7b4014d4e0031b
+  metadata.gz: a03eb8b4f7df4223fe098516a23443beac52be00
+  data.tar.gz: 230450c3d105e2327c681f1bc1beb225926361be
 SHA512:
-  metadata.gz: 0c967abee0d044b9bb9e74775c48dd0e677c0594eeb1ba37c48319ac5df381c97fcbd6eba3ed2c0b66f71cdf55e9c28b4820b6a9afad7e537b82068328ff17b1
-  data.tar.gz: 3f5320ef0bb5e1adf71f1c51e26ac97e0802aac7a83cefd732a2a3de19bd8a7bcd4fb09b4ba23ba46cc43cc08e0d4bc36b43b5d16734b3ef5d58e83024926d35
+  metadata.gz: 151efb283126cbf585549a0e1d0a7f766f48c51e3f21e18ab373c34d59a247c64fbcc1188157ccaadb7b5940ccdcc480c24b87f33a2f0e890e019be817b9bcab
+  data.tar.gz: db35a7cc044f47bb4de83ea1467e35186389471485f71f48e3f4363087dc8f9326150a0df93a7844d901b804bf38348d51ab2cfb452a2aa708240cbd08a5ba76

data/README.md

@@ -1,3 +1,5 @@
+[![Gem Version](https://badge.fury.io/rb/druid_config.svg)](https://badge.fury.io/rb/druid_config) [![Build Status](https://travis-ci.org/redBorder/druid_config.svg)](https://travis-ci.org/redBorder/druid_config)
+
 # DruidConfig
 
 DruidConfig is a gem to access the information about Druid cluster status. You can check a node capacity, number of segments, tiers... It uses [zookeeper](https://zookeeper.apache.org/) to get coordinator and overlord URIs.
@@ -5,9 +7,13 @@ DruidConfig is a gem to access the information about Druid cluster status. You c
 To use in your application, add this line to your Gemfile:
 
 ```ruby
-require 'druid_config'
+gem 'druid_config'
 ```
 
+## Query Druid data
+
+The purpose of this gem is to query Druid cluster status. If you want to query Druid data, I recommend you to use [ruby-druid gem](https://github.com/ruby-druid/ruby-druid).
+
 # Initialization
 
 `Cluster` is the base class to perform queries. To initialize it send the Zookeeper URI and options as arguments:
@@ -44,11 +50,11 @@ Call methods defined in `DruidConfig::Cluster` to access to the data. To get mor
 
 Some methods return an instance of an `Entity` class. These entities provide multiple methods to access data. Defined entities are inside `druid_config/entities` folder.
 
-* [DataSource]()
-* [Node]()
-* [Segment]()
-* [Tier]()
-* [Worker]()
+* [DataSource](https://github.com/redBorder/druid_config/blob/master/lib/druid_config/entities/data_source.rb)
+* [Segment](https://github.com/redBorder/druid_config/blob/master/lib/druid_config/entities/segment.rb)
+* [Tier](https://github.com/redBorder/druid_config/blob/master/lib/druid_config/entities/tier.rb)
+* [Node](https://github.com/redBorder/druid_config/blob/master/lib/druid_config/entities/node.rb)
+* [Worker](https://github.com/redBorder/druid_config/blob/master/lib/druid_config/entities/worker.rb)
 
 ## Exceptions
 

data/lib/druid_config.rb

@@ -6,6 +6,7 @@ require 'druid_config/zk'
 require 'druid_config/version'
 require 'druid_config/util'
 require 'druid_config/entities/segment'
+require 'druid_config/entities/task'
 require 'druid_config/entities/worker'
 require 'druid_config/entities/node'
 require 'druid_config/entities/tier'

data/lib/druid_config/cluster.rb

@@ -136,7 +136,7 @@ module DruidConfig
     def datasources
       datasource_status = load_status
       secure_query do
-        self.class.get('/datasources?full').map do |data|
+        self.class.get('/datasources?simple').map do |data|
           DruidConfig::Entities::DataSource.new(
             data,
             datasource_status.select { |k, _| k == data['name'] }.values.first)
@@ -202,8 +202,8 @@ module DruidConfig
     #
     def servers
       secure_query do
-        queue = load_queue('full')
-        self.class.get('/servers?full').map do |data|
+        queue = load_queue('simple')
+        self.class.get('/servers?simple').map do |data|
           DruidConfig::Entities::Node.new(
             data,
             queue.select { |k, _| k == data['host'] }.values.first)
@@ -253,22 +253,14 @@ module DruidConfig
     # Array of Workers
     #
     def workers
-      # Stash the base_uri
-      stash_uri
-      self.class.base_uri(
-        "#{DruidConfig.client.overlord}"\
-        "druid/indexer/#{DruidConfig::Version::API_VERSION}")
       workers = []
       # Perform a query
-      begin
+      query_overlord do
         secure_query do
           workers = self.class.get('/workers').map do |worker|
             DruidConfig::Entities::Worker.new(worker)
           end
         end
-      ensure
-        # Recover it
-        pop_uri
       end
       # Return
       workers
@@ -281,6 +273,68 @@ module DruidConfig
       @physical_workers ||= workers.map(&:host).uniq
     end
 
+    # Tasks
+    # -----------------
+    
+    #
+    # Return tasks based on the status
+    #
+    # == Returns:
+    # Array of Tasks
+    #
+    %w(running pending waiting).each do |status|
+      define_method("#{status}_tasks") do
+        tasks = []
+        query_overlord do
+          tasks = self.class.get("/#{status}Tasks").map do |task|
+            DruidConfig::Entities::Task.new(
+              task['id'],
+              DruidConfig::Entities::Task::STATUS[status.to_sym],
+              created_time: task['createdTime'],
+              query_insertion_time: task['queueInsertionTime'])
+          end
+        end
+      end
+    end
+
+    #
+    # Find a task
+    #
+    def task(id)
+      DruidConfig::Entities::Task.find(id)
+    end
+    
+    #
+    # Return complete tasks
+    #
+    # == Returns:
+    # Array of Tasks
+    #
+    def complete_tasks
+      tasks = []
+      query_overlord do
+        tasks = self.class.get('/completeTasks').map do |task|
+          DruidConfig::Entities::Task.new(
+            task['id'],
+            task['statusCode'])
+        end
+      end
+    end
+
+    #
+    # Return failed completed tasks
+    #
+    def failed_tasks
+      complete_tasks.select(&:failed?)
+    end
+
+    #
+    # Return success completed tasks
+    #
+    def success_tasks
+      complete_tasks.select(&:success?)
+    end
+
     # Services
     # -----------------
     
@@ -302,23 +356,5 @@ module DruidConfig
       # Return nodes
       @services = services
     end
-
-    private
-
-    #
-    # Stash current base_uri
-    #
-    def stash_uri
-      @uri_stack ||= []
-      @uri_stack.push self.class.base_uri
-    end
-
-    #
-    # Pop next base_uri
-    #
-    def pop_uri
-      return if @uri_stack.nil? || @uri_stack.empty?
-      self.class.base_uri(@uri_stack.pop)
-    end
   end
 end

data/lib/druid_config/entities/data_source.rb

@@ -1,6 +1,6 @@
 module DruidConfig
   #
-  # Module of info
+  # Module of Entities
   #
   module Entities
     #
@@ -8,6 +8,7 @@ module DruidConfig
     class DataSource
       # HTTParty Rocks!
       include HTTParty
+      include DruidConfig::Util
 
       attr_reader :name, :properties, :load_status
 
@@ -31,28 +32,36 @@ module DruidConfig
       # http://druid.io/docs/0.8.1/design/coordinator.html
       #
 
-      def info(params = '')
-        @info ||= self.class.get("/datasources/#{@name}?#{params}")
+      def info
+        secure_query do
+          @info ||= self.class.get("/datasources/#{@name}")
+        end
       end
 
       # Intervals
       # -----------------
       def intervals(params = '')
-        self.class.get("/datasources/#{@name}/intervals?#{params}")
+        secure_query do
+          self.class.get("/datasources/#{@name}/intervals?#{params}")
+        end
       end
 
       def interval(interval, params = '')
-        self.class.get("/datasources/#{@name}/intervals/#{interval}"\
-                       "?#{params}")
+        secure_query do
+          self.class.get("/datasources/#{@name}/intervals/#{interval}"\
+                         "?#{params}")
+        end
       end
 
       # Segments and Tiers
       # -----------------
       def segments
-        @segments ||=
-          self.class.get("/datasources/#{@name}/segments?full").map do |s|
-            DruidConfig::Entities::Segment.new(s)
-          end
+        secure_query do
+          @segments ||=
+            self.class.get("/datasources/#{@name}/segments?full").map do |s|
+              DruidConfig::Entities::Segment.new(s)
+            end
+        end
       end
 
       def segment(segment)
@@ -66,12 +75,16 @@ module DruidConfig
       # Rules
       # -----------------
       def rules(params = '')
-        self.class.get("/rules/#{@name}?#{params}")
+        secure_query do
+          self.class.get("/rules/#{@name}?#{params}")
+        end
       end
 
       def history_rules(interval)
-        self.class.get("/rules/#{@name}/history"\
-                       "?interval=#{interval}")
+        secure_query do
+          self.class.get("/rules/#{@name}/history"\
+                         "?interval=#{interval}")
+        end
       end
     end
   end

data/lib/druid_config/entities/node.rb

@@ -6,10 +6,11 @@ module DruidConfig
     class Node
       # HTTParty Rocks!
       include HTTParty
+      include DruidConfig::Util
 
       # Readers
       attr_reader :host, :port, :max_size, :type, :tier, :priority, :size,
-                  :segments, :segments_to_load, :segments_to_drop,
+                  :segments_to_load_count, :segments_to_drop_count,
                   :segments_to_load_size, :segments_to_drop_size
 
       #
@@ -28,21 +29,20 @@ module DruidConfig
         @tier = metadata['tier']
         @priority = metadata['priority']
         @size = metadata['currSize']
-        @segments = metadata['segments'].map do |_, sdata|
-          DruidConfig::Entities::Segment.new(sdata)
-        end
+        # Set end point for HTTParty
+        self.class.base_uri(
+          "#{DruidConfig.client.coordinator}"\
+          "druid/coordinator/#{DruidConfig::Version::API_VERSION}")
+
+        # Load more data from queue
         if queue.nil?
-          @segments_to_load, @segments_to_drop = [], []
+          @segments_to_load_count, @segments_to_drop_count = 0, 0
           @segments_to_load_size, @segments_to_drop_size = 0, 0
         else
-          @segments_to_load = queue['segmentsToLoad'].map do |segment|
-            DruidConfig::Entities::Segment.new(segment)
-          end
-          @segments_to_drop = queue['segmentsToDrop'].map do |segment|
-            DruidConfig::Entities::Segment.new(segment)
-          end
-          @segments_to_load_size = @segments_to_load.map(&:size).reduce(:+)
-          @segments_to_drop_size = @segments_to_drop.map(&:size).reduce(:+)
+          @segments_to_load_count = queue['segmentsToLoad']
+          @segments_to_drop_count = queue['segmentsToDrop']
+          @segments_to_load_size = queue['segmentsToLoadSize']
+          @segments_to_drop_size = queue['segmentsToLoadSize']
         end
       end
 
@@ -63,12 +63,64 @@ module DruidConfig
         max_size - size
       end
 
+      #
+      # Return all segments of this node
+      #
+      def segments_count
+        @segments_count ||=
+          self.class.get("/servers/#{uri}/segments").size
+      end
+
+      #
+      # Return all segments of this node
+      #
+      def segments
+        @segments ||=
+          self.class.get("/servers/#{uri}/segments?full").map do |s|
+            DruidConfig::Entities::Segment.new(s)
+          end
+      end
+
+      #
+      # Get segments to load
+      #
+      def segments_to_load
+        current_queue = queue
+        return [] unless current_queue
+        current_queue['segmentsToLoad'].map do |segment|
+          DruidConfig::Entities::Segment.new(segment)
+        end
+      end
+
+      #
+      # Get segments to drop
+      #
+      def segments_to_drop
+        current_queue = queue
+        return [] unless current_queue
+        current_queue['segmentsToDrop'].map do |segment|
+          DruidConfig::Entities::Segment.new(segment)
+        end
+      end
+
       #
       # Return the URI of this node
       #
       def uri
         "#{@host}:#{@port}"
       end
+
+      private
+
+      #
+      # Get load queue for this node
+      #
+      def queue
+        secure_query do
+          self.class.get('/loadqueue?full')
+            .select { |k, _| k == uri }.values.first
+        end
+      end
     end
   end
 end

data/lib/druid_config/entities/task.rb

@@ -0,0 +1,97 @@
+module DruidConfig
+  #
+  # Module of Entities
+  #
+  module Entities
+    #
+    # Init a DataSource
+    class Task
+      # HTTParty Rocks!
+      include HTTParty
+      include DruidConfig::Util
+
+      # Statuses constants
+      STATUS = {
+        running: 'RUNNING',
+        pending: 'PENDING',
+        success: 'SUCCESS',
+        waiting: 'WAITING',
+        failed: 'FAILED'
+      }
+
+      attr_reader :id, :status, :created_time, :query_insertion_time
+
+      #
+      # Find a task based on id
+      #
+      def self.find(id)
+        # Set end point for HTTParty
+        base_uri(
+          "#{DruidConfig.client.overlord}"\
+          "druid/indexer/#{DruidConfig::Version::API_VERSION}/task")
+        # Get data
+        status = get("/#{id}/status")
+        new(id, status['status']['status'])
+      end
+
+      #
+      # Initialize a task
+      #
+      # == Parameters:
+      # id::
+      #   String with identifier
+      # status::
+      #   Current status of the task
+      #
+      def initialize(id, status, extended_info = {})
+        @id = id
+        @status = status.downcase.to_sym
+        @created_time = extended_info[:created_time]
+        @query_insertion_time = extended_info[:query_insertion_time]
+        # Set end point for HTTParty
+        self.class.base_uri(
+          "#{DruidConfig.client.overlord}"\
+          "druid/indexer/#{DruidConfig::Version::API_VERSION}/task")
+      end
+
+      #
+      # Multiple methods to check status of the tasks
+      #
+      STATUS.keys.each do |s|
+        define_method("#{s}?") do
+          s.to_sym == @status
+        end
+      end
+
+      alias_method :completed?, :success?
+
+      #
+      # Get payload of the task
+      #
+      def payload
+        @payload ||= self.class.get("/#{@id}")['payload']
+      end
+
+      #
+      # Current log of a task
+      #
+      def log
+        self.class.get("/#{@id}/log")
+      end
+
+      #
+      # Get the dataSource of this task
+      #
+      def datasource
+        payload['dataSource']
+      end
+
+      #
+      # Group of the task
+      #
+      def group_id
+        payload['groupId']
+      end
+    end
+  end
+end

data/lib/druid_config/entities/tier.rb

@@ -42,14 +42,28 @@ module DruidConfig
                       .flatten.sort_by { |seg| seg.interval.first }
       end
 
+      def segments_count
+        @segments_count ||= nodes.map(&:segments_count).inject(:+)
+      end
+
       def segments_to_load
-        @segments_to_load ||=
-          nodes.map { |node| node.segments_to_load.count }.inject(:+)
+        @segments_to_load ||= nodes.map(&:segments_to_load)
+                              .flatten.sort_by { |seg| seg.interval.first }
       end
 
       def segments_to_drop
-        @segments_to_drop ||=
-          nodes.map { |node| node.segments_to_drop.count }.inject(:+)
+        @segments_to_drop ||= nodes.map(&:segments_to_drop)
+                              .flatten.sort_by { |seg| seg.interval.first }
+      end
+
+      def segments_to_load_count
+        @segments_to_load_count ||=
+          nodes.map(&:segments_to_load_count).inject(:+)
+      end
+
+      def segments_to_drop_count
+        @segments_to_drop_count ||=
+          nodes.map(&:segments_to_drop_count).inject(:+)
       end
 
       def segments_to_load_size

data/lib/druid_config/entities/worker.rb

@@ -6,7 +6,7 @@ module DruidConfig
     class Worker
       # Readers
       attr_reader :last_completed_task_time, :host, :port, :ip, :capacity,
-                   :version, :running_tasks, :current_capacity_used
+                  :version, :running_tasks, :capacity_used
 
       #
       # Initialize it with received info
@@ -21,21 +21,29 @@ module DruidConfig
         @capacity = metadata['worker']['capacity']
         @version = metadata['worker']['version']
         @last_completed_task_time = metadata['lastCompletedTaskTime']
-        @running_tasks = metadata['runningTasks']
+        @running_tasks = metadata['runningTasks'].map do |task|
+          DruidConfig::Entities::Task.new(
+            task,
+            DruidConfig::Entities::Task::STATUS[:running],
+            created_time: task['createdTime'],
+            query_insertion_time: task['queueInsertionTime'])
+        end
         @capacity_used = metadata['currCapacityUsed']
       end
 
+      alias_method :used, :capacity_used
+
       #
       # Return free capacity
       #
       def free
-        @free ||= (capacity - current_capacity_used)
+        @free ||= (capacity - capacity_used)
       end
 
       #
       # Return capacity used
       #
-      def used
+      def used_percent
         return 0 unless @capacity && @capacity != 0
         ((@capacity_used.to_f / @capacity) * 100).round(2)
       end

data/lib/druid_config/util.rb

@@ -31,5 +31,39 @@ module DruidConfig
         retry
       end
     end
+
+    #
+    # Update the URI of HTTParty to perform queries to Overlord.
+    # After perform the query, the URI is reverted to coordinator.
+    #
+    def query_overlord
+      return unless block_given?
+      stash_uri
+      self.class.base_uri(
+        "#{DruidConfig.client.overlord}"\
+        "druid/indexer/#{DruidConfig::Version::API_VERSION}")
+      begin
+        yield
+      ensure
+        # Ensure we revert the URI
+        pop_uri
+      end
+    end
+
+    #
+    # Stash current base_uri
+    #
+    def stash_uri
+      @uri_stack ||= []
+      @uri_stack.push self.class.base_uri
+    end
+
+    #
+    # Pop next base_uri
+    #
+    def pop_uri
+      return if @uri_stack.nil? || @uri_stack.empty?
+      self.class.base_uri(@uri_stack.pop)
+    end
   end
 end

data/lib/druid_config/version.rb

@@ -4,7 +4,7 @@
 module DruidConfig
   module Version
     # Version of the gem
-    VERSION = '0.1.0'
+    VERSION = '0.2.0'
 
     # Base URI foor coordinator queries
     API_VERSION = 'v1'

data/lib/druid_config/zk.rb

@@ -1,3 +1,9 @@
+#
+# This Class is an adaptation from ZK Class of ruby-druid gem.
+# Please, check the following link:
+#   https://github.com/ruby-druid/ruby-druid/blob/master/lib/druid/zk.rb
+#
+
 require 'zk'
 require 'rest_client'
 

data/spec/cluster_spec.rb

@@ -2,31 +2,27 @@ require 'spec_helper'
 require 'pry'
 require 'pry-nav'
 
-# describe DruidConfig::Cluster do
-#   before(:each) do
-#     @cluster = DruidConfig::Cluster.new('localhost', zk_keepalive: true)
-#   end
+describe DruidConfig::Cluster do
+  before(:each) do
+    @cluster = DruidConfig::Cluster.new('localhost', zk_keepalive: true)
+  end
 
-#   it 'must get the leader' do
-#     expect(@cluster.leader).to eq 'coordinator.stub'
-#   end
+  it 'must get the leader' do
+    expect(@cluster.leader).to eq 'coordinator.stub'
+  end
 
-#   it 'must get load datasources of a cluster' do
-#     datasources = @cluster.datasources
+  it 'must get load status' do
+    basic = @cluster.load_status
+    expect(basic.keys).to eq %w(datasource1 datasource2)
+    expect(basic[basic.keys.first]).to eq 100
 
+    simple = @cluster.load_status('simple')
+    expect(simple.keys).to eq %w(datasource1 datasource2)
+    expect(simple[simple.keys.first]).to eq 0
 
-#     # basic = @cluster.load_status
-#     # expect(basic.keys).to eq %w(datasource1 datasource2)
-#     # expect(basic[basic.keys.first]).to eq 100
-
-#     # simple = @cluster.load_status('simple')
-#     # expect(simple.keys).to eq %w(datasource1 datasource2)
-#     # expect(simple[simple.keys.first]).to eq 0
-
-#     # # Use tiers
-#     # simple = @cluster.load_status('full')
-#     # expect(simple.keys).to eq %w(_default_tier hot)
-#     # expect(simple['_default_tier']['datasource1']).to eq 0
-#     # expect(simple['hot']['datasource2']).to eq 0
-#   end
-# end
+    full = @cluster.load_status('full')
+    expect(full.keys).to eq %w(_default_tier hot)
+    expect(full['_default_tier']['datasource1']).to eq 0
+    expect(full['hot']['datasource2']).to eq 0
+  end
+end

data/spec/data/druid_responses.yml

@@ -0,0 +1,4 @@
+leader: coordinator.stub
+loadstatus: '{"datasource1":100.0,"datasource2":100.0}'
+loadstatus_simple: '{"datasource1":0,"datasource2":0}'
+loadstatus_full: '{"_default_tier":{"datasource1":0}, "hot":{"datasource2":0}}'

data/spec/node_spec.rb

@@ -28,8 +28,10 @@ describe DruidConfig::Entities::Node do
                   'priority' => 0, 'segments' => @segments, 'currSize' => @size }
     
     @queue = {
-      'segmentsToLoad' => [],
-      'segmentsToDrop' => []
+      'segmentsToLoad' => 0,
+      'segmentsToDrop' => 0,
+      'segmentsToLoadSize' => 0,
+      'segmentsToDropSize' => 0
     }
   end
 
@@ -41,8 +43,8 @@ describe DruidConfig::Entities::Node do
     expect(datasource.type).to eq @type.to_sym
     expect(datasource.priority).to eq @priority
     expect(datasource.size).to eq @size
-    expect(datasource.segments_to_load).to eq []
-    expect(datasource.segments_to_drop).to eq []
+    expect(datasource.segments_to_load_count).to eq 0
+    expect(datasource.segments_to_drop_count).to eq 0
   end
 
   it 'calculate free space' do

data/spec/spec_helper.rb

@@ -46,8 +46,11 @@ RSpec.configure do |config|
         def coordinator
           'coordinator.stub/'
         end
+        def overlord
+          'overlord.stub/'
+        end
       end
-      allow(DruidConfig).to receive(:client) { ClientStub.new }
+      allow(DruidConfig::Client).to receive(:new) { ClientStub.new }
 
       # Stub queries
       # ----------------------------------
@@ -56,24 +59,27 @@ RSpec.configure do |config|
       # leader: coordinator.stub
       # datasources: datasource1, datasource2
       # tiers: _default_tier, hot
-      # stub_request(:get, 'http://coordinator.stub/druid/coordinator/v1/leader')
-      #   .with(headers: { 'Accept' => '*/*', 'User-Agent' => 'Ruby' })
-      #   .to_return(status: 200, body: 'coordinator.stub', headers: {})
+      
+      # Load data
+      responses = YAML.load(File.read('spec/data/druid_responses.yml'))
+      head = { 'Accept' => '*/*', 'User-Agent' => 'Ruby' }
+      rhead = { 'Content-Type' => 'application/json' }
+      base = 'http://coordinator.stub/druid/coordinator/v1/'
+
+      # Stub all queries to the API
+      stub_request(:get, "#{base}leader").with(headers: head)
+        .to_return(status: 200, body: responses['leader'], headers: {})
 
-      # stub_request(:get, 'http://coordinator.stub/druid/coordinator/v1/loadstatus')
-      #   .with(headers: { 'Accept' => '*/*', 'User-Agent' => 'Ruby' })
-      #   .to_return(status: 200, body: '{"datasource1":100.0,"datasource2":100.0}',
-      #              headers: { 'Content-Type' => 'application/json' })
+      stub_request(:get, "#{base}loadstatus").with(headers: head)
+        .to_return(status: 200, body: responses['loadstatus'], headers: rhead)
 
-      # stub_request(:get, 'http://coordinator.stub/druid/coordinator/v1/loadstatus?simple')
-      #   .with(headers: { 'Accept' => '*/*', 'User-Agent' => 'Ruby' })
-      #   .to_return(status: 200, body: '{"datasource1":0,"datasource2":0}',
-      #              headers: { 'Content-Type' => 'application/json' })
+      stub_request(:get, "#{base}loadstatus?simple").with(headers: head)
+        .to_return(status: 200, body: responses['loadstatus_simple'],
+                   headers: rhead)
 
-      # stub_request(:get, 'http://coordinator.stub/druid/coordinator/v1/loadstatus?full')
-      #   .with(headers: { 'Accept' => '*/*', 'User-Agent' => 'Ruby' })
-      #   .to_return(status: 200, body: '{"_default_tier":{"datasource1":0}, "hot":{"datasource2":0}}',
-      #              headers: { 'Content-Type' => 'application/json' })
+      stub_request(:get, "#{base}loadstatus?full").with(headers: head)
+        .to_return(status: 200, body: responses['loadstatus_full'],
+                   headers: rhead)
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: druid_config
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Angel M Miguel
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-10-27 00:00:00.000000000 Z
+date: 2015-10-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: zk
@@ -152,12 +152,14 @@ files:
 - lib/druid_config/entities/node.rb
 - lib/druid_config/entities/rule.rb
 - lib/druid_config/entities/segment.rb
+- lib/druid_config/entities/task.rb
 - lib/druid_config/entities/tier.rb
 - lib/druid_config/entities/worker.rb
 - lib/druid_config/util.rb
 - lib/druid_config/version.rb
 - lib/druid_config/zk.rb
 - spec/cluster_spec.rb
+- spec/data/druid_responses.yml
 - spec/data_source_spec.rb
 - spec/node_spec.rb
 - spec/spec_helper.rb
@@ -187,6 +189,7 @@ specification_version: 4
 summary: Fetch Druid info and configure your cluster
 test_files:
 - spec/cluster_spec.rb
+- spec/data/druid_responses.yml
 - spec/data_source_spec.rb
 - spec/node_spec.rb
 - spec/spec_helper.rb