Skip to content

Commit 8577eb7

Browse files
authored
Add additional trace points for AWS Kinesis (#1569)
1 parent 4cb31b1 commit 8577eb7

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

newrelic/hooks/external_botocore.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1388,6 +1388,7 @@ def wrap_serialize_to_request(wrapped, instance, args, kwargs):
13881388
extract_agent_attrs=extract_kinesis_agent_attrs,
13891389
library="Kinesis",
13901390
),
1391+
("kinesis", "describe_account_settings"): aws_function_trace("describe_account_settings", library="Kinesis"),
13911392
("kinesis", "describe_limits"): aws_function_trace("describe_limits", library="Kinesis"),
13921393
("kinesis", "describe_stream"): aws_function_trace(
13931394
"describe_stream", extract_kinesis, extract_agent_attrs=extract_kinesis_agent_attrs, library="Kinesis"
@@ -1465,6 +1466,7 @@ def wrap_serialize_to_request(wrapped, instance, args, kwargs):
14651466
("kinesis", "untag_resource"): aws_function_trace(
14661467
"untag_resource", extract_kinesis, extract_agent_attrs=extract_kinesis_agent_attrs, library="Kinesis"
14671468
),
1469+
("kinesis", "update_account_settings"): aws_function_trace("update_account_settings", library="Kinesis"),
14681470
("kinesis", "update_max_record_size"): aws_function_trace(
14691471
"update_max_record_size", extract_kinesis, extract_agent_attrs=extract_kinesis_agent_attrs, library="Kinesis"
14701472
),
@@ -1474,6 +1476,12 @@ def wrap_serialize_to_request(wrapped, instance, args, kwargs):
14741476
("kinesis", "update_stream_mode"): aws_function_trace(
14751477
"update_stream_mode", extract_kinesis, extract_agent_attrs=extract_kinesis_agent_attrs, library="Kinesis"
14761478
),
1479+
("kinesis", "update_stream_warm_throughput"): aws_function_trace(
1480+
"update_stream_warm_throughput",
1481+
extract_kinesis,
1482+
extract_agent_attrs=extract_kinesis_agent_attrs,
1483+
library="Kinesis",
1484+
),
14771485
("kinesis", "put_record"): aws_message_trace(
14781486
"Produce", "Stream", extract_kinesis, extract_agent_attrs=extract_kinesis_agent_attrs, library="Kinesis"
14791487
),

0 commit comments

Comments
 (0)