codeclimate-kafka 0.0.2 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 60a376171edc1e992b10b08d773cc48e3ea88149
-  data.tar.gz: c22754e9df2eb3de27a43fc8ad7f2e9cef5f88c4
+  metadata.gz: dd36735cbf74eaa7090b513cbfb2936a39c19a40
+  data.tar.gz: e2c34ca572d9f68716a395658f7c32b9c228ea7a
 SHA512:
-  metadata.gz: 90dd796b1c45ccf2e84c90cbaebfa00a0445420f821f2997f36879c8e5c4903497a6b82120ee58b0e29a31ea65690e183fb30ea97fb6aa124040078865a71c25
-  data.tar.gz: 02f9ef5f677afa75ab78c48050a78c987f9e4b016923e3931fc3be318298b1c9cd31d135a50643a2bf890f7533953466ea6c06f78690c8fa7920b05928db41db
+  metadata.gz: 4c44c82fb498a857e87d02bf5c02cef37836f39ce2727d7091cc07faa1247c3b3eec93a0e3c85fb2efd4e75f9638cc35629f4a71ad594d343f44bea9d9b103fe
+  data.tar.gz: 69534b7d9f0604835e1e3169a0b2ff038e5e8d6aef21737558ae74fd4a29d0e07eb78acbeb3ad404735114820df2e1ec37c41d28243140e472c912b25603a96e

data/lib/cc/kafka/producer.rb

@@ -4,28 +4,69 @@ require "poseidon"
 module CC
   module Kafka
     class Producer
-      def initialize(client_id, url)
-        @client_id = client_id
+      SCHEMES = [
+        HTTP = "http",
+        KAFKA = "kafka",
+      ]
+
+      HTTPError = Class.new(StandardError)
+      HTTP_TIMEOUT= 60 # seconds
+
+      def initialize(url, client_id = nil)
         @url = url
+        @client_id = client_id
       end
 
       def send_message(data, key = nil)
         serialized = BSON.serialize(data).to_s
-        message = Poseidon::MessageToSend.new(topic, serialized, key)
-
-        producer.send_messages([message])
-      rescue
-        close
 
-        raise
+        if http?
+          send_http(serialized, key)
+        else
+          send_poseidon(serialized, key)
+        end
       end
 
       def close
-        producer.close
+        unless http?
+          producer.close
+        end
       end
 
       private
 
+      def http?
+        uri.scheme == HTTP
+      end
+
+      def send_http(serialized, key)
+        data = {
+          "topic" => topic,
+          "message" => serialized,
+        }
+        data["key"] = key if key
+
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.open_timeout = HTTP_TIMEOUT
+        http.read_timeout = HTTP_TIMEOUT
+        request = Net::HTTP::Post.new("/message")
+        request.set_form_data(data)
+
+        response = http.request(request)
+
+        unless response.is_a?(Net::HTTPSuccess)
+          raise HTTPError, "request not successful: #{response.inspect}"
+        end
+      end
+
+      def send_poseidon(serialized, key)
+        message = Poseidon::MessageToSend.new(topic, serialized, key)
+        producer.send_messages([message])
+      rescue
+        producer.close
+        raise
+      end
+
       def producer
         @producer ||= Poseidon::Producer.new(
           broker,

data/lib/cc/kafka/version.rb

@@ -1,5 +1,5 @@
 module CC
   module Kafka
-    VERSION = "0.0.2"
+    VERSION = "0.1.0"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: codeclimate-kafka
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
 platform: ruby
 authors:
 - Code Climate
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-29 00:00:00.000000000 Z
+date: 2015-07-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bson