Skip to content

Commit 221faed

Browse files
author
aldrick.castro
committed
Pulled out the section needed to be tested
1 parent f6570d0 commit 221faed

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

postgres/datadog_checks/postgres/explain_parameterized_queries.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -142,26 +142,27 @@ def _get_number_of_parameters_for_prepared_statement(self, dbname, query_signatu
142142
dbname, PARAM_TYPES_COUNT_QUERY.format(query_signature=query_signature)
143143
)
144144
return rows[0][0] if rows else 0
145-
145+
146146
@tracked_method(agent_check_getter=agent_check_getter)
147-
def _explain_prepared_statement(self, dbname, statement, obfuscated_statement, query_signature):
148-
num_params = self._get_number_of_parameters_for_prepared_statement(dbname, query_signature)
149-
147+
def _generate_prepared_statement_query(self, dbname: str, query_signature: str) -> str:
150148
parameters = ""
149+
num_params = self._get_number_of_parameters_for_prepared_statement(dbname, query_signature)
151150

152151
if num_params > 0:
153152
null_parameters = ','.join('null' for _ in range(num_params))
154153
parameters = f"({null_parameters})"
155154

156-
execute_prepared_statement_query = EXECUTE_PREPARED_STATEMENT_QUERY.format(
157-
prepared_statement=query_signature, parameters=parameters
158-
)
155+
return EXECUTE_PREPARED_STATEMENT_QUERY.format(prepared_statement=query_signature, parameters=parameters)
156+
157+
@tracked_method(agent_check_getter=agent_check_getter)
158+
def _explain_prepared_statement(self, dbname, statement, obfuscated_statement, query_signature):
159+
prepared_statement_query = self._generate_prepared_statement_query(dbname, query_signature)
159160
try:
160161
return self._execute_query_and_fetch_rows(
161162
dbname,
162163
EXPLAIN_QUERY.format(
163164
explain_function=self._explain_function,
164-
statement=execute_prepared_statement_query,
165+
statement=prepared_statement_query,
165166
),
166167
)
167168
except Exception as e:

0 commit comments

Comments
 (0)