async-fluent-plugin-azureeventhubs 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 895d5c57e671be39e4ccd9be2def670d02f87b1171af01b41d926f4bfbb8f021
+  data.tar.gz: dea3b6593fd2716aa7d22d430a97ba4433c7e584b36e4b2820b44966b0232ce4
+SHA512:
+  metadata.gz: 88c23e1607e61424d62ffe3645ab7ece75d996f129942432b1b3da7e056cfab6964a1547d374dfb454360df8f986789e07511e4819c314c67c058d9769eb9571
+  data.tar.gz: 999590adf69222f41e63cb4f06d592a92853ed4f27ecfbe25817d2e6f4f1527c5fa44838b3efccb0b69f9bf7201e9e74a636be99a3e40c7280f2b393810974e1

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,23 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+Gem::Specification.new do |spec|
+  spec.name          = "async-fluent-plugin-azureeventhubs"
+  spec.version       = "0.0.1"
+  spec.authors       = ["Sandeep Kotha"]
+  spec.email         = ["sandeep.kotha@live.com"]
+  spec.summary       = "Fluentd output plugin for Azure Event Hubs"
+  spec.description   = "Fluentd output plugin for Azure Event Hubs"
+  spec.homepage      = "https://github.com/htgc/fluent-plugin-azureeventhubs"
+  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.7"
+  spec.add_development_dependency "rake", "~> 12.3"
+  spec.add_dependency "fluentd", [">= 0.14.15", "< 2"]
+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,63 @@
+module Fluent::Plugin
+
+  class AzureEventHubsOutputBuffered < Output
+    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-plugin-azureeventhubs
+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: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.3'
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.14.15
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.14.15
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
+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/htgc/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