Module: LogStash::Config::Mixin::DSL

Defined in:
lib/logstash/config/mixin.rb

Overview

def config_init

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#flags ⇒ Object

Returns the value of attribute flags.



130
131
132
 
# File 'lib/logstash/config/mixin.rb', line 130

def flags
  @flags
end

Instance Method Details

#config(name, opts = {}) ⇒ Object

Define a new configuration setting



154
155
156
157
158
159
160
161
162
163
164
165
 
# File 'lib/logstash/config/mixin.rb', line 154

def config(name, opts={})
  @config ||= Hash.new
  # TODO(sissel): verify 'name' is of type String, Symbol, or Regexp

  name = name.to_s if name.is_a?(Symbol)
  @config[name] = opts  # ok if this is empty

  if name.is_a?(String)
    define_method(name) { instance_variable_get("@#{name}") }
    define_method("#{name}=") { |v| instance_variable_set("@#{name}", v) }
  end
end

#config_name(name = nil) ⇒ Object

If name is given, set the name and return it. If no name given (nil), return the current name.



134
135
136
137
138
 
# File 'lib/logstash/config/mixin.rb', line 134

def config_name(name = nil)
  @config_name = name if !name.nil?
  LogStash::Config::Registry.registry[@config_name] = self
  return @config_name
end

#default(name, value) ⇒ Object

def config



167
168
169
170
 
# File 'lib/logstash/config/mixin.rb', line 167

def default(name, value)
  @defaults ||= {}
  @defaults[name.to_s] = value
end

#default?(name) ⇒ Boolean

Returns:

  • (Boolean) 


180
181
182
 
# File 'lib/logstash/config/mixin.rb', line 180

def default?(name)
  return @defaults && @defaults.include?(name)
end

#get_config ⇒ Object 



172
173
174
 
# File 'lib/logstash/config/mixin.rb', line 172

def get_config
  return @config
end

#get_default(name) ⇒ Object

def get_config



176
177
178
 
# File 'lib/logstash/config/mixin.rb', line 176

def get_default(name)
  return @defaults && @defaults[name]
end

#hash_or_array(value) ⇒ Object

def validate_value



498
499
500
501
502
503
 
# File 'lib/logstash/config/mixin.rb', line 498

def hash_or_array(value)
  if !value.is_a?(Hash)
    value = [*value] # coerce scalar to array if necessary
  end
  return value
end

#inherited(subclass) ⇒ Object

This is called whenever someone subclasses a class that has this mixin.



196
197
198
199
200
201
202
203
204
205
206
207
208
 
# File 'lib/logstash/config/mixin.rb', line 196

def inherited(subclass)
  # Copy our parent's config to a subclass.
  # This method is invoked whenever someone subclasses us, like:
  # class Foo < Bar ...
  subconfig = Hash.new
  if !@config.nil?
    @config.each do |key, val|
      subconfig[key] = val
    end
  end
  subclass.instance_variable_set("@config", subconfig)
  @@version_notice_given = false
end

#milestone(m = nil) ⇒ Object

Deprecated: Declare the version of the plugin inside the gemspec.



148
149
150
151
 
# File 'lib/logstash/config/mixin.rb', line 148

def milestone(m = nil)
  @logger = Cabin::Channel.get(LogStash)
  @logger.warn(I18n.t('logstash.plugin.deprecated_milestone', :plugin => config_name))
end

#options(opts) ⇒ Object 



184
185
186
187
188
189
190
191
192
193
 
# File 'lib/logstash/config/mixin.rb', line 184

def options(opts)
  # add any options from this class
  prefix = self.name.split("::").last.downcase
  @flags.each do |flag|
    flagpart = flag[:args].first.gsub(/^--/,"")
    # TODO(sissel): logger things here could help debugging.

    opts.on("--#{prefix}-#{flagpart}", *flag[:args][1..-1], &flag[:block])
  end
end

#plugin_status(status = nil) ⇒ Object

Deprecated: Declare the version of the plugin inside the gemspec.



142
143
144
 
# File 'lib/logstash/config/mixin.rb', line 142

def plugin_status(status = nil)
  milestone(status)
end

#print_version_notice ⇒ Object

def validate



225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
 
# File 'lib/logstash/config/mixin.rb', line 225

def print_version_notice
  return if @@version_notice_given

  begin
    plugin_version = LogStash::Util::PluginVersion.find_plugin_version!(@plugin_type, @config_name)

    if plugin_version < PLUGIN_VERSION_1_0_0
      if plugin_version < PLUGIN_VERSION_0_9_0
        @logger.info(I18n.t("logstash.plugin.version.0-1-x", 
                            :type => @plugin_type,
                            :name => @config_name,
                            :LOGSTASH_VERSION => LOGSTASH_VERSION))
      else
        @logger.info(I18n.t("logstash.plugin.version.0-9-x", 
                            :type => @plugin_type,
                            :name => @config_name,
                            :LOGSTASH_VERSION => LOGSTASH_VERSION))
      end
    end
  rescue LogStash::PluginNoVersionError
    # If we cannot find a version in the currently installed gems we
    # will display this message. This could happen in the test, if you 
    # create an anonymous class to test a plugin.
    @logger.warn(I18n.t("logstash.plugin.no_version",
                            :type => @plugin_type,
                            :name => @config_name,
                            :LOGSTASH_VERSION => LOGSTASH_VERSION))
  ensure 
    @@version_notice_given = true
  end
end

#validate(params) ⇒ Object

def inherited



210
211
212
213
214
215
216
217
218
219
220
221
222
223
 
# File 'lib/logstash/config/mixin.rb', line 210

def validate(params)
  @plugin_name = config_name
  @plugin_type = ancestors.find { |a| a.name =~ /::Base$/ }.config_name
  @logger = Cabin::Channel.get(LogStash)
  is_valid = true

  print_version_notice

  is_valid &&= validate_check_invalid_parameter_names(params)
  is_valid &&= validate_check_required_parameter_names(params)
  is_valid &&= validate_check_parameter_values(params)

  return is_valid
end

#validate_check_invalid_parameter_names(params) ⇒ Object 



257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
 
# File 'lib/logstash/config/mixin.rb', line 257

def validate_check_invalid_parameter_names(params)
  invalid_params = params.keys
  # Filter out parameters that match regexp keys.
  # These are defined in plugins like this:
  #   config /foo.*/ => ...
  @config.each_key do |config_key|
    if config_key.is_a?(Regexp)
      invalid_params.reject! { |k| k =~ config_key }
    elsif config_key.is_a?(String)
      invalid_params.reject! { |k| k == config_key }
    end
  end

  if invalid_params.size > 0
    invalid_params.each do |name|
      @logger.error("Unknown setting '#{name}' for #{@plugin_name}")
    end
    return false
  end # if invalid_params.size > 0
  return true
end

#validate_check_parameter_values(params) ⇒ Object 



299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
 
# File 'lib/logstash/config/mixin.rb', line 299

def validate_check_parameter_values(params)
  # Filter out parametrs that match regexp keys.
  # These are defined in plugins like this:
  #   config /foo.*/ => ... 
  is_valid = true

  params.each do |key, value|
    @config.keys.each do |config_key|
      next unless (config_key.is_a?(Regexp) && key =~ config_key) \
                  || (config_key.is_a?(String) && key == config_key)
      config_val = @config[config_key][:validate]
      #puts "  Key matches."
      success, result = validate_value(value, config_val)
      if success 
        # Accept coerced value if success
        # Used for converting values in the config to proper objects.
        params[key] = result if !result.nil?
      else
        @logger.error(I18n.t("logstash.agent.configuration.setting_invalid",
                             :plugin => @plugin_name, :type => @plugin_type,
                             :setting => key, :value => value.inspect,
                             :value_type => config_val,
                             :note => result))
      end
      #puts "Result: #{key} / #{result.inspect} / #{success}"
      is_valid &&= success

      break # done with this param key
    end # config.each
  end # params.each

  return is_valid
end

#validate_check_required_parameter_names(params) ⇒ Object

def validate_check_invalid_parameter_names



279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
 
# File 'lib/logstash/config/mixin.rb', line 279

def validate_check_required_parameter_names(params)
  is_valid = true

  @config.each do |config_key, config|
    next unless config[:required]

    if config_key.is_a?(Regexp)
      next if params.keys.select { |k| k =~ config_key }.length > 0
    elsif config_key.is_a?(String)
      next if params.keys.member?(config_key)
    end
    @logger.error(I18n.t("logstash.agent.configuration.setting_missing",
                         :setting => config_key, :plugin => @plugin_name,
                         :type => @plugin_type))
    is_valid = false
  end

  return is_valid
end

#validate_value(value, validator) ⇒ Object 



343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
 
# File 'lib/logstash/config/mixin.rb', line 343

def validate_value(value, validator)
  # Validator comes from the 'config' pieces of plugins.
  # They look like this
  #   config :mykey => lambda do |value| ... end
  # (see LogStash::Inputs::File for example)
  result = nil

  if validator.nil?
    return true
  elsif validator.is_a?(Array)
    value = [*value]
    if value.size > 1
      return false, "Expected one of #{validator.inspect}, got #{value.inspect}"
    end

    if !validator.include?(value.first)
      return false, "Expected one of #{validator.inspect}, got #{value.inspect}"
    end
    result = value.first
  elsif validator.is_a?(Symbol)
    # TODO(sissel): Factor this out into a coersion method?
    # TODO(sissel): Document this stuff.
    value = hash_or_array(value)

    case validator
      when :codec
        if value.first.is_a?(String)
          value = LogStash::Plugin.lookup("codec", value.first).new
          return true, value
        else
          value = value.first
          return true, value
        end
      when :hash
        if value.is_a?(Hash)
          return true, value
        end

        if value.size % 2 == 1
          return false, "This field must contain an even number of items, got #{value.size}"
        end

        # Convert the array the config parser produces into a hash.
        result = {}
        value.each_slice(2) do |key, value|
          entry = result[key]
          if entry.nil?
            result[key] = value
          else
            if entry.is_a?(Array)
              entry << value
            else
              result[key] = [entry, value]
            end
          end
        end
      when :array
        result = value
      when :string
        if value.size > 1 # only one value wanted
          return false, "Expected string, got #{value.inspect}"
        end
        result = value.first
      when :number
        if value.size > 1 # only one value wanted
          return false, "Expected number, got #{value.inspect} (type #{value.class})"
        end

        v = value.first
        case v
          when Numeric
            result = v
          when String
            if v.to_s.to_f.to_s != v.to_s \
               && v.to_s.to_i.to_s != v.to_s
              return false, "Expected number, got #{v.inspect} (type #{v})"
            end
            if v.include?(".")
              # decimal value, use float.
              result = v.to_f
            else
              result = v.to_i
            end
        end # case v
      when :boolean
        if value.size > 1 # only one value wanted
          return false, "Expected boolean, got #{value.inspect}"
        end

        bool_value = value.first
        if !!bool_value == bool_value
          # is_a does not work for booleans
          # we have Boolean and not a string
          result = bool_value
        else
          if bool_value !~ /^(true|false)$/
            return false, "Expected boolean 'true' or 'false', got #{bool_value.inspect}"
          end

          result = (bool_value == "true")
        end
      when :ipaddr
        if value.size > 1 # only one value wanted
          return false, "Expected IPaddr, got #{value.inspect}"
        end

        octets = value.split(".")
        if octets.length != 4
          return false, "Expected IPaddr, got #{value.inspect}"
        end
        octets.each do |o|
          if o.to_i < 0 or o.to_i > 255
            return false, "Expected IPaddr, got #{value.inspect}"
          end
        end
        result = value.first
      when :password
        if value.size > 1
          return false, "Expected password (one value), got #{value.size} values?"
        end

        result = value.first.is_a?(::LogStash::Util::Password) ? value.first : ::LogStash::Util::Password.new(value.first)
      when :path
        if value.size > 1 # Only 1 value wanted
          return false, "Expected path (one value), got #{value.size} values?"
        end

        # Paths must be absolute
        #if !Pathname.new(value.first).absolute?
          #return false, "Require absolute path, got relative path #{value.first}?"
        #end

        if !File.exists?(value.first) # Check if the file exists
          return false, "File does not exist or cannot be opened #{value.first}"
        end

        result = value.first
      when :bytes
        begin
          bytes = Integer(value.first) rescue nil
          result = bytes || Filesize.from(value.first).to_i
        rescue ArgumentError
          return false, "Unparseable filesize: #{value.first}. possible units (KiB, MiB, ...) e.g. '10 KiB'. doc reference: http://www.elastic.co/guide/en/logstash/current/configuration.html#bytes"
        end
      else
        return false, "Unknown validator symbol #{validator}"
    end # case validator
  else
    return false, "Unknown validator #{validator.class}"
  end

  # Return the validator for later use, like with type coercion.
  return true, result
end

#validator_find(key) ⇒ Object

def validate_check_parameter_values



333
334
335
336
337
338
339
340
341
 
# File 'lib/logstash/config/mixin.rb', line 333

def validator_find(key)
  @config.each do |config_key, config_val|
    if (config_key.is_a?(Regexp) && key =~ config_key) \
       || (config_key.is_a?(String) && key == config_key)
      return config_val
    end
  end # @config.each
  return nil
end