Class: AzureEventHubsHttpSender

Inherits:
Object
  • Object
show all
Defined in:
lib/fluent/plugin/azureeventhubs/http.rb

Instance Method Summary collapse

Constructor Details

#initialize(connection_string, hub_name, expiry = 3600, proxy_addr = '', proxy_port = 3128, open_timeout = 60, read_timeout = 60) ⇒ AzureEventHubsHttpSender

Returns a new instance of AzureEventHubsHttpSender.



3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
# File 'lib/fluent/plugin/azureeventhubs/http.rb', line 3

def initialize(connection_string, hub_name, expiry=3600,proxy_addr='',proxy_port=3128,open_timeout=60,read_timeout=60)
  require 'openssl'
  require 'base64'
  require 'rest-client'
  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

Instance Method Details

#generate_sas_token(uri) ⇒ Object 



34
35
36
37
38
39
40
41
42
 
# File 'lib/fluent/plugin/azureeventhubs/http.rb', line 34

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

#send(payload) ⇒ Object 



45
46
47
 
# File 'lib/fluent/plugin/azureeventhubs/http.rb', line 45

def send(payload)
  send_w_properties(payload, nil)
end

#send_w_properties(payload, properties) ⇒ Object 



49
50
51
52
53
54
55
 
# File 'lib/fluent/plugin/azureeventhubs/http.rb', line 49

def send_w_properties(payload, properties)
  token = generate_sas_token(@uri.to_s)
  RestClient.post("#{@uri.to_s}?timeout=10&api-version=2014-01", payload.to_json,
                      { 'Content-Type' => 'application/atom+xml;type=entry;charset=utf-8',
                          'Authorization' => token
                      })
end