Skip to content

Commit 215fc61

Browse files
committed
Preflight check initialization optimization.
1 parent ddaa1a3 commit 215fc61

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

lib/logstash/filters/elastic_integration.rb

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -135,9 +135,6 @@ def register
135135
initialize_event_processor!
136136

137137
perform_preflight_check!
138-
if serverless?
139-
set_api_version_to_rest_client!
140-
end
141138
check_versions_alignment
142139
end # def register
143140

@@ -346,6 +343,12 @@ def extract_immutable_config
346343
def initialize_elasticsearch_rest_client!
347344
config = extract_immutable_config
348345
@elasticsearch_rest_client = _elasticsearch_rest_client(config)
346+
347+
if serverless?
348+
@elasticsearch_rest_client = _elasticsearch_rest_client(config) do |builder|
349+
builder.configureElasticApi { |elasticApi| elasticApi.setApiVersion(ELASTIC_API_VERSION) }
350+
end
351+
end
349352
end
350353

351354
def _elasticsearch_rest_client(config, &builder_interceptor)
@@ -372,21 +375,24 @@ def initialize_event_processor!
372375
end
373376

374377
def perform_preflight_check!
375-
java_import('co.elastic.logstash.filters.elasticintegration.PreflightCheck')
376-
@preflight_check = PreflightCheck.new(@elasticsearch_rest_client)
377378
connected_es_version_info
378379
check_user_privileges!
379380
check_es_cluster_license!
380381
rescue => e
381382
raise_config_error!(e.message)
382383
end
383384

385+
def preflight_check_instance
386+
java_import('co.elastic.logstash.filters.elasticintegration.PreflightCheck')
387+
@preflight_check |= PreflightCheck.new(@elasticsearch_rest_client)
388+
end
389+
384390
def connected_es_version_info
385-
@connected_es_version_info |= @preflight_check.getElasticsearchVersionInfo
391+
@connected_es_version_info |= preflight_check_instance.getElasticsearchVersionInfo
386392
end
387393

388394
def check_user_privileges!
389-
@preflight_check.checkUserPrivileges
395+
preflight_check_instance.checkUserPrivileges
390396
rescue => e
391397
security_error_message = "no handler found for uri [/_security/user/_has_privileges]"
392398
if e.message.include?(security_error_message)
@@ -409,7 +415,7 @@ def check_user_privileges!
409415
end
410416

411417
def check_es_cluster_license!
412-
@preflight_check.checkLicense
418+
preflight_check_instance.checkLicense
413419
rescue => e
414420
raise_config_error!(e.message)
415421
end

0 commit comments

Comments
 (0)