From 8d4756a0323a8f97f1acfb43eb45939f5e3bfe65 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mr=C2=B7x?= <as2252258@163.com>
Date: Fri, 30 Oct 2020 11:33:37 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E5=90=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Kafka/Producer.php | 30 ++++++++++++++++--------------
 1 file changed, 16 insertions(+), 14 deletions(-)

diff --git a/Kafka/Producer.php b/Kafka/Producer.php
index 8ecb3a43..41e6cfea 100644
--- a/Kafka/Producer.php
+++ b/Kafka/Producer.php
@@ -8,6 +8,7 @@ use RdKafka\TopicConf;
 use ReflectionException;
 use Snowflake\Abstracts\Component;
 use Snowflake\Event;
+use Snowflake\Exception\ComponentException;
 use Snowflake\Exception\NotFindClassException;
 use Snowflake\Snowflake;
 
@@ -56,17 +57,6 @@ class Producer extends Component
 	}
 
 
-	/**
-	 * @param bool $value
-	 * @return $this
-	 */
-	public function setAck(bool $value)
-	{
-//		$this->topicConf->set('request.required.acks', (string)$value);
-		return $this;
-	}
-
-
 	/**
 	 * @param $servers
 	 * @return Producer
@@ -81,10 +71,12 @@ class Producer extends Component
 	/**
 	 * @param $message
 	 * @param null $key
-	 * @param int $timeout
-	 * @throws
+	 * @param bool $isAck
+	 * @throws NotFindClassException
+	 * @throws ReflectionException
+	 * @throws ComponentException
 	 */
-	public function delivery($message, $key = null, $timeout = 10)
+	public function delivery($message, $key = null, $isAck = false)
 	{
 		if (!$this->conf || !$this->topicConf) {
 			throw new \Exception('Error. Please set kafka conf.');
@@ -99,6 +91,16 @@ class Producer extends Component
 		if ($this->producer === null) {
 			$this->producer = Snowflake::createObject(\RdKafka\Producer::class, [$this->conf]);
 		}
+
+		if ($isAck) {
+			if ($this->producer->getOutQLen()>0) {
+				$this->flush();
+			}
+			$this->topicConf->set('request.required.acks', '1');
+		} else {
+			$this->topicConf->set('request.required.acks', '0');
+		}
+
 		$topic = $this->producer->newTopic($this->_topic, $this->topicConf);
 		$topic->produce(RD_KAFKA_PARTITION_UA, 0, $message, $key);
 		$this->producer->poll(0);