lib/fluent/plugin/kinesis_helper/client.rb

#
#  Copyright 2014-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
#  Licensed under the Amazon Software License (the "License").
#  You may not use this file except in compliance with the License.
#  A copy of the License is located at
#
#  http://aws.amazon.com/asl/
#
#  or in the "license" file accompanying this file. This file is distributed
#  on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
#  express or implied. See the License for the specific language governing
#  permissions and limitations under the License.

module Fluent
  module KinesisHelper
    module Client
      def client
        @client ||= client_class.new(client_options)
      end

      private

      def client_class
        case request_type
        when :streams
          Aws::Kinesis::Client
        when :firehose
          Aws::Firehose::Client
        when :producer
          KinesisProducer::Library
        end
      end
    end
  end
end