kafka-rest-rb 0.1.0.alpha2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: ef34042990b7684ddc6a59d1ac9ba8ee44f87d30
+  data.tar.gz: 95dd2134ff748d790ffdb5b6ab216d99d326b59c
+SHA512:
+  metadata.gz: 246f29462bbf834ae1952c7b010bea02a7bfbd8b0613b6cbfd506755144d8dcab06291346ce6db3f5b2556ed9a0002e14257152c9403c91b8c5816c190b04d35
+  data.tar.gz: eb2a8e82e9740b6e677605b848a2e2cf70330e76670533392059689acf317dac87380014fc864f0cc666185981a56a0aa6920b69aefd965bc94c4cddff117e25

data/.gitignore

@@ -0,0 +1,11 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+
+*.gem

data/.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

data/.travis.yml

@@ -0,0 +1,5 @@
+sudo: false
+language: ruby
+rvm:
+  - 2.2.2
+before_install: gem install bundler -v 1.12.5

data/CODE_OF_CONDUCT.md

@@ -0,0 +1,49 @@
+# Contributor Code of Conduct
+
+As contributors and maintainers of this project, and in the interest of
+fostering an open and welcoming community, we pledge to respect all people who
+contribute through reporting issues, posting feature requests, updating
+documentation, submitting pull requests or patches, and other activities.
+
+We are committed to making participation in this project a harassment-free
+experience for everyone, regardless of level of experience, gender, gender
+identity and expression, sexual orientation, disability, personal appearance,
+body size, race, ethnicity, age, religion, or nationality.
+
+Examples of unacceptable behavior by participants include:
+
+* The use of sexualized language or imagery
+* Personal attacks
+* Trolling or insulting/derogatory comments
+* Public or private harassment
+* Publishing other's private information, such as physical or electronic
+  addresses, without explicit permission
+* Other unethical or unprofessional conduct
+
+Project maintainers have the right and responsibility to remove, edit, or
+reject comments, commits, code, wiki edits, issues, and other contributions
+that are not aligned to this Code of Conduct, or to ban temporarily or
+permanently any contributor for other behaviors that they deem inappropriate,
+threatening, offensive, or harmful.
+
+By adopting this Code of Conduct, project maintainers commit themselves to
+fairly and consistently applying these principles to every aspect of managing
+this project. Project maintainers who do not follow or enforce the Code of
+Conduct may be permanently removed from the project team.
+
+This code of conduct applies both within project spaces and in public spaces
+when an individual is representing the project or its community.
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be
+reported by contacting a project maintainer at komolov.f@gmail.com. All
+complaints will be reviewed and investigated and will result in a response that
+is deemed necessary and appropriate to the circumstances. Maintainers are
+obligated to maintain confidentiality with regard to the reporter of an
+incident.
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage],
+version 1.3.0, available at
+[http://contributor-covenant.org/version/1/3/0/][version]
+
+[homepage]: http://contributor-covenant.org
+[version]: http://contributor-covenant.org/version/1/3/0/

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in kafka-rest-rb.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2016 Theodore Konukhov
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,3 @@
+# Kafka-REST client, producer/consumer DSLs and worker daemon for Ruby/Rails apps. [![CircleCI](https://circleci.com/gh/konukhov/kafka-rest-rb.svg?style=shield)](https://circleci.com/gh/konukhov/kafka-rest-rb)
+
+In early development stage. Come back later to see the docs!

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "kafka-rest"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/kafka-rest

@@ -0,0 +1,19 @@
+#!/usr/bin/env ruby
+
+# TODO, pids, config and stuff
+
+ENV['RAILS_ENV'] ||= ENV['RACK_ENV'] || 'development'
+
+app_path = ENV['APP_PATH'] || '.'
+
+require 'kafka_rest/logging'
+require 'kafka_rest/worker'
+
+require File.expand_path(app_path, 'config/environment.rb')
+
+Rails.application.eager_load!
+
+client  = KafkaRest::Client.new
+worker  = KafkaRest::Worker.new(client)
+
+worker.start

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/kafka-rest-rb.gemspec

@@ -0,0 +1,30 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'kafka_rest/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "kafka-rest-rb"
+  spec.version       = KafkaRest::VERSION
+  spec.authors       = ["Theodore Konukhov"]
+  spec.email         = ["me@thdr.io"]
+
+  spec.summary       = %q{Kafka-REST proxy client for Ruby on Rails.}
+  spec.description   = %q{Kafka-REST client, DSLs and consumer workers for Ruby.}
+  spec.homepage      = "https://github.com/konukhov/kafka-rest-rb"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = 'bin'
+  spec.executables   = ['kafka-rest']
+  spec.require_paths = ['lib']
+
+  spec.add_runtime_dependency 'faraday', '~> 0.9'
+  spec.add_runtime_dependency 'faraday_middleware', '~> 0.10'
+  spec.add_runtime_dependency 'concurrent-ruby', '~> 1.0'
+  spec.add_runtime_dependency 'oj', '~> 2.17'
+
+  spec.add_development_dependency "bundler", "~> 1.12"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+end

data/lib/kafka-rest.rb

@@ -0,0 +1 @@
+require 'kafka_rest'

data/lib/kafka_rest/client/middleware.rb

@@ -0,0 +1,69 @@
+require 'faraday'
+require 'faraday_middleware/response_middleware'
+require 'oj'
+
+module KafkaRest
+  class Client
+    class KafkaRestClientException < StandardError
+      attr_reader :body, :status
+
+      def initialize(resp)
+        @body = resp.body
+        @status = resp.status
+
+        super "#{@body['message']}" +
+              " (HTTP Status: #{@status}; " +
+              "error code: #{@body['error_code']})"
+      end
+    end
+
+    class DefaultHeaders < Faraday::Middleware
+      def initialize(app = nil, default_headers = {})
+        @default_headers = default_headers
+        super(app)
+      end
+
+      def call(env)
+        env[:request_headers] = @default_headers.merge env[:request_headers]
+        @app.call(env)
+      end
+    end
+
+    class JsonRequest < Faraday::Middleware
+      def call(env)
+        if env[:body]
+          env[:body] = Oj.dump env[:body], mode: :compat, symbol_keys: false
+        end
+
+        @app.call(env)
+      end
+    end
+
+    class JsonResponse < FaradayMiddleware::ResponseMiddleware
+      define_parser do |body|
+        Oj.load(body)
+      end
+    end
+
+    class RaiseException < FaradayMiddleware::ResponseMiddleware
+      def call(env)
+        response = @app.call(env)
+        response.on_complete do
+          unless response.success?
+            raise KafkaRestClientException.new(response)
+          end
+        end
+      end
+    end
+
+    Faraday::Request.register_middleware(
+      default_headers: DefaultHeaders,
+      encode_json: JsonRequest
+    )
+
+    Faraday::Response.register_middleware(
+      decode_json: JsonResponse,
+      raise_exception: RaiseException
+    )
+  end
+end

data/lib/kafka_rest/client.rb

@@ -0,0 +1,137 @@
+require 'kafka_rest/client/middleware.rb'
+require 'faraday'
+
+module KafkaRest
+  class Client
+    def initialize
+      @conn = Faraday.new(url: KafkaRest.config.url) do |c|
+        c.request :encode_json
+        c.request :default_headers, default_headers
+
+        c.response :raise_exception
+        c.response :decode_json
+
+        c.adapter :net_http_persistent
+      end
+    end
+
+    # Get list of topics
+    ### returns: array of topics
+    def topics
+      @conn.get("/topics")
+    end
+
+    # Get topic metadata by name
+    ### returns: name, configs, partitions
+    def topic(topic)
+      @conn.get("/topics/#{topic}")
+    end
+
+    # Get topic's partitions
+    ###
+    def topic_partitions(topic)
+      @conn.get("/topics/#{topic}/partitions")
+    end
+
+    # Get topic's partition metadata
+    def topic_partition(topic, partition)
+      @conn.get("/topics/#{topic}/partitions/#{partition}")
+    end
+
+    # Get messages from topic's partition.
+    def topic_partition_messages(topic, partition, params = {})
+      params[:count] ||= 1
+      format = params.delete(:format) || 'binary'
+
+      @conn.get(
+        "/topics/#{topic}/partitions/#{partition}/messages",
+        params,
+        accept(format)
+      )
+    end
+
+    def produce_message(path, records, format, params)
+      body = params.merge(
+        records: records.is_a?(Array) ? records : [records]
+      )
+
+      @conn.post(path, body, content_type(format))
+    end
+    private :produce_message
+
+    # Produce message into a topic
+    ### params: key_schema, value_schema, key_schema_id, value_schema_id, records { key, value, partition }
+    ### returns: key_schema_id, value_schema_id, offsets { partition, offset, error_code, error }
+    def topic_produce_message(topic, records, format = 'json', params = {})
+      produce_message("topics/#{topic}", records, format, params)
+    end
+
+    # Produce message into a topic and partition
+    ### see topic_produce_message
+    def topic_partition_produce_message(topic, partition, records, format = 'json', params = {})
+      produce_message("topics/#{topic}/partitions/#{partition}", records, format, params)
+    end
+
+    # Add new consumer to a group
+    ### params: name, format, auto.offset.reset, auto.commit.enable
+    ### returns: instance_id, base_uri
+    def consumer_add(group_name, params = {})
+      body = {}
+      body['auto.offset.reset'] = params[:auth_offset_reset] || 'largest'
+      body['auto.commit.enable'] = params[:auto_commit_enable] == true || false
+      body['format'] = params[:format] || 'json'
+
+      @conn.post("consumers/#{group_name}", body)
+    end
+
+    def consumer_commit_offsets(group_name, consumer_id)
+      @conn.post("consumers/#{group_name}/instances/#{consumer_id}/offsets")
+    end
+
+    def consumer_remove(group_name, consumer_id)
+      @conn.delete("consumers/#{group_name}/instances/#{consumer_id}")
+    end
+
+    def consumer_consume_from_topic(group_name, consumer_id, topic, params = {})
+      format = params.delete(:format) || 'json'
+
+      @conn.get(
+        "consumers/#{group_name}/instances/#{consumer_id}/topics/#{topic}",
+        params,
+        accept(format)
+      )
+    end
+
+    def brokers
+      @conn.get("/brokers")
+    end
+
+    private
+
+    def default_headers
+      {
+        'Accept' => 'application/vnd.kafka.v1+json, application/vnd.kafka+json, application/json',
+        'Content-Type' => 'application/vnd.kafka.v1+json'
+      }
+    end
+
+    def kafka_mime_type(format = :json)
+      case format.to_sym
+      when :avro
+        'application/vnd.kafka.avro.v1+json'
+      when :binary
+        'application/vnd.kafka.binary.v1+json'
+      when :json
+        'application/vnd.kafka.json.v1+json'
+      end
+    end
+
+    def content_type(format = nil, headers = {})
+      headers.merge 'Content-Type' => kafka_mime_type(format)
+    end
+
+    def accept(format, headers = {})
+      headers.merge 'Accept' => kafka_mime_type(format)
+    end
+  end
+end

data/lib/kafka_rest/config.rb

@@ -0,0 +1,29 @@
+module KafkaRest
+  class Config
+    attr_accessor :url,
+                  :message_format,
+                  :serialization_adapter,
+                  :worker_min_threads,
+                  :worker_max_threads,
+                  :worker_max_queue
+
+    def initialize
+      @url = 'http://localhost:8082'
+      @message_format = 'json'
+      @serialization_adapter = nil
+      @worker_min_threads = 4
+      @worker_max_threads = 4
+      @worker_max_queue = nil
+    end
+  end
+
+  @@config = Config.new
+
+  def self.configure(&block)
+    block.call @@config
+  end
+
+  def self.config
+    @@config
+  end
+end

data/lib/kafka_rest/consumer.rb

@@ -0,0 +1,37 @@
+require 'kafka_rest/dsl'
+
+module KafkaRest
+  module Consumer
+    def self.included(base)
+      base.class_eval do
+        extend Dsl
+
+        option :topic, required: true
+
+        option :group_name, required: true
+
+        option :format, default: KafkaRest.config.message_format, validate: ->(v){
+          %w(json binary avro).include?(v.to_s)
+        }, error_message: 'Format must be either `json`, `avro` or `binary`'
+
+        option :auto_commit, default: false
+
+        option :offset_reset, default: :largest, validate: ->(val){
+          %w(smallest largest).include?(val.to_s)
+        }, error_message: 'Offset reset strategy must be `smallest` or `largest`'
+
+        option :max_bytes
+
+        option :poll_delay, default: 0.5, validate: ->(val){
+          val > 0
+        }, error_message: 'Poll delay should be a number greater than zero'
+      end
+
+      Worker::ConsumerManager.register!(base)
+    end
+
+    def receive(*args)
+      raise NotImplementedError
+    end
+  end
+end

data/lib/kafka_rest/dsl.rb

@@ -0,0 +1,46 @@
+module KafkaRest
+  module Dsl
+    class MissingRequiredOption  < StandardError; end
+    class InvalidOptionValue     < StandardError; end
+
+    def option(name, opts = {})
+      name         = name.to_s
+      required     = opts[:required] || false
+      default      = opts[:default]
+      validate     = opts[:validate] || ->(val) { true }
+      error_msg    = opts[:error_message] || "`#{name}`'s value is invalid"
+
+      class_eval do
+        metaclass = class << self; self; end
+        instance_variable_set "@#{name}", default
+        metaclass.send :define_method, "_validate_#{name}", ->(val) { validate.call(val) }
+      end
+
+      class_eval %Q{
+        def #{name}
+          self.class.get_#{name}
+        end
+
+        class << self
+          def get_#{name}
+            @#{name}.tap do |v|
+              if #{required} && v.nil?
+                raise KafkaRest::Dsl::MissingRequiredOption.new(
+                  "Missing required option `#{name}`"
+                )
+              end
+            end
+          end
+
+          def #{name}(val)
+            unless _validate_#{name}(val)
+              raise KafkaRest::Dsl::InvalidOptionValue.new("#{error_msg}")
+            end
+
+            @#{name} = val
+          end
+        end
+      }
+    end
+  end
+end

data/lib/kafka_rest/logging.rb

@@ -0,0 +1,20 @@
+module KafkaRest
+  module Logging
+    def self.logger
+      @logger ||= (
+        require 'logger'
+        ::Logger.new(STDOUT).tap do |l|
+          l.level = ::Logger::INFO
+        end
+      )
+    end
+
+    def self.logger=(l)
+      @logger = l
+    end
+
+    def logger
+      Logging.logger
+    end
+  end
+end

data/lib/kafka_rest/producer/serialization/active_model.rb

@@ -0,0 +1,27 @@
+module KafkaRest
+  module Producer
+    module Serialization
+      class ActiveModel < Adapter
+        def serialize(obj, opts = {})
+          klass.new(obj, opts).as_json
+        end
+
+        private
+
+        def klass
+          @klass ||= (
+            unless defined?(::ActiveModel::Serializer)
+              'ActiveModel::Serializer cannot be found'
+            end
+
+            if (kl = @args.first) && kl < ::ActiveModel::Serializer
+              kl
+            else
+              raise 'Provide ActiveModel::Serializer child as an argunent to `serializer`'
+            end
+          )
+        end
+      end
+    end
+  end
+end

data/lib/kafka_rest/producer/serialization/adapter.rb

@@ -0,0 +1,15 @@
+module KafkaRest
+  module Producer
+    module Serialization
+      class Adapter
+        def initialize(*args)
+          @args = args
+        end
+
+        def serialize(obj, options = {})
+          raise NotImplementedError
+        end
+      end
+    end
+  end
+end

data/lib/kafka_rest/producer/serialization/noop.rb

@@ -0,0 +1,11 @@
+module KafkaRest
+  module Producer
+    module Serialization
+      class Noop < Adapter
+        def serialize(obj, opts = {})
+          obj.to_s
+        end
+      end
+    end
+  end
+end

data/lib/kafka_rest/producer.rb

@@ -0,0 +1,69 @@
+require 'kafka_rest/dsl'
+
+module KafkaRest
+  module Producer
+    DEFAULT_KEY_SCHEMA = "{\"type\": \"string\"}"
+
+    def self.included(base)
+      base.class_eval do
+        extend ClassMethods
+        extend Dsl
+
+        option :topic, required: true
+
+        option :format, default: KafkaRest.config.message_format, validate: ->(v){
+          %w(json binary avro).include?(v.to_s)
+        }, error_message: 'Format must be `avro`, `json` or `binary`.'
+
+
+        option :key_schema, validate: ->(v){
+          v.is_a?(Symbol) || v.is_a?(String) || v.is_a?(Proc)
+        }, default: DEFAULT_KEY_SCHEMA
+
+        option :value_schema, validate: ->(v){
+          v.is_a?(Symbol) || v.is_a?(String) || v.is_a?(Proc)
+        }
+
+        option :key, validate: ->(val) {
+          if val
+            val.is_a?(Symbol) || val.is_a?(Proc)
+          else
+            true
+          end
+        }
+
+        option :serialization_adapter, validate: ->(val){
+          if val
+            val.is_a?(Class) && val < Serialization::Adapter
+          else
+            true
+          end
+        }, default: KafkaRest.config.serialization_adapter
+
+        option :serializer
+
+        class << base
+          # right away override default get_serializer and get_value_schema
+          def get_serializer
+            @serializer_inst ||= get_serialization_adapter.new @serializer
+          end
+
+          def get_value_schema
+            if get_format.to_s == 'avro' && @value_schema.nil?
+              raise 'Format `avro` requires providing `value_schema`'
+            end
+
+            @value_schema
+          end
+        end
+      end
+    end
+
+    module ClassMethods
+      def send!(obj, opts = {}, producer = nil)
+        (producer || KafkaRest::Sender.instance)
+          .send!(self, obj, opts)
+      end
+    end
+  end
+end

data/lib/kafka_rest/sender/payload/avro_builder.rb

@@ -0,0 +1,15 @@
+module KafkaRest
+  class Sender
+    class Payload
+      class AvroBuilder < Builder
+        # TODO: get rid of this
+        def build
+          {
+            key: @payload.key,
+            value: @payload.value,
+          }
+        end
+      end
+    end
+  end
+end

data/lib/kafka_rest/sender/payload/binary_builder.rb

@@ -0,0 +1,16 @@
+require 'base64'
+
+module KafkaRest
+  class Sender
+    class Payload
+      class BinaryBuilder < Builder
+        def build
+          {
+            key: @payload.key,
+            value: Base64.strict_encode64(@payload.value)
+          }
+        end
+      end
+    end
+  end
+end

data/lib/kafka_rest/sender/payload/builder.rb

@@ -0,0 +1,15 @@
+module KafkaRest
+  class Sender
+    class Payload
+      class Builder
+        def initialize(payload)
+          @payload = payload
+        end
+
+        def build
+          raise NotImplementedError
+        end
+      end
+    end
+  end
+end

data/lib/kafka_rest/sender/payload/json_builder.rb

@@ -0,0 +1,14 @@
+module KafkaRest
+  class Sender
+    class Payload
+      class JsonBuilder < Builder
+        def build
+          {
+            key: @payload.key,
+            value: @payload.value
+          }
+        end
+      end
+    end
+  end
+end

data/lib/kafka_rest/sender/payload.rb

@@ -0,0 +1,72 @@
+require 'kafka_rest/sender/payload/builder'
+require 'kafka_rest/sender/payload/avro_builder'
+require 'kafka_rest/sender/payload/json_builder'
+require 'kafka_rest/sender/payload/binary_builder'
+
+module KafkaRest
+  class Sender
+    class Payload
+      attr_reader :klass
+
+      def initialize(klass, obj, opts = {})
+        @klass   = klass
+        @obj     = obj
+        @opts    = opts
+        @builder = get_builder.new(self)
+
+        @key       = @opts.delete(:key)
+        @timestamp = @opts.delete(:timestamp)
+        @partition = @opts.delete(:partition)
+      end
+
+      def build
+        @builder.build.tap do |pl|
+          @timestamp and pl[:timestamp] = @timestamp
+          @partition and pl[:partition] = @partition
+        end
+      end
+
+      def value
+        @klass.get_serializer.serialize(@obj, @opts)
+      end
+
+      def key
+        return @key if @key
+
+        k = @klass.get_key
+
+        case k
+        when NilClass
+          k
+        when Symbol
+          if inst.respond_to?(k)
+            inst.send(k, @obj)
+          elsif @obj.respond_to?(k)
+            @obj.send(k)
+          else
+            raise NoMethodError.new("Undefined method \"#{k}\"")
+          end
+        when Proc
+          k.call(@obj)
+        end
+      end
+
+      private
+
+      def get_builder
+        case klass.get_format
+        when :avro
+          AvroBuilder
+        when :json
+          JsonBuilder
+        when :binary
+          BinaryBuilder
+        end
+      end
+
+      def inst
+        @inst ||= @klass.new
+      end
+    end
+  end
+end

data/lib/kafka_rest/sender.rb

@@ -0,0 +1,81 @@
+require 'thread'
+require 'kafka_rest/sender/payload'
+
+module KafkaRest
+  class Sender
+    @@lock = Mutex.new
+
+    class << self
+      def instance
+        @@lock.synchronize do
+          @instance ||= self.new(Client.new, lock: @@lock)
+        end
+      end
+    end
+
+    attr_reader :key_schema_cache, :value_schema_cache
+
+    # TODO: buffering???
+    def initialize(client, opts = {})
+      @lock = opts[:lock] || Mutex.new
+      @client = client
+      @key_schema_cache = {}
+      @value_schema_cache = {}
+    end
+
+    # TODO: back-off retry if offset[i].errors is a retriable error
+    def send!(klass, obj, opts = {})
+      topic, payload, format, params = build_request(klass, obj, opts)
+      send_produce_request!(topic, payload, format, params)
+    end
+
+    private
+
+    def build_request(klass, obj, opts)
+      # TODO: oooh, dirty and weird - this should not be here.
+      #       come up with something good!
+      topic    = klass.get_topic.to_s
+      key      = klass.get_key
+      payload  = Payload.new(klass, obj, opts).build
+      format   = klass.get_format.to_s
+      params   = {}.tap do |_p|
+        if format == 'avro'
+          unless key.nil?
+            if kid = @key_schema_cache[topic]
+              _p[:key_schema_id] = kid
+            else
+              _p[:key_schema] = klass.get_key_schema
+            end
+          end
+
+          if vid = @value_schema_cache[topic]
+            _p[:value_schema_id] = vid
+          else
+            _p[:value_schema] = klass.get_value_schema
+          end
+        end
+      end
+
+      [topic, payload, format, params]
+    end
+
+    def send_produce_request!(topic, payload, format, params)
+      @client.topic_produce_message(topic, payload, format, params).body.tap do |re|
+        # this too (line 27)
+        cache_schema_ids!(re, topic) if format == 'avro'
+      end['offsets']
+    end
+
+    def cache_schema_ids!(resp, topic)
+      @lock.synchronize do
+        if @key_schema_cache[topic].nil? && kid = resp['key_schema_id']
+          @key_schema_cache[topic] = kid
+        end
+
+        if @value_schema_cache[topic].nil? && vid = resp['value_schema_id']
+          @value_schema_cache[topic] = vid
+        end
+      end
+    end
+  end
+end

data/lib/kafka_rest/version.rb

@@ -0,0 +1,3 @@
+module KafkaRest
+  VERSION = '0.1.0.alpha2'
+end

data/lib/kafka_rest/worker/consumer_manager.rb

@@ -0,0 +1,134 @@
+require 'concurrent/utility/monotonic_time'
+
+module KafkaRest
+  class Worker
+    class ConsumerManager
+      STATES = [:initial, :idle, :working, :dead]
+
+      include KafkaRest::Logging
+
+      class << self
+        @@consumers = []
+
+        def register!(consumer_class)
+          # TODO: raise exception if group_id + topic are not unique
+          # TODO: Thread.current???
+          @@consumers << consumer_class
+        end
+
+        def consumers
+          @@consumers
+        end
+      end
+
+      extend Forwardable
+
+      def_delegators :@consumer,
+                     :topic,
+                     :group_name,
+                     :poll_delay,
+                     :auto_commit,
+                     :offset_reset,
+                     :format,
+                     :max_bytes
+
+      def initialize(client, consumer)
+        @client    = client
+        @consumer  = consumer.new
+        @id        = nil
+        @uri       = nil
+        @state     = :initial
+        @next_poll = Concurrent.monotonic_time
+        @lock      = Mutex.new
+      end
+
+      STATES.each do |state|
+        class_eval %Q{
+          def #{state}?(lock = true)
+            with_lock(lock) { @state == :#{state} }
+          end
+        }
+      end
+
+      def poll?
+        with_lock {
+          idle?(false) && Concurrent.monotonic_time > @next_poll
+        }
+      end
+
+      def add!
+        params = {}.tap do |h|
+          auto_commit.nil? or h[:auto_commit_enable] = auto_commit
+          offset_reset and h[:auto_offset_reset] = offset_reset
+          format and h[:format] = format
+        end
+
+        resp   = @client.consumer_add(group_name, params)
+        @id    = resp.body['instance_id']
+        @uri   = resp.body['base_uri']
+        @state = :idle
+
+        logger.info "[Kafka REST] Added consumer #{@id}"
+      end
+
+      def remove!
+        resp = @client.consumer_remove(group_name, @id)
+        logger.info "[Kafka REST] Removed consumer #{@id}"
+      end
+
+      def poll!
+        begin
+          with_lock do
+            return false unless idle?(false)
+            @state = :working
+          end
+
+          logger.debug "Polling #{group_name}..."
+
+          params = {}.tap do |h|
+            format and h[:format] = format
+            max_bytes and h[:max_bytes] = max_bytes
+          end
+
+          messages = @client.consumer_consume_from_topic(
+            group_name,
+            @id,
+            topic,
+            params
+          ).body
+
+          if messages.any?
+            messages.each do |msg|
+              logger.debug "[Kafka REST] Consumer #{@id} got message: #{msg}"
+              @consumer.receive(msg)
+            end
+
+            unless auto_commit
+              @client.consumer_commit_offsets(group_name, @id)
+            end
+
+            with_lock { @state = :idle }
+          else
+            with_lock do
+              @next_poll = Concurrent.monotonic_time + poll_delay
+              @state = :idle
+            end
+          end
+        rescue Exception => e # TODO: handle errors
+          logger.warn "[Kafka REST] Consumer died due to error: #{e.class}, #{e.message}"
+          with_lock { @state = :dead }
+        end
+      end
+
+      private
+
+      def with_lock(lock = true, &block)
+        if lock
+          @lock.synchronize &block
+        else
+          block.call
+        end
+      end
+    end
+  end
+end

data/lib/kafka_rest/worker.rb

@@ -0,0 +1,96 @@
+require 'kafka_rest/client'
+require 'kafka_rest/worker/consumer_manager'
+require 'concurrent/executor/thread_pool_executor'
+
+module KafkaRest
+  class Worker
+    BUSY_THREAD_POOL_DELAY = 0.5
+    NO_WORK_DELAY = 0.1
+
+    include KafkaRest::Logging
+
+    def initialize(client)
+      @client = client
+      @started = false
+      @thread_pool = Concurrent::ThreadPoolExecutor.new(
+        min_threads: KafkaRest.config.worker_min_threads,
+        max_threads: KafkaRest.config.worker_max_threads,
+        max_queue:   max_queue,
+        fallback_policy: :discard
+      )
+
+      @consumers = ConsumerManager.consumers.map do |kl|
+        ConsumerManager.new(@client, kl)
+      end
+    end
+
+    def start
+      begin
+        @running = true
+
+        trap(:SIGINT) do
+          stop
+        end
+
+        init_consumers
+        run_work_loop
+      rescue => e
+        logger.error "[Kafka REST] Got exception: #{e.class} (#{e.message})"
+        e.backtrace.each { |msg| logger.error "\t #{msg}" }
+        stop
+      end
+    end
+
+    def stop
+      logger.info "[Kafka REST] Stopping worker..."
+      @running = false
+      remove_consumers
+    end
+
+    private
+
+    def run_work_loop
+      while @running
+        check_dead!
+
+        jobs = @consumers.select(&:poll?)
+
+        if jobs.empty?
+          sleep(NO_WORK_DELAY)
+          next
+        end
+
+        pool_available = jobs.each do |c|
+          unless @thread_pool.post { c.poll! }
+            break(false)
+          end
+        end
+
+        unless pool_available
+          sleep(BUSY_THREAD_POOL_DELAY)
+        end
+      end
+    end
+
+    def check_dead!
+      # Do we need this?
+      if @consumers.all?(&:dead?)
+        logger.info "[Kafka REST] All consumers are dead. Quitting..."
+        stop
+      end
+    end
+
+    def init_consumers
+      @consumers.map &:add!
+    end
+
+    def remove_consumers
+      @consumers.reject(&:initial?).map &:remove!
+    end
+
+    def max_queue
+      KafkaRest.config.worker_max_queue ||
+        ConsumerManager.consumers.size * 2
+    end
+  end
+end

data/lib/kafka_rest.rb

@@ -0,0 +1,24 @@
+require 'kafka_rest/config'
+require 'kafka_rest/logging'
+require 'kafka_rest/client'
+require 'kafka_rest/worker'
+require 'kafka_rest/producer'
+require 'kafka_rest/producer/serialization/adapter'
+require 'kafka_rest/sender'
+require 'kafka_rest/consumer'
+
+KafkaRest.configure do |c|
+  serializers = KafkaRest::Producer::Serialization
+
+  if defined?(ActiveModelSerializers) || defined?(::ActiveModel::Serializer)
+    require 'kafka_rest/producer/serialization/active_model'
+    c.serialization_adapter = serializers::ActiveModel
+  # elsif defined?(JBuilder)
+  # TODO jbuilder is default
+  # elsif defined?(Rabl)
+  # TODO rabl is default
+  else
+    require 'kafka_rest/producer/serialization/noop'
+    c.serialization_adapter = serializers::Noop
+  end
+end

metadata

@@ -0,0 +1,176 @@
+--- !ruby/object:Gem::Specification
+name: kafka-rest-rb
+version: !ruby/object:Gem::Version
+  version: 0.1.0.alpha2
+platform: ruby
+authors:
+- Theodore Konukhov
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-10-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: faraday_middleware
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.10'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.10'
+- !ruby/object:Gem::Dependency
+  name: concurrent-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: oj
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.17'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.17'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: Kafka-REST client, DSLs and consumer workers for Ruby.
+email:
+- me@thdr.io
+executables:
+- kafka-rest
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/kafka-rest
+- bin/setup
+- kafka-rest-rb.gemspec
+- lib/kafka-rest.rb
+- lib/kafka_rest.rb
+- lib/kafka_rest/client.rb
+- lib/kafka_rest/client/middleware.rb
+- lib/kafka_rest/config.rb
+- lib/kafka_rest/consumer.rb
+- lib/kafka_rest/dsl.rb
+- lib/kafka_rest/logging.rb
+- lib/kafka_rest/producer.rb
+- lib/kafka_rest/producer/serialization/active_model.rb
+- lib/kafka_rest/producer/serialization/adapter.rb
+- lib/kafka_rest/producer/serialization/noop.rb
+- lib/kafka_rest/sender.rb
+- lib/kafka_rest/sender/payload.rb
+- lib/kafka_rest/sender/payload/avro_builder.rb
+- lib/kafka_rest/sender/payload/binary_builder.rb
+- lib/kafka_rest/sender/payload/builder.rb
+- lib/kafka_rest/sender/payload/json_builder.rb
+- lib/kafka_rest/version.rb
+- lib/kafka_rest/worker.rb
+- lib/kafka_rest/worker/consumer_manager.rb
+homepage: https://github.com/konukhov/kafka-rest-rb
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">"
+    - !ruby/object:Gem::Version
+      version: 1.3.1
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Kafka-REST proxy client for Ruby on Rails.
+test_files: []