ruby-kafka 0.7.0 → 0.7.1.beta1

data/lib/kafka/transaction_state_machine.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+module Kafka
+  class TransactionStateMachine
+    class InvalidTransitionError < StandardError; end
+    class InvalidStateError < StandardError; end
+
+    STATES = [
+      UNINITIALIZED          = :uninitialized,
+      READY                  = :ready,
+      IN_TRANSACTION         = :in_trasaction,
+      COMMITTING_TRANSACTION = :committing_transaction,
+      ABORTING_TRANSACTION   = :aborting_transaction,
+      ERROR                  = :error
+    ]
+
+    TRANSITIONS = {
+      UNINITIALIZED          => [READY, ERROR],
+      READY                  => [UNINITIALIZED, COMMITTING_TRANSACTION, ABORTING_TRANSACTION],
+      IN_TRANSACTION         => [READY],
+      COMMITTING_TRANSACTION => [IN_TRANSACTION],
+      ABORTING_TRANSACTION   => [IN_TRANSACTION],
+      # Any states can transition to error state
+      ERROR                  => STATES
+    }
+
+    def initialize(logger:)
+      @state = UNINITIALIZED
+      @mutex = Mutex.new
+      @logger = logger
+    end
+
+    def transition_to!(next_state)
+      raise InvalidStateError unless STATES.include?(next_state)
+      unless TRANSITIONS[next_state].include?(@state)
+        raise InvalidTransitionError, "Could not transition from state '#{@state}' to state '#{next_state}'"
+      end
+      @logger.debug("Transaction state changed to '#{next_state}'!")
+      @mutex.synchronize { @state = next_state }
+    end
+
+    def uninitialized?
+      in_state?(UNINITIALIZED)
+    end
+
+    def ready?
+      in_state?(READY)
+    end
+
+    def in_transaction?
+      in_state?(IN_TRANSACTION)
+    end
+
+    def committing_transaction?
+      in_state?(COMMITTING_TRANSACTION)
+    end
+
+    def aborting_transaction?
+      in_state?(ABORTING_TRANSACTION)
+    end
+
+    def error?
+      in_state?(ERROR)
+    end
+
+    private
+
+    def in_state?(state)
+      @mutex.synchronize { @state == state }
+    end
+  end
+end

data/lib/kafka/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Kafka
-  VERSION = "0.7.0"
+  VERSION = "0.7.1.beta1"
 end

data/ruby-kafka.gemspec

@@ -41,7 +41,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "extlz4"
   spec.add_development_dependency "colored"
   spec.add_development_dependency "rspec_junit_formatter", "0.2.2"
-  spec.add_development_dependency "dogstatsd-ruby", ">= 3.0.0"
+  spec.add_development_dependency "dogstatsd-ruby", ">= 3.0.0", "<4.0.0"
   spec.add_development_dependency "statsd-ruby"
   spec.add_development_dependency "ruby-prof"
   spec.add_development_dependency "timecop"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-kafka
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.7.1.beta1
 platform: ruby
 authors:
 - Daniel Schierbeck
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-08-14 00:00:00.000000000 Z
+date: 2018-08-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: digest-crc
@@ -199,6 +199,9 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 3.0.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 4.0.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -206,6 +209,9 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 3.0.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 4.0.0
 - !ruby/object:Gem::Dependency
   name: statsd-ruby
   requirement: !ruby/object:Gem::Requirement
@@ -338,7 +344,9 @@ files:
 - lib/kafka/datadog.rb
 - lib/kafka/fetch_operation.rb
 - lib/kafka/fetched_batch.rb
+- lib/kafka/fetched_batch_generator.rb
 - lib/kafka/fetched_message.rb
+- lib/kafka/fetched_offset_resolver.rb
 - lib/kafka/fetcher.rb
 - lib/kafka/gzip_codec.rb
 - lib/kafka/heartbeat.rb
@@ -353,6 +361,8 @@ files:
 - lib/kafka/produce_operation.rb
 - lib/kafka/producer.rb
 - lib/kafka/protocol.rb
+- lib/kafka/protocol/add_partitions_to_txn_request.rb
+- lib/kafka/protocol/add_partitions_to_txn_response.rb
 - lib/kafka/protocol/alter_configs_request.rb
 - lib/kafka/protocol/alter_configs_response.rb
 - lib/kafka/protocol/api_versions_request.rb
@@ -370,12 +380,16 @@ files:
 - lib/kafka/protocol/describe_groups_request.rb
 - lib/kafka/protocol/describe_groups_response.rb
 - lib/kafka/protocol/encoder.rb
+- lib/kafka/protocol/end_txn_request.rb
+- lib/kafka/protocol/end_txn_response.rb
 - lib/kafka/protocol/fetch_request.rb
 - lib/kafka/protocol/fetch_response.rb
 - lib/kafka/protocol/find_coordinator_request.rb
 - lib/kafka/protocol/find_coordinator_response.rb
 - lib/kafka/protocol/heartbeat_request.rb
 - lib/kafka/protocol/heartbeat_response.rb
+- lib/kafka/protocol/init_producer_id_request.rb
+- lib/kafka/protocol/init_producer_id_response.rb
 - lib/kafka/protocol/join_group_request.rb
 - lib/kafka/protocol/join_group_response.rb
 - lib/kafka/protocol/leave_group_request.rb
@@ -412,6 +426,8 @@ files:
 - lib/kafka/ssl_context.rb
 - lib/kafka/ssl_socket_with_timeout.rb
 - lib/kafka/statsd.rb
+- lib/kafka/transaction_manager.rb
+- lib/kafka/transaction_state_machine.rb
 - lib/kafka/version.rb
 - lib/ruby-kafka.rb
 - ruby-kafka.gemspec
@@ -430,9 +446,9 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: 2.1.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - ">="
+  - - ">"
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 1.3.1
 requirements: []
 rubyforge_project: 
 rubygems_version: 2.7.6