async-fluent-azure-eventhubs-plugin 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 6ef121f2e386234907016780997478329045c40e756f62f4b5c730d905ba1052
+  data.tar.gz: a713a0981edfee6961bb20a8718fa0210ec55d3448185fd70c63cd683b74698e
+SHA512:
+  metadata.gz: 9c38dfe625801d2f3b8809d820d2a50eac8dd8a50ff7561d38d84196c4da9c158e1675fa0f47e7679e1526f96f16b1324e4f9c2e4c3c17588e8035a2a81d38ae
+  data.tar.gz: a8e5093ef6261b49c37a70d969ee3f59f32c148b5bab1cea09d686e9a6d957387bb4d20bbe3445e22d2dab70cc79346fcac0f96a2ba69541824b2ce8946d592b

data/.gitignore

@@ -0,0 +1,4 @@
+/.idea/vcs.xml
+/.idea/async-fluent-plugin-azureeventhubs.iml
+/.idea/modules.xml
+/.idea/workspace.xml

data/.travis.yml

@@ -0,0 +1,28 @@
+language: ruby
+
+rvm:
+  - 2.1
+  - 2.2
+  - 2.3.1
+  - 2.4.1
+  - 2.5.0
+  - ruby-head
+
+before_install:
+  - gem update --system
+
+script:
+  - bundle exec rake test
+
+sudo: false
+
+matrix:
+  allow_failures:
+    - rvm: ruby-head
+deploy:
+  provider: rubygems
+  api_key: "bc5cc687adff822e861d4df96c0986e4"
+  gemspec: aync-fluent-plugin-azureeventhubs.gemspec
+  on:
+    tags: true
+    all_branches: true

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2018 sio2k
+
+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 @@
+# async-fluent-plugin-azureeventhubs-
+[![Build Status](https://travis-ci.com/sio2k/async-fluent-plugin-azureeventhubs-.svg?branch=master)](https://travis-ci.com/sio2k/async-fluent-plugin-azureeventhubs-)
+Inspired by https://github.com/htgc/fluent-plugin-azureeventhubs

data/Rakefile

@@ -0,0 +1,8 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+require 'rake/testtask'
+
+
+
+task :default => [:build]

data/aync-fluent-plugin-azureeventhubs.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Sandeep Kotha"]
+  gem.email         = ["sandeep.kotha@live.com"]
+  gem.summary       = "Fluentd output plugin for Azure Event Hubs"
+  gem.description   = "Fluentd output plugin for Azure Event Hubs"
+  gem.homepage      = "https://github.com/sio2k/async-fluent-plugin-azureeventhubs-"
+  gem.license       = "MIT"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+  gem.name          = "async-fluent-azure-eventhubs-plugin"
+  gem.version       = "0.0.1"
+
+  gem.add_development_dependency "rake", ">= 0.9.2"
+  gem.add_dependency "fluentd", [">= 0.10.58", "< 2"]
+  gem.add_development_dependency "test-unit", ">= 3.0.8"
+end

data/lib/fluent/plugin/azureeventhubs/http.rb

@@ -0,0 +1,67 @@
+
+class AzureEventHubsHttpSender
+  def initialize(connection_string, hub_name, expiry=3600,proxy_addr='',proxy_port=3128,open_timeout=60,read_timeout=60)
+    require 'openssl'
+    require 'base64'
+    require 'unirest'
+    require 'json'
+    require 'cgi'
+    require 'time'
+    @connection_string = connection_string
+    @hub_name = hub_name
+    @expiry_interval = expiry
+    @proxy_addr = proxy_addr
+    @proxy_port = proxy_port
+    @open_timeout = open_timeout
+    @read_timeout = read_timeout
+
+    if @connection_string.count(';') != 2
+      raise "Connection String format is not correct"
+    end
+
+    @connection_string.split(';').each do |part|
+      if ( part.index('Endpoint') == 0 )
+        @endpoint = 'https' + part[11..-1]
+      elsif ( part.index('SharedAccessKeyName') == 0 )
+        @sas_key_name = part[20..-1]
+      elsif ( part.index('SharedAccessKey') == 0 )
+        @sas_key_value = part[16..-1]
+      end
+    end
+    @uri = URI.parse("#{@endpoint}#{@hub_name}/messages")
+  end
+
+  def generate_sas_token(uri)
+    target_uri = CGI.escape(uri.downcase).downcase
+    expiry = Time.now.to_i + @expiry_interval
+    to_sign = "#{target_uri}\n#{expiry}";
+    signature = CGI.escape(Base64.encode64(OpenSSL::HMAC.digest(OpenSSL::Digest.new('sha256'), @sas_key_value, to_sign)).strip())
+
+    token = "SharedAccessSignature sr=#{target_uri}&sig=#{signature}&se=#{expiry}&skn=#{@sas_key_name}"
+    return token
+  end
+
+  private :generate_sas_token
+
+  def send(payload)
+    send_w_properties(payload, nil)
+  end
+
+  def send_w_properties(payload, properties)
+    token = generate_sas_token(@uri.to_s)
+    headers = {
+      'Content-Type' => 'application/atom+xml;type=entry;charset=utf-8',
+      'Authorization' => token
+    }
+    if not properties.nil?
+      headers = headers.merge(properties)
+    end
+    Unirest.post @uri.request_uri, 
+                        headers:{ 'Content-Type' => 'application/atom+xml;type=entry;charset=utf-8',
+                            'Authorization' => token
+                        }, 
+                        parameters:payload {|response|
+    response.code 
+}   
+  end
+end

data/lib/fluent/plugin/out_azureeventhubs_buffered.rb

@@ -0,0 +1,61 @@
+class Fluent::AzureEventHubsOutputBuffered < Fluent::BufferedOutput
+    Fluent::Plugin.register_output('azure_eventhub_buffered', self)
+
+    helpers :compat_parameters, :inject
+
+    DEFAULT_BUFFER_TYPE = "memory"
+
+    config_param :connection_string, :string
+    config_param :hub_name, :string
+    config_param :include_tag, :bool, :default => false
+    config_param :include_time, :bool, :default => false
+    config_param :tag_time_name, :string, :default => 'time'
+    config_param :expiry_interval, :integer, :default => 3600 # 60min
+    config_param :type, :string, :default => 'https' # https / amqps (Not Implemented)
+    config_param :proxy_addr, :string, :default => ''
+    config_param :proxy_port, :integer,:default => 3128
+    config_param :open_timeout, :integer,:default => 60
+    config_param :read_timeout, :integer,:default => 60
+    config_param :message_properties, :hash, :default => nil
+
+    config_section :buffer do
+      config_set_default :@type, DEFAULT_BUFFER_TYPE
+      config_set_default :chunk_keys, ['tag']
+    end
+
+    def configure(conf)
+      compat_parameters_convert(conf, :buffer, :inject)
+      super
+      case @type
+      when 'amqps'
+        raise NotImplementedError
+      else
+        require_relative 'azureeventhubs/http'
+        @sender = AzureEventHubsHttpSender.new(@connection_string, @hub_name, @expiry_interval,@proxy_addr,@proxy_port,@open_timeout,@read_timeout)
+      end
+      raise Fluent::ConfigError, "'tag' in chunk_keys is required." if not @chunk_key_tag
+    end
+
+    def format(tag, time, record)
+      record = inject_values_to_record(tag, time, record)
+      [tag, time, record].to_msgpack
+    end
+
+    def formatted_to_msgpack_binary?
+      true
+    end
+
+    def write(chunk)
+      chunk.msgpack_each { |tag, time, record|
+        p record.to_s
+        if @include_tag
+          record['tag'] = tag
+        end
+        if @include_time
+          record[@tag_time_name] = time
+        end
+        @sender.send_w_properties(record, @message_properties)
+      }
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,26 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'fluent/test'
+unless ENV.has_key?('VERBOSE')
+  nulllogger = Object.new
+  nulllogger.instance_eval {|obj|
+    def method_missing(method, *args)
+    end
+  }
+  $log = nulllogger
+end
+
+
+class Test::Unit::TestCase
+end

data/test/plugin/test_out_azure_event_hub.rb

@@ -0,0 +1,28 @@
+require 'helper'
+require 'fluent/output'
+
+class AzureEventHubTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+  end
+
+  BASE_CONFIG = %[
+    type azure_eventhub_buffered
+  ]
+
+  CONFIG = BASE_CONFIG + %[
+    connection_string sb://sample
+  ]
+
+  def create_driver(conf = CONFIG, tag='test')
+    "sample"
+  end
+
+
+
+  def test_format
+    d = create_driver
+    assert_equal 'sample', d
+  end
+
+end

metadata

@@ -0,0 +1,105 @@
+--- !ruby/object:Gem::Specification
+name: async-fluent-azure-eventhubs-plugin
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Sandeep Kotha
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-10-31 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.9.2
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.10.58
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.10.58
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.8
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.8
+description: Fluentd output plugin for Azure Event Hubs
+email:
+- sandeep.kotha@live.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- aync-fluent-plugin-azureeventhubs.gemspec
+- lib/fluent/plugin/azureeventhubs/http.rb
+- lib/fluent/plugin/out_azureeventhubs_buffered.rb
+- test/helper.rb
+- test/plugin/test_out_azure_event_hub.rb
+homepage: https://github.com/sio2k/async-fluent-plugin-azureeventhubs-
+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.7.7
+signing_key: 
+specification_version: 4
+summary: Fluentd output plugin for Azure Event Hubs
+test_files:
+- test/helper.rb
+- test/plugin/test_out_azure_event_hub.rb