| <?php |
| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you under the Apache License, Version 2.0 (the |
| // "License"); you may not use this file except in compliance |
| // with the License. You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, |
| // software distributed under the License is distributed on an |
| // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| // KIND, either express or implied. See the License for the |
| // specific language governing permissions and limitations |
| // under the License. |
| |
| declare(strict_types=1); |
| |
| use Iggy\Client; |
| use Iggy\PollingStrategy; |
| use Iggy\SendMessage; |
| |
| function iggy_connection_string(): string |
| { |
| $configured = getenv('IGGY_CONNECTION_STRING'); |
| if ($configured !== false && $configured !== '') { |
| return $configured; |
| } |
| |
| $host = getenv('IGGY_HOST') ?: '127.0.0.1'; |
| $port = getenv('IGGY_PORT') ?: '8090'; |
| $username = rawurlencode(getenv('IGGY_USERNAME') ?: 'iggy'); |
| $password = rawurlencode(getenv('IGGY_PASSWORD') ?: 'iggy'); |
| |
| return "iggy+tcp://{$username}:{$password}@{$host}:{$port}"; |
| } |
| |
| function iggy_client(): Client |
| { |
| $client = Client::fromConnectionString(iggy_connection_string()); |
| $client->connect(); |
| |
| return $client; |
| } |
| |
| function ensure_stream_and_topic(Client $client, string $stream, string $topic): void |
| { |
| if ($client->getStream($stream) === null) { |
| $client->createStream($stream); |
| } |
| |
| if ($client->getTopic($stream, $topic) === null) { |
| $client->createTopic($stream, $topic, 1); |
| } |
| } |
| |
| function send_payloads(Client $client, string $stream, string $topic, array $payloads): void |
| { |
| $messages = array_map( |
| static fn (string $payload): SendMessage => new SendMessage($payload), |
| $payloads, |
| ); |
| |
| $client->sendMessages($stream, $topic, 0, $messages); |
| } |
| |
| function print_polled_messages(Client $client, string $stream, string $topic, int $count): void |
| { |
| $messages = $client->pollMessages($stream, $topic, 0, PollingStrategy::first(), $count, true); |
| |
| if (count($messages) < $count) { |
| throw new RuntimeException("Expected {$count} messages, received " . count($messages)); |
| } |
| |
| foreach ($messages as $message) { |
| echo "Received message at offset {$message->offset()}: {$message->payload()}", PHP_EOL; |
| } |
| } |
| |
| function example_payloads(string $prefix, int $count): array |
| { |
| return array_map( |
| static fn (int $index): string => "{$prefix}-message-{$index}", |
| range(1, $count), |
| ); |
| } |