Contents

require 'logstasher/device'
require 'redis'

module LogStasher
  module Device
    class Redis
      include ::LogStasher::Device

      attr_reader :options, :redis

      def initialize(options = {})
        @options = default_options.merge(stringify_keys(options))

        validate_options
        configure_redis
      end

      def data_type
        options['data_type']
      end

      def key
        options['key']
      end

      def redis_options
        unless @redis_options
          default_keys = default_options.keys
          @redis_options = options.select { |k, v| !default_keys.include?(k) }
        end

        @redis_options
      end

      def write(log)
        case data_type
        when 'list'
          redis.rpush(key, log)
        when 'channel'
          redis.publish(key, log)
        else
          fail "Unknown data type #{data_type}"
        end
      end

      def close
        redis.quit
      end

      private

      def configure_redis
        @redis = ::Redis.new(redis_options)
      end

      def default_options
        { 'key' => 'logstash', 'data_type' => 'list' }
      end

      def validate_options
        unless ['list', 'channel'].include?(options['data_type'])
          fail 'Expected data_type to be either "list" or "channel"'
        end
      end
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
md-logstasher-1.0.5	lib/logstasher/device/redis.rb
md-logstasher-1.0.4	lib/logstasher/device/redis.rb
md-logstasher-1.0.3	lib/logstasher/device/redis.rb
md-logstasher-1.0.2	lib/logstasher/device/redis.rb
md-logstasher-1.0.1	lib/logstasher/device/redis.rb
md-logstasher-1.0.0	lib/logstasher/device/redis.rb
md-logstasher-1.0.0.rc5	lib/logstasher/device/redis.rb
md-logstasher-1.0.0.rc4	lib/logstasher/device/redis.rb