kazoo-ruby 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 89757bee4b135ad94b20ef03dd75627928ac6430
-  data.tar.gz: babf18793c68d48949a54fadaa9c26a846cc90df
+  metadata.gz: 6b5646dda6763c39d80505cd273ec369a5710738
+  data.tar.gz: baac20d245bb954a07788bf7672fca21bfe838e4
 SHA512:
-  metadata.gz: 0fe533d27ff5eceb5cde6b60cb45137382c2498cf1ed97d51e5e59ed8755ec033ccc421915cd4cb16bf9729652d2d2c39675af31eeb1a9a528d468f8c8489625
-  data.tar.gz: d7cf53f56a7c2765d3f85b890e3acab71d5969970fc26ada8247404101d350f51f64b89a16c1987ad3303a2390909f94b8bc38bbbd4a3c56fb0e7e857becc5f7
+  metadata.gz: 9df02b3597772f201d64ac7e0c5108856454297ad3994e00474bd70d1b59e22fc7541f45650487cb06c2b108e5fafab09de4ae0a13d1c482f8fab59d9d6a2eac
+  data.tar.gz: 0892bc22d1cd9b8d2df3353a4b0a108578c70cc5ec33c65bf59d87f3ded03b98c4f4308d50a840e8d301a4ef3675f323be23152e1548c6b019b2707b392ccfaf

data/lib/kazoo/cli/cluster.rb

@@ -23,6 +23,13 @@ module Kazoo
           $stdout.puts "Broker #{broker_name} is non-critical and can be stopped safely."
         end
       end
+
+      desc "election", "Triggers a preferred leader election"
+      def election
+        validate_class_options!
+
+        kafka_cluster.preferred_leader_election
+      end
     end
   end
 end

data/lib/kazoo/cluster.rb

@@ -102,6 +102,21 @@ module Kazoo
       partitions.any?(&:under_replicated?)
     end
 
+    # Triggers a preferred leader elections for the provided list of partitions. If no list of
+    # partitions is provided, the preferred leader will be elected for all partitions in the cluster.
+    def preferred_leader_election(partitions: nil)
+      partitions = self.partitions if partitions.nil?
+      result = zk.create(path: "/admin/preferred_replica_election", data: JSON.generate(version: 1, partitions: partitions))
+      case result.fetch(:rc)
+      when Zookeeper::Constants::ZOK
+        return true
+      when Zookeeper::Constants::ZNODEEXISTS
+        raise Kazoo::Error, "Another preferred leader election is still in progress"
+      else
+        raise Kazoo::Error, "Failed to start preferred leadership election. Result code: #{result.fetch(:rc)}"
+      end
+    end
+
     # Closes the zookeeper connection and clears all the local caches.
     def close
       zk.close

data/lib/kazoo/partition.rb

@@ -81,6 +81,10 @@ module Kazoo
       end
     end
 
+    def to_json(generator)
+      generator.generate(topic: topic.name, partition: id)
+    end
+
     protected
 
     def refresh_state

data/lib/kazoo/version.rb

@@ -1,3 +1,3 @@
 module Kazoo
-  VERSION = "0.4.0"
+  VERSION = "0.5.0"
 end

data/test/cluster_test.rb

@@ -13,4 +13,30 @@ class ClusterTest < Minitest::Test
     @cluster.topics['test.4'].partitions[2].expects(:isr).returns([@cluster.brokers[1]])
     assert @cluster.under_replicated?
   end
+
+  def test_preferred_leader_election
+    @cluster.zk.stubs(:create).with do |parameters|
+      payload = JSON.parse(parameters.fetch(:data))
+
+      [
+        parameters.fetch(:path) == "/admin/preferred_replica_election",
+        payload.fetch('version') == 1,
+        payload.fetch('partitions').length == 1,
+        payload.fetch('partitions').first.fetch('topic') == 'test.1',
+        payload.fetch('partitions').first.fetch('partition') == 0,
+      ].all?
+    end.returns(rc: Zookeeper::Constants::ZOK)
+
+    assert @cluster.preferred_leader_election(partitions: [@cluster.topic('test.1').partition(0)])
+  end
+
+
+  def test_preferred_leader_election_in_progress
+    @cluster.zk.stubs(:create).returns(rc: Zookeeper::Constants::ZNODEEXISTS)
+    exception = assert_raises(Kazoo::Error) do
+      @cluster.preferred_leader_election
+    end
+
+    assert_equal "Another preferred leader election is still in progress", exception.message
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kazoo-ruby
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Willem van Bergen
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-10-09 00:00:00.000000000 Z
+date: 2015-12-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor