takus/fluent-plugin-ec2-metadata

require 'fluent/plugin/output'
require_relative 'ec2_metadata'

module Fluent::Plugin
  class EC2MetadataOutput < Output
    include Fluent::EC2Metadata

    Fluent::Plugin.register_output('ec2_metadata', self)

    helpers :event_emitter

    config_param :output_tag, :string
    config_param :aws_key_id, :string, default: ENV['AWS_ACCESS_KEY_ID'], secret: true
    config_param :aws_sec_key, :string, default: ENV['AWS_SECRET_ACCESS_KEY'], secret: true
    config_param :metadata_refresh_seconds, :integer, default: 300
    config_param :imdsv2, :bool, default: false

    attr_reader :ec2_metadata

    def process(tag, es)
      tag_parts = tag.split('.')
      es.each { |time, record|
        new_tag, new_record = modify(@output_tag, record, tag, tag_parts)
        router.emit(new_tag, time, new_record)
      }
    rescue => e
      log.warn "ec2-metadata: #{e.class} #{e.message} #{e.backtrace.join(', ')}"
    end
  end
end