Sha256: 6bc2f280944ab5df82ea67f434170ed63cb1ade5b16b0fd7c0289c39c6912bac

Contents?: true

Size: 1.97 KB

Versions: 1

Compression:

Stored size: 1.97 KB

Contents

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)    
    Unirest.post "#{@uri.to_s}?timeout=10&api-version=2014-01", 
                        headers:{ 'Content-Type' => 'application/atom+xml;type=entry;charset=utf-8',
                            'Authorization' => token
                        }, 
                        parameters: payload.to_s {|response|
    response.code 
}   
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version Path
sk-fluent-plugin-azureeventhubs-0.0.6 lib/fluent/plugin/azureeventhubs/http.rb