Class: Fluent::Plugin::KinesisStreamsOutput

Inherits:
KinesisOutput
  • Object
show all
Includes:
Fluent::Plugin::KinesisHelper::API::BatchRequest
Defined in:
lib/fluent/plugin/out_kinesis_streams.rb

Constant Summary collapse

RequestType = 
:streams
BatchRequestLimitCount = 
500
BatchRequestLimitSize = 
5 * 1024 * 1024

Constants included from Fluent::Plugin::KinesisHelper::API

Fluent::Plugin::KinesisHelper::API::MaxRecordSize

Instance Method Summary collapse

Methods included from Fluent::Plugin::KinesisHelper::API::BatchRequest

included, #size_of_values

Methods inherited from KinesisOutput

#multi_workers_ready?

Methods included from Fluent::Plugin::KinesisHelper::API

included

Methods included from Fluent::Plugin::KinesisHelper::Client

#client, included

Instance Method Details

#configure(conf) ⇒ Object 



30
31
32
33
 
# File 'lib/fluent/plugin/out_kinesis_streams.rb', line 30

def configure(conf)
  super
  @key_formatter = key_formatter_create
end

#format(tag, time, record) ⇒ Object 



35
36
37
38
39
40
41
 
# File 'lib/fluent/plugin/out_kinesis_streams.rb', line 35

def format(tag, time, record)
  format_for_api do
    data = @data_formatter.call(tag, time, record)
    key = @key_formatter.call(record)
    [data, key]
  end
end

#write(chunk) ⇒ Object 



43
44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/fluent/plugin/out_kinesis_streams.rb', line 43

def write(chunk)
  stream_name = extract_placeholders(@stream_name, chunk)
  write_records_batch(chunk) do |batch|
    records = batch.map{|(data, partition_key)|
      { data: data, partition_key: partition_key }
    }
    client.put_records(
      stream_name: stream_name,
      records: records,
    )
  end
end