feat: add silent option

galax-io · May 9, 2024 · be313b8 · be313b8
1 parent f130728
commit be313b8
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/src/main/scala/org/galaxio/gatling/kafka/actions/KafkaRequestReplyAction.scala b/src/main/scala/org/galaxio/gatling/kafka/actions/KafkaRequestReplyAction.scala
@@ -99,7 +99,7 @@ class KafkaRequestReplyAction[K: ClassTag, V: ClassTag](
             session,
             next,
             requestNameString,
-            attributes.silent.getOrElse(false)
+            attributes.silent.getOrElse(false),
           )
       },
       e => {

diff --git a/src/main/scala/org/galaxio/gatling/kafka/client/KafkaMessageTracker.scala b/src/main/scala/org/galaxio/gatling/kafka/client/KafkaMessageTracker.scala
@@ -16,7 +16,7 @@ class KafkaMessageTracker(actor: ActorRef) {
       session: Session,
       next: Action,
       requestName: String,
-      silentRequest: Boolean
+      silentRequest: Boolean,
   ): Unit =
     actor ! MessagePublished(
       matchId,
@@ -26,6 +26,6 @@ class KafkaMessageTracker(actor: ActorRef) {
       session,
       next,
       requestName,
-      silentRequest
+      silentRequest,
     )
 }
diff --git a/src/main/scala/org/galaxio/gatling/kafka/client/KafkaMessageTrackerActor.scala b/src/main/scala/org/galaxio/gatling/kafka/client/KafkaMessageTrackerActor.scala
@@ -28,7 +28,7 @@ object KafkaMessageTrackerActor {
       session: Session,
       next: Action,
       requestName: String,
-      silentRequest: Boolean
+      silentRequest: Boolean,
   )
 
   case class MessageConsumed(
@@ -71,9 +71,9 @@ class KafkaMessageTrackerActor(statsEngine: StatsEngine, clock: Clock) extends A
       requestName: String,
       responseCode: Option[String],
       message: Option[String],
-      silentRequest: Boolean
+      silentRequest: Boolean,
   ): Unit = {
-    if(! silentRequest) {
+    if (!silentRequest) {
       statsEngine.logResponse(
         session.scenario,
         session.groups,
@@ -112,7 +112,7 @@ class KafkaMessageTrackerActor(statsEngine: StatsEngine, clock: Clock) extends A
           requestName,
           message.responseCode,
           Some(errorMessage),
-          silentRequest
+          silentRequest,
         )
       case _                           =>
         executeNext(newSession, sent, received, OK, next, requestName, None, None, silentRequest)
@@ -159,7 +159,7 @@ class KafkaMessageTrackerActor(statsEngine: StatsEngine, clock: Clock) extends A
           requestName,
           None,
           Some(s"Reply timeout after $receivedTimeout ms"),
-          silentRequest
+          silentRequest,
         )
       }
       timedOutMessages.clear()