feat: add silent option

galax-io · May 22, 2024 · 9fcbdb6 · 9fcbdb6
1 parent 46ca837
commit 9fcbdb6
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 13 deletions.
diff --git a/src/main/scala/org/galaxio/gatling/kafka/actions/KafkaRequestActionBuilder.scala b/src/main/scala/org/galaxio/gatling/kafka/actions/KafkaRequestActionBuilder.scala
@@ -1,6 +1,5 @@
 package org.galaxio.gatling.kafka.actions
 
-import com.softwaremill.quicklens._
 import io.gatling.core.action.Action
 import io.gatling.core.action.builder.ActionBuilder
 import io.gatling.core.structure.ScenarioContext

diff --git a/src/main/scala/org/galaxio/gatling/kafka/actions/KafkaRequestReplyActionBuilder.scala b/src/main/scala/org/galaxio/gatling/kafka/actions/KafkaRequestReplyActionBuilder.scala
@@ -7,7 +7,7 @@ import io.gatling.core.structure.ScenarioContext
 import io.gatling.core.util.NameGen
 import org.galaxio.gatling.kafka.KafkaCheck
 import org.galaxio.gatling.kafka.protocol.KafkaProtocol
-import org.galaxio.gatling.kafka.request.builder.{KafkaRequestReplyAttributes, RequestBuilder}
+import org.galaxio.gatling.kafka.request.builder.KafkaRequestReplyAttributes
 
 import scala.reflect.ClassTag
 

diff --git a/src/main/scala/org/galaxio/gatling/kafka/protocol/KafkaProtocolBuilderNew.scala b/src/main/scala/org/galaxio/gatling/kafka/protocol/KafkaProtocolBuilderNew.scala
@@ -1,10 +1,9 @@
 package org.galaxio.gatling.kafka.protocol
 
-import io.gatling.core.session.Expression
 import org.apache.kafka.clients.producer.ProducerConfig
 import org.apache.kafka.common.serialization.Serdes
 import org.apache.kafka.streams.StreamsConfig
-import KafkaProtocol._
+import org.galaxio.gatling.kafka.protocol.KafkaProtocol._
 import org.galaxio.gatling.kafka.request.KafkaProtocolMessage
 
 import scala.concurrent.duration.{DurationInt, FiniteDuration}

diff --git a/src/main/scala/org/galaxio/gatling/kafka/request/builder/KafkaRequestBuilderBase.scala b/src/main/scala/org/galaxio/gatling/kafka/request/builder/KafkaRequestBuilderBase.scala
@@ -15,19 +15,17 @@ case class KafkaRequestBuilderBase(requestName: Expression[String]) {
       payload: Expression[V],
       headers: Expression[Headers] = List.empty[Header],
       silent: Boolean = false,
-  )(implicit
-      sender: Sender[K, V],
-  ): RequestBuilder[K, V] = {
-    if (key == null)
-      sender.send(requestName, None, payload, Some(headers), Some(silent))
-    else
-      sender.send(requestName, Some(key), payload, Some(headers), Some(silent))
-  }
+  )(implicit sender: Sender[K, V]): RequestBuilder[K, V] = if (key == null)
+    sender.send(requestName, None, payload, Some(headers), Some(silent))
+  else
+    sender.send(requestName, Some(key), payload, Some(headers), Some(silent))
 
   def send[V](payload: Expression[V])(implicit
       sender: Sender[Nothing, V],
+      headers: Expression[Headers] = List.empty[Header],
+      silent: Boolean = false,
   ): RequestBuilder[_, V] =
-    sender.send(requestName = requestName, key = None, payload = payload, headers = None, silent = None)
+    sender.send(requestName = requestName, key = None, payload = payload, headers = Some(headers), silent = Some(silent))
 
   def requestReply: ReqRepBase.type = ReqRepBase