instrument tracing without capturing inputs or outputs

Dwolla · Dec 4, 2023 · b6431f0 · b6431f0
1 parent 18d3949
commit b6431f0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/draining/src/main/scala/com/dwolla/autoscaling/ecs/draining/TerminationEventHandler.scala b/draining/src/main/scala/com/dwolla/autoscaling/ecs/draining/TerminationEventHandler.scala
@@ -32,10 +32,10 @@ class TerminationEventHandler extends IOLambda[SnsEvent, INothing] {
       given Random[IO] <- Random.scalaUtilRandom[IO].toResource
       entryPoint <- XRay.entryPoint[IO]()
       awsEnv <- AwsEnvironment.default(client, AwsRegion.US_WEST_2)
-      ecs <- AwsClient(ECS, awsEnv).map(_.traceWithInputs).map(EcsAlg(_))
+      ecs <- AwsClient(ECS, awsEnv).map(_.instrumentAndTrace).map(EcsAlg(_))
       autoscalingClient <- AwsClient(AutoScaling, awsEnv)
-      sns <- AwsClient(SNS, awsEnv).map(SnsAlg[IO](_).traceWithInputs)
-      autoscaling = AutoScalingAlg[IO](autoscalingClient, sns).traceWithInputs
+      sns <- AwsClient(SNS, awsEnv).map(SnsAlg[IO](_).instrumentAndTrace)
+      autoscaling = AutoScalingAlg[IO](autoscalingClient, sns).instrumentAndTrace
       bridgeFunction = TerminationEventBridge(ecs, autoscaling)
       given Logger[IO] <- LoggerFactory[IO].create.toResource
     } yield LifecycleHookHandler(entryPoint, "TerminationEventHandler")(bridgeFunction)

diff --git a/...k/src/main/scala/com/dwolla/autoscaling/ecs/registrator/ScaleOutPendingEventHandler.scala b/...k/src/main/scala/com/dwolla/autoscaling/ecs/registrator/ScaleOutPendingEventHandler.scala
@@ -36,7 +36,7 @@ class ScaleOutPendingEventHandler extends IOLambda[SnsEvent, INothing] {
       given Random[IO] <- Random.scalaUtilRandom[IO].toResource
       entryPoint <- XRay.entryPoint[IO]()
       awsEnv <- AwsEnvironment.default(client, AwsRegion.US_WEST_2)
-      ecs <- AwsClient(ECS, awsEnv).map(_.traceWithInputs).map(EcsAlg(_))
+      ecs <- AwsClient(ECS, awsEnv).map(_.instrumentAndTrace).map(EcsAlg(_))
       autoscalingClient <- AwsClient(AutoScaling, awsEnv)
       sns <- AwsClient(SNS, awsEnv).map(SnsAlg[IO](_))
       ec2Client <- AwsClient(EC2, awsEnv).map(Ec2Alg[IO](_))