| <?php |
| |
| /* |
| * This file is part of the Symfony package. |
| * |
| * (c) Fabien Potencier <fabien@symfony.com> |
| * |
| * For the full copyright and license information, please view the LICENSE |
| * file that was distributed with this source code. |
| */ |
| |
| namespace Symfony\Component\VarDumper\Caster; |
| |
| use RdKafka\Conf; |
| use RdKafka\Exception as RdKafkaException; |
| use RdKafka\KafkaConsumer; |
| use RdKafka\Message; |
| use RdKafka\Metadata\Broker as BrokerMetadata; |
| use RdKafka\Metadata\Collection as CollectionMetadata; |
| use RdKafka\Metadata\Partition as PartitionMetadata; |
| use RdKafka\Metadata\Topic as TopicMetadata; |
| use RdKafka\Topic; |
| use RdKafka\TopicConf; |
| use RdKafka\TopicPartition; |
| use Symfony\Component\VarDumper\Cloner\Stub; |
| |
| /** |
| * Casts RdKafka related classes to array representation. |
| * |
| * @author Romain Neutron <imprec@gmail.com> |
| */ |
| class RdKafkaCaster |
| { |
| public static function castKafkaConsumer(KafkaConsumer $c, array $a, Stub $stub, bool $isNested) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| try { |
| $assignment = $c->getAssignment(); |
| } catch (RdKafkaException $e) { |
| $assignment = []; |
| } |
| |
| $a += [ |
| $prefix.'subscription' => $c->getSubscription(), |
| $prefix.'assignment' => $assignment, |
| ]; |
| |
| $a += self::extractMetadata($c); |
| |
| return $a; |
| } |
| |
| public static function castTopic(Topic $c, array $a, Stub $stub, bool $isNested) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| $a += [ |
| $prefix.'name' => $c->getName(), |
| ]; |
| |
| return $a; |
| } |
| |
| public static function castTopicPartition(TopicPartition $c, array $a) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| $a += [ |
| $prefix.'offset' => $c->getOffset(), |
| $prefix.'partition' => $c->getPartition(), |
| $prefix.'topic' => $c->getTopic(), |
| ]; |
| |
| return $a; |
| } |
| |
| public static function castMessage(Message $c, array $a, Stub $stub, bool $isNested) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| $a += [ |
| $prefix.'errstr' => $c->errstr(), |
| ]; |
| |
| return $a; |
| } |
| |
| public static function castConf(Conf $c, array $a, Stub $stub, bool $isNested) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| foreach ($c->dump() as $key => $value) { |
| $a[$prefix.$key] = $value; |
| } |
| |
| return $a; |
| } |
| |
| public static function castTopicConf(TopicConf $c, array $a, Stub $stub, bool $isNested) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| foreach ($c->dump() as $key => $value) { |
| $a[$prefix.$key] = $value; |
| } |
| |
| return $a; |
| } |
| |
| public static function castRdKafka(\RdKafka $c, array $a, Stub $stub, bool $isNested) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| $a += [ |
| $prefix.'out_q_len' => $c->getOutQLen(), |
| ]; |
| |
| $a += self::extractMetadata($c); |
| |
| return $a; |
| } |
| |
| public static function castCollectionMetadata(CollectionMetadata $c, array $a, Stub $stub, bool $isNested) |
| { |
| $a += iterator_to_array($c); |
| |
| return $a; |
| } |
| |
| public static function castTopicMetadata(TopicMetadata $c, array $a, Stub $stub, bool $isNested) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| $a += [ |
| $prefix.'name' => $c->getTopic(), |
| $prefix.'partitions' => $c->getPartitions(), |
| ]; |
| |
| return $a; |
| } |
| |
| public static function castPartitionMetadata(PartitionMetadata $c, array $a, Stub $stub, bool $isNested) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| $a += [ |
| $prefix.'id' => $c->getId(), |
| $prefix.'err' => $c->getErr(), |
| $prefix.'leader' => $c->getLeader(), |
| ]; |
| |
| return $a; |
| } |
| |
| public static function castBrokerMetadata(BrokerMetadata $c, array $a, Stub $stub, bool $isNested) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| $a += [ |
| $prefix.'id' => $c->getId(), |
| $prefix.'host' => $c->getHost(), |
| $prefix.'port' => $c->getPort(), |
| ]; |
| |
| return $a; |
| } |
| |
| private static function extractMetadata($c) |
| { |
| $prefix = Caster::PREFIX_VIRTUAL; |
| |
| try { |
| $m = $c->getMetadata(true, null, 500); |
| } catch (RdKafkaException $e) { |
| return []; |
| } |
| |
| return [ |
| $prefix.'orig_broker_id' => $m->getOrigBrokerId(), |
| $prefix.'orig_broker_name' => $m->getOrigBrokerName(), |
| $prefix.'brokers' => $m->getBrokers(), |
| $prefix.'topics' => $m->getTopics(), |
| ]; |
| } |
| } |