jruby-kafka 1.3.0-java → 1.4.0-java

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0c0f698d22e7c8ef1c889c342b79d79788cbcd53
-  data.tar.gz: 4dd19fdd27dad557914b1ceb87614ef97f22792f
+  metadata.gz: 6a5fbbd24312699a8ecebe3875234e202c20fc0b
+  data.tar.gz: 7c6f1a9a2085ec7574797fc79c29cd0f7eacd8e8
 SHA512:
-  metadata.gz: a24b0153e5bcfbf6cc78891dcfe3d71f7edc46578da8765be01c749794299dda63421d2385220e1673a02e207130fd35016ac0e979a1c5d2443be9cdbaf3cfa4
-  data.tar.gz: 3b68ee09f663bfa55a8e13b41386b5c7dea8b74cce03e9f569af1c540eb1774cdbdab2707f3e6844d6ca40c0c6e3d70f42618bca51ab2b40d32e787c418c12a0
+  metadata.gz: f52d35afd1daa035e48f5bd8da20c12e13d01ae51b8e31314dc9cb3905f47be1feef1e642c06a2e425e7e8c863c46ffd8ebcddaf0bcac7a621efdc1092769311
+  data.tar.gz: 528e5846605ec8d144174f6de4157e5f1ec06619c4fd7b2fc9be9a74bf5c076ae4508d43d6e04b73f831c7bd27b2d88042da2a5a37e271da139f7d67afc6bfa5

data/lib/jruby-kafka/kafka-producer.rb

@@ -0,0 +1,76 @@
+require 'jruby-kafka/namespace'
+require 'jruby-kafka/error'
+
+# noinspection JRubyStringImportInspection
+class Kafka::KafkaProducer
+  java_import 'org.apache.kafka.clients.producer.ProducerRecord'
+  KAFKA_PRODUCER = Java::org.apache.kafka.clients.producer.KafkaProducer
+
+  VALIDATIONS = {
+      :'required.codecs' => %w[
+      none gzip snappy lz4
+    ]
+  }
+
+  REQUIRED = %w[
+    bootstrap.servers key.serializer
+  ]
+
+  KNOWN = %w[
+    acks                      batch.size                block.on.buffer.full
+    bootstrap.servers         buffer.memory             client.id
+    compression.type          key.serializer            linger.ms
+    max.in.flight.requests.per.connection               max.request.size
+    metadata.fetch.timeout.ms metadata.max.age.ms       metric.reporters
+    metrics.num.samples       metrics.sample.window.ms  receive.buffer.bytes
+    reconnect.backoff.ms      retries                   retry.backoff.ms
+    send.buffer.bytes         timeout.ms                value.serializer
+  ]
+
+  attr_reader :producer, :send_method, :options
+
+  def initialize(opts = {})
+    @options = opts.reduce({}) do |opts_array, (k, v)|
+      unless v.nil?
+        opts_array[k.to_s.gsub(/_/, '.')] = v
+      end
+      opts_array
+    end
+    validate_arguments
+    @send_method = proc { throw StandardError.new 'Producer is not connected' }
+  end
+
+  def connect
+    @producer = KAFKA_PRODUCER.new(create_producer_config)
+    @send_method = producer.java_method :send, [ProducerRecord]
+  end
+
+  # throws FailedToSendMessageException or if not connected, StandardError.
+  def send_msg(topic, partition, key, value)
+    send_method.call(ProducerRecord.new(topic, partition, key, value))
+  end
+
+  def close
+    @producer.close
+  end
+
+  private
+
+  def validate_arguments
+    errors = []
+    missing = REQUIRED.reject { |opt| options[opt] }
+    errors = ["Required settings: #{ missing.join(', ')}"] if missing.any?
+    invalid = VALIDATIONS.reject { |opt, valid| options[opt].nil? or valid.include? options[opt].to_s }
+    errors += invalid.map { |opt, valid| "#{ opt } should be one of: [#{ valid.join(', ')}]" }
+    fail StandardError.new "Invalid configuration arguments: #{ errors.join('; ') }" if errors.any?
+    options.keys.each do |opt|
+      STDERR.puts "WARNING: Unknown configuration key: #{opt}" unless KNOWN.include? opt
+    end
+  end
+
+  def create_producer_config
+    properties = java.util.Properties.new
+    options.each { |opt, value| properties.put opt, value.to_s }
+    properties
+  end
+end

data/lib/jruby-kafka.rb

@@ -2,6 +2,7 @@ require 'jruby-kafka_jars.rb'
 require 'jruby-kafka/consumer'
 require 'jruby-kafka/group'
 require 'jruby-kafka/producer'
+require 'jruby-kafka/kafka-producer'
 
 module Kafka
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: jruby-kafka
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.4.0
 platform: java
 authors:
 - Joseph Lawson
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2015-03-20 00:00:00.000000000 Z
+date: 2015-03-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   requirement: !ruby/object:Gem::Requirement
@@ -71,6 +71,7 @@ files:
 - lib/jruby-kafka/consumer.rb
 - lib/jruby-kafka/error.rb
 - lib/jruby-kafka/group.rb
+- lib/jruby-kafka/kafka-producer.rb
 - lib/jruby-kafka/namespace.rb
 - lib/jruby-kafka/producer.rb
 - lib/jruby-kafka_jars.rb