改名

2021-08-24 18:24:46 +08:00
parent 575f4b8520
commit 24c93edf2b
110 changed files with 5 additions and 6 deletions
@@ -0,0 +1,161 @@
+<?php
+
+namespace Kafka;
+
+use Exception;
+use Kiri\Abstracts\Config;
+use Kiri\Exception\ConfigException;
+use Kiri\Exception\NotFindClassException;
+use Kiri\Kiri;
+use RdKafka\Producer;
+use RdKafka\ProducerTopic;
+use ReflectionException;
+
+
+/**
+ *
+ */
+class KafkaClient
+{
+
+
+	private Configuration $conf;
+	private TopicConfig $topicConf;
+
+	private bool $isAck = true;
+
+	/**
+	 * Producer constructor.
+	 * @param string $topic
+	 * @param string $groupId
+	 * @throws ConfigException
+	 * @throws NotFindClassException
+	 * @throws ReflectionException
+	 */
+	public function __construct(public string $topic, public string $groupId)
+	{
+		$this->conf = di(Configuration::class);
+		$this->topicConf = di(TopicConfig::class);
+		$this->setConfig();
+	}
+
+
+	/**
+	 * @return TopicConfig
+	 */
+	public function getTopicConfig(): TopicConfig
+	{
+		return $this->topicConf;
+	}
+
+
+	/**
+	 * @return Configuration
+	 */
+	public function getConfiguration(): Configuration
+	{
+		return $this->conf;
+	}
+
+
+	/**
+	 * @throws ConfigException
+	 */
+	private function setConfig()
+	{
+		$config = Config::get('kafka.producers.' . $this->topic, null, true);
+		if (!isset($config['brokers'])) {
+			throw new ConfigException('Please configure relevant information.');
+		}
+		$this->conf->setMetadataBrokerList($config['brokers']);
+		$this->conf->setGroupId($this->groupId);
+		$this->conf->setClientId(current(swoole_get_local_ip()));
+		$this->conf->setErrorCb(function ($kafka, $err, $reason) {
+			logger()->error(sprintf("Kafka error: %s (reason: %s)", rd_kafka_err2str($err), $reason));
+		});
+	}
+
+
+	/**
+	 * @param string $key
+	 * @param string|array $params
+	 * @param bool $isAck
+	 * @throws Exception
+	 */
+	public function push(string $key, string|array $params, bool $isAck = false)
+	{
+		$this->sendMessage([$params], $key, $isAck);
+	}
+
+
+	/**
+	 * @param string|null $key
+	 * @param array $data
+	 * @param bool $isAck
+	 * @throws Exception
+	 */
+	public function batch(?string $key, array $data, bool $isAck = false)
+	{
+		$this->sendMessage($data, $key, $isAck);
+	}
+
+
+	/**
+	 * @return Producer
+	 * @throws Exception
+	 */
+	private function getProducer(): Producer
+	{
+		return Kiri::getDi()->get(Producer::class, [$this->conf]);
+	}
+
+
+	/**
+	 * @param Producer $producer
+	 * @param $topic
+	 * @param $isAck
+	 * @return ProducerTopic
+	 */
+	private function getProducerTopic(Producer $producer, $topic, $isAck): ProducerTopic
+	{
+		$this->topicConf->setRequestRequiredAcks($isAck ? '1' : '0');
+		return $producer->newTopic($topic, $this->topicConf);
+	}
+
+
+	/**
+	 * @param array $message
+	 * @param string $key
+	 * @param bool $isAck
+	 * @throws Exception
+	 */
+	private function sendMessage(array $message, string $key = '', bool $isAck = false)
+	{
+		$producer = $this->getProducer();
+		$producerTopic = $this->getProducerTopic($producer, $this->topic, $isAck);
+		if ($this->isAck) {
+			$this->flush($producer);
+		}
+		foreach ($message as $value) {
+			$producerTopic->produce(RD_KAFKA_PARTITION_UA, 0, swoole_serialize($value), $key);
+			$producer->poll(0);
+		}
+		$this->flush($producer);
+	}
+
+
+	/**
+	 * @param Producer $producer
+	 */
+	private function flush(Producer $producer)
+	{
+		while ($producer->getOutQLen() > 0) {
+			$result = $producer->flush(100);
+			if (RD_KAFKA_RESP_ERR_NO_ERROR === $result) {
+				break;
+			}
+		}
+	}
+
+
+}