avro2kafka 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 442cbcdebce5cdbf488aff50c61ab23dafbece43
+  data.tar.gz: 46a629b705531af0ada32f54987af81814772fd8
+SHA512:
+  metadata.gz: 6e9ec180f4fbe39e175fce8caa2dd97180f99faf35dfce6a50bde5b4fd4775380978b937602c57d6c3cb3cfe68fb301568de19a3c543eecb9d4d7b11b169531c
+  data.tar.gz: b4ee9b33830dbfbd22b9ee8b7064263c9df79d8b149968353e69c8a70b7035086ca19661d447a5dbd5d161b3fedd7c026af88e0454b838b9611cbef822680a68

data/CHANGELOG.md

@@ -0,0 +1,17 @@
+# Changelog
+
+All notable changes to this project are documented in this file.
+This project adheres to [Semantic Versioning](http://semver.org/).
+
+## [Unreleased]
+
+### Added
+
+### Changed
+
+## 0.1.0 (2015-06-09)
+Initial release
+
+### Added
+ * CLI (`avro2kafa`) to Avro files to Kafka (#1)
+ * Gem packaging

data/Gemfile

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

data/Gemfile.lock

@@ -0,0 +1,48 @@
+PATH
+  remote: .
+  specs:
+    avro2kafka (0.1.0)
+      avro (~> 1.7)
+      poseidon (~> 0.0.5)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    avro (1.7.7)
+      multi_json
+    bump (0.5.2)
+    coderay (1.1.0)
+    diff-lcs (1.2.5)
+    method_source (0.8.2)
+    multi_json (1.11.0)
+    poseidon (0.0.5)
+    pry (0.10.1)
+      coderay (~> 1.1.0)
+      method_source (~> 0.8.1)
+      slop (~> 3.4)
+    rake (10.4.2)
+    rspec (3.2.0)
+      rspec-core (~> 3.2.0)
+      rspec-expectations (~> 3.2.0)
+      rspec-mocks (~> 3.2.0)
+    rspec-core (3.2.3)
+      rspec-support (~> 3.2.0)
+    rspec-expectations (3.2.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.2.0)
+    rspec-mocks (3.2.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.2.0)
+    rspec-support (3.2.2)
+    slop (3.6.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  avro2kafka!
+  bump (~> 0.5)
+  bundler (~> 1.6)
+  pry (~> 0.10)
+  rake (~> 10.0)
+  rspec (~> 3.2)

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2015 Secret Sauce Partners, Inc.
+
+MIT License
+
+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,47 @@
+# Avro2Kafka
+
+Publish data from Avro files to Kafka in JSON format
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'avro2kafka'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install avro2kafka
+
+## Usage
+
+```
+Usage: avro2kafka [options] [file]
+```
+
+### Options
+
+`-b, --broker BROKER`
+The Kafka broker, eg. localhost:9092. This value is required.
+
+`-t, --topic TOPIC`
+The Kafka topic to publish to. This value is required.
+
+`-k, --key KEY`
+The fields in the avro that will act as keys for Kafka. If not supplied, unique and undefined keys will be generated, hence log compaction won't work.
+
+`-h, --help`
+Prints help
+
+## Contributing
+
+1. Fork it ( https://github.com/[my-github-username]/avro2kafka/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request

data/Rakefile

@@ -0,0 +1,18 @@
+require 'rspec/core/rake_task'
+require 'bundler/gem_tasks'
+require 'bump/tasks'
+
+# Remove pre and set rake tasks
+Rake.application.instance_eval do
+  %w[bump:pre bump:set].each do |task|
+    @tasks.delete(task)
+  end
+end
+
+# Default directory to look in is `/spec`
+# Run with `rake spec`
+RSpec::Core::RakeTask.new(:spec) do |task|
+  task.rspec_opts = ['--color', '--format', 'documentation']
+end
+
+task :default => :spec

data/avro2kafka.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'avro2kafka/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "avro2kafka"
+  spec.version       = Avro2Kafka::VERSION
+  spec.authors       = ["Peter Marton"]
+  spec.email         = ["martonpe@secretsaucepartners.com"]
+  spec.summary       = %q{Publish Avro files to Kafka}
+  spec.description   = %q{Publish Avro files to Kafka in JSON format}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.2"
+  spec.add_development_dependency "pry", "~> 0.10"
+  spec.add_development_dependency "bump", "~> 0.5"
+
+  spec.add_dependency "avro", "~> 1.7"
+  spec.add_dependency "poseidon", "~> 0.0.5"
+end

data/bin/avro2kafka

@@ -0,0 +1,46 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH << File.dirname(__FILE__) + '/../lib' if $0 == __FILE__
+require 'optparse'
+require 'avro2kafka'
+
+options = {}
+
+option_parser = OptionParser.new do |opts|
+  opts.banner = "Version #{Avro2Kafka::VERSION} of Avro2Kafka\n" \
+    "Usage: #{File.basename(__FILE__)} [options] [file]"
+
+  opts.on('-b', '--broker BROKER', 'The Kafka broker, eg. localhost:9092. This value is required.') do |broker|
+    options[:broker] = broker
+  end
+
+  opts.on('-t', '--topic TOPIC', 'The Kafka topic to publish to. This value is required.') do |topic|
+    options[:topic] = topic
+  end
+
+  opts.on('-k', '--key KEY', 'The fields in the avro that will act as keys for Kafka.') do |key|
+    options[:key] = key
+  end
+
+  opts.on('-h', '--help', 'Prints help') do
+    puts opts
+    exit
+  end
+end
+
+option_parser.parse!
+
+begin
+  raise OptionParser::MissingArgument.new('--broker') if options[:broker].nil?
+  raise OptionParser::MissingArgument.new('--topic') if options[:topic].nil?
+
+  Avro2Kafka.new(options).publish
+rescue OptionParser::MissingArgument => ex
+  puts ex.message
+
+  puts option_parser
+rescue Exception => e
+  puts 'Uh oh, something went wrong!'
+  puts e.message
+  puts e.backtrace.join("\n")
+end

data/lib/avro2kafka.rb

@@ -0,0 +1,28 @@
+require 'avro2kafka/version'
+require 'avro2kafka/avro_reader'
+require 'avro2kafka/kafka_publisher'
+
+class Avro2Kafka
+  attr_reader :input_path, :kafka_broker, :kafka_topic, :kafka_key
+
+  def initialize(options)
+    @input_path = ARGV.first
+    @kafka_broker = options.delete(:broker)
+    @kafka_topic = options.delete(:topic)
+    @kafka_key = options[:key]
+
+    @options = options
+  end
+
+  def reader
+    ARGF.lineno = 0
+    ARGF
+  end
+
+  def publish
+    records = AvroReader.new(reader).read
+    KafkaPublisher.new(kafka_broker, kafka_topic, kafka_key).publish(records)
+    puts "Avro file published to #{kafka_topic} topic on #{kafka_broker}!"
+  end
+
+end

data/lib/avro2kafka/avro_reader.rb

@@ -0,0 +1,16 @@
+require 'avro'
+
+class Avro2Kafka
+  class AvroReader
+    attr_reader :io
+
+    def initialize(io)
+      @io = io
+      @reader = Avro::IO::DatumReader.new()
+    end
+
+    def read
+      Avro::DataFile::Reader.new(io, @reader).to_enum
+    end
+  end
+end

data/lib/avro2kafka/kafka_publisher.rb

@@ -0,0 +1,28 @@
+require 'poseidon'
+require 'json'
+
+class Avro2Kafka
+  class KafkaPublisher
+    attr_reader :producer, :topic, :keys
+
+    def initialize(broker, topic, keys)
+      @producer = Poseidon::Producer.new([broker], "avro2kafka")
+      @topic = topic
+      @keys = keys.split(',').map(&:strip) if keys
+    end
+
+    def publish(records)
+      records.each_slice(100) do |batch|
+        messages = batch.map do |record|
+          if keys
+            message_key = keys.map { |key| record[key] }.join
+            Poseidon::MessageToSend.new(topic, record.to_json, message_key)
+          else
+            Poseidon::MessageToSend.new(topic, record.to_json)
+          end
+        end
+        producer.send_messages(messages)
+      end
+    end
+  end
+end

data/lib/avro2kafka/version.rb

@@ -0,0 +1,3 @@
+class Avro2Kafka
+  VERSION = "0.1.0"
+end

data/spec/avro2kafka/avro_reader_spec.rb

@@ -0,0 +1,36 @@
+require 'spec_helper'
+require 'avro'
+
+RSpec.describe Avro2Kafka::AvroReader do
+  describe '#read' do
+    let(:path) { './spec/support/data.avro' }
+
+    before do
+      @io = File.open(path, 'r')
+      @avro = Avro2Kafka::AvroReader.new(@io).read
+    end
+
+    after do
+      @io.close
+    end
+
+    it 'should return an Enumerator object' do
+      expect(@avro).to be_an_instance_of Enumerator
+    end
+
+    it 'should return 3 rows' do
+      expect(@avro.count).to eq 3
+    end
+
+    it 'should return the correct data' do
+      expect(@avro.map { |record| record }).to eq(
+        [
+          { 'id'=> 1, 'name'=> 'dresses',     'description'=> 'Dresses' },
+          { 'id'=> 2, 'name'=> 'female-tops', 'description'=> 'Female Tops' },
+          { 'id'=> 3, 'name'=> 'bras',        'description'=> 'Bras' }
+        ]
+      )
+    end
+
+  end
+end

data/spec/avro2kafka/kafka_publisher_spec.rb

@@ -0,0 +1,19 @@
+require 'spec_helper'
+require 'poseidon'
+
+RSpec.describe Avro2Kafka::KafkaPublisher do
+  describe '#publish' do
+    let(:broker) { 'localhost:9092' }
+    let(:topic) { 'feeds' }
+    let(:key) { 'name, id' }
+    let(:messages) { %w[message1, message2, message3] }
+
+    it 'should call publisher#send_messages' do
+      kafka_publisher = Avro2Kafka::KafkaPublisher.new(broker, topic, key)
+      mock_producer = double("Producer")
+      allow(kafka_publisher).to receive(:producer).and_return(mock_producer)
+      expect(mock_producer).to receive(:send_messages).with(all(be_kind_of(Poseidon::MessageToSend)))
+      kafka_publisher.publish(messages)
+    end
+  end
+end

data/spec/avro2kafka_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+require 'poseidon'
+require 'json'
+
+RSpec.describe Avro2Kafka do
+  describe '#publish' do
+    let(:options) do
+      {
+        broker: 'localhost:9092',
+        topic: 'feeds',
+        key: 'name, id'
+      }
+    end
+
+    before do
+      ARGV.replace ['./spec/support/data.avro']
+    end
+
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,3 @@
+require 'avro2kafka'
+
+Dir[File.dirname(__FILE__) + '/support/**/*.rb'].each {|f| require f }

metadata

@@ -0,0 +1,166 @@
+--- !ruby/object:Gem::Specification
+name: avro2kafka
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Peter Marton
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-06-11 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !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.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.10'
+- !ruby/object:Gem::Dependency
+  name: bump
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+- !ruby/object:Gem::Dependency
+  name: avro
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: poseidon
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.5
+description: Publish Avro files to Kafka in JSON format
+email:
+- martonpe@secretsaucepartners.com
+executables:
+- avro2kafka
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- avro2kafka.gemspec
+- bin/avro2kafka
+- lib/avro2kafka.rb
+- lib/avro2kafka/avro_reader.rb
+- lib/avro2kafka/kafka_publisher.rb
+- lib/avro2kafka/version.rb
+- spec/avro2kafka/avro_reader_spec.rb
+- spec/avro2kafka/kafka_publisher_spec.rb
+- spec/avro2kafka_spec.rb
+- spec/spec_helper.rb
+- spec/support/data.avro
+homepage: ''
+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: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Publish Avro files to Kafka
+test_files:
+- spec/avro2kafka/avro_reader_spec.rb
+- spec/avro2kafka/kafka_publisher_spec.rb
+- spec/avro2kafka_spec.rb
+- spec/spec_helper.rb
+- spec/support/data.avro
+has_rdoc: