mirror of
https://github.com/elastic/logstash.git
synced 2025-04-25 07:07:54 -04:00
c6795731f1
Refactor: move PipelineConfig from Ruby to Java Reimplement the Ruby class PipelinceConfig in Java trying to keep the method signatures to limit the changes in client code, this is a step of other that intend to move all the configuration code in Java language. Having all that code in Java unlock some reasoning about how to better implement it and probably an improvement in performance during process startup. Moved also the spec into a JUnit and fixed here and there the failing tests Closes: #11824
127 lines
4 KiB
Ruby
127 lines
4 KiB
Ruby
# Licensed to Elasticsearch B.V. under one or more contributor
|
|
# license agreements. See the NOTICE file distributed with
|
|
# this work for additional information regarding copyright
|
|
# ownership. Elasticsearch B.V. licenses this file to you under
|
|
# the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License 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.
|
|
|
|
require "stud/task"
|
|
|
|
def silence_warnings
|
|
warn_level = $VERBOSE
|
|
$VERBOSE = nil
|
|
yield
|
|
ensure
|
|
$VERBOSE = warn_level
|
|
end
|
|
|
|
def clear_data_dir
|
|
if defined?(agent_settings)
|
|
data_path = agent_settings.get("path.data")
|
|
else
|
|
data_path = LogStash::SETTINGS.get("path.data")
|
|
end
|
|
|
|
Dir.foreach(data_path) do |f|
|
|
next if f == "." || f == ".." || f == ".gitkeep"
|
|
FileUtils.rm_rf(File.join(data_path, f))
|
|
end
|
|
end
|
|
|
|
def mock_settings(settings_values={})
|
|
settings = LogStash::SETTINGS.clone
|
|
|
|
settings_values.each do |key, value|
|
|
settings.set(key, value)
|
|
end
|
|
|
|
settings
|
|
end
|
|
|
|
def make_test_agent(settings=mock_settings)
|
|
sl = LogStash::Config::SourceLoader.new
|
|
sl.add_source(LogStash::Config::Source::Local.new(settings))
|
|
sl
|
|
|
|
::LogStash::Agent.new(settings, sl)
|
|
end
|
|
|
|
def mock_pipeline(pipeline_id, reloadable = true, config_hash = nil)
|
|
config_string = "input { stdin { id => '#{pipeline_id}' }}"
|
|
settings = mock_settings("pipeline.id" => pipeline_id.to_s,
|
|
"config.string" => config_string,
|
|
"config.reload.automatic" => reloadable)
|
|
pipeline_config = mock_pipeline_config(pipeline_id, config_string, settings)
|
|
LogStash::Pipeline.new(pipeline_config)
|
|
end
|
|
|
|
def mock_pipeline_from_string(config_string, settings = LogStash::SETTINGS, metric = nil)
|
|
pipeline_config = mock_pipeline_config(settings.get("pipeline.id"), config_string, settings)
|
|
LogStash::Pipeline.new(pipeline_config, metric)
|
|
end
|
|
|
|
def mock_java_pipeline_from_string(config_string, settings = LogStash::SETTINGS, metric = nil)
|
|
pipeline_config = mock_pipeline_config(settings.get("pipeline.id"), config_string, settings)
|
|
LogStash::JavaPipeline.new(pipeline_config, metric)
|
|
end
|
|
|
|
def mock_pipeline_config(pipeline_id, config_string = nil, settings = {})
|
|
config_string = "input { stdin { id => '#{pipeline_id}' }}" if config_string.nil?
|
|
|
|
# This is for older tests when we already have a config
|
|
unless settings.is_a?(LogStash::Settings)
|
|
settings.merge!({ "pipeline.id" => pipeline_id.to_s })
|
|
settings = mock_settings(settings)
|
|
end
|
|
|
|
config_part = org.logstash.common.SourceWithMetadata.new("config_string", "config_string", 0, 0, config_string)
|
|
|
|
org.logstash.config.ir.PipelineConfig.new(LogStash::Config::Source::Local, pipeline_id.to_sym, [config_part], settings)
|
|
end
|
|
|
|
def start_agent(agent)
|
|
agent_task = Stud::Task.new do
|
|
begin
|
|
agent.execute
|
|
rescue => e
|
|
raise "Start Agent exception: #{e}"
|
|
end
|
|
end
|
|
|
|
wait(30).for { agent.running? }.to be(true)
|
|
agent_task
|
|
end
|
|
|
|
def temporary_file(content, file_name = Time.now.to_i.to_s, directory = Stud::Temporary.pathname)
|
|
FileUtils.mkdir_p(directory)
|
|
target = ::File.join(directory, file_name)
|
|
|
|
File.open(target, "w+") do |f|
|
|
f.write(content)
|
|
end
|
|
target
|
|
end
|
|
|
|
RSpec::Matchers.define :ir_eql do |expected|
|
|
match do |actual|
|
|
next unless expected.java_kind_of?(org.logstash.config.ir.SourceComponent) && actual.java_kind_of?(org.logstash.config.ir.SourceComponent)
|
|
|
|
expected.sourceComponentEquals(actual)
|
|
end
|
|
|
|
failure_message do |actual|
|
|
"actual value \n#{actual.to_s}\nis not .sourceComponentEquals to the expected value: \n#{expected.to_s}\n"
|
|
end
|
|
end
|
|
|
|
SUPPORT_DIR = Pathname.new(::File.join(::File.dirname(__FILE__), "support"))
|