skonto commented on a change in pull request #19096: [SPARK-21869][SS] A cached
Kafka producer should not be closed if any task is using it - adds inuse
tracking.
URL: https://github.com/apache/spark/pull/19096#discussion_r263320525
##########
File path:
external/kafka-0-10-sql/src/main/scala/org/apache/spark/sql/kafka010/CachedKafkaProducer.scala
##########
@@ -29,51 +30,76 @@ import scala.util.control.NonFatal
import org.apache.spark.SparkEnv
import org.apache.spark.internal.Logging
-private[kafka010] object CachedKafkaProducer extends Logging {
+private[kafka010] case class CachedKafkaProducer(id: String, inUseCount:
AtomicInteger,
+ kafkaProducer: KafkaProducer[Array[Byte], Array[Byte]]) extends Logging {
+ private var closed: Boolean = false
+ private def close(): Unit = this.synchronized {
+ if (!closed) {
+ closed = true
+ kafkaProducer.close()
+ logInfo(s"Closed kafka producer: $kafkaProducer")
+ }
+ }
+
+ def flush(): Unit = {
Review comment:
do we need the wrapper?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
