diff --git a/pubsub/api/src/create_topic_with_aws_msk_ingestion.php b/pubsub/api/src/create_topic_with_aws_msk_ingestion.php new file mode 100644 index 0000000000..b3d04c1702 --- /dev/null +++ b/pubsub/api/src/create_topic_with_aws_msk_ingestion.php @@ -0,0 +1,71 @@ + $projectId, + ]); + + $topic = $pubsub->createTopic($topicName, [ + 'ingestionDataSourceSettings' => [ + 'aws_msk' => [ + 'cluster_arn' => $clusterArn, + 'topic' => $mskTopic, + 'aws_role_arn' => $awsRoleArn, + 'gcp_service_account' => $gcpServiceAccount + ] + ] + ]); + + printf('Topic created: %s' . PHP_EOL, $topic->name()); +} +# [END pubsub_create_topic_with_aws_msk_ingestion] +require_once __DIR__ . '/../../../testing/sample_helpers.php'; +\Google\Cloud\Samples\execute_sample(__FILE__, __NAMESPACE__, $argv); diff --git a/pubsub/api/test/pubsubTest.php b/pubsub/api/test/pubsubTest.php index d570411ad2..3bc53d0cc4 100644 --- a/pubsub/api/test/pubsubTest.php +++ b/pubsub/api/test/pubsubTest.php @@ -624,6 +624,7 @@ public function testUpdateTopicType() $this->assertMatchesRegularExpression('/Topic updated:/', $output); $this->assertMatchesRegularExpression(sprintf('/%s/', $topic), $output); } + public function testCreateTopicWithCloudStorageIngestion() { $this->requireEnv('PUBSUB_EMULATOR_HOST'); @@ -648,4 +649,28 @@ public function testCreateTopicWithCloudStorageIngestion() $this->assertMatchesRegularExpression('/Topic deleted:/', $output); $this->assertMatchesRegularExpression(sprintf('/%s/', $topic), $output); } + + public function testCreateTopicWithAwsMskIngestion() + { + $this->requireEnv('PUBSUB_EMULATOR_HOST'); + + $topic = 'test-topic-' . rand(); + $output = $this->runFunctionSnippet('create_topic_with_aws_msk_ingestion', [ + self::$projectId, + $topic, + 'arn:aws:kafka:us-east-1:111111111111:cluster/fake-cluster-name/11111111-1111-1', + 'fake-msk-topic-name', + self::$awsRoleArn, + self::$gcpServiceAccount + ]); + $this->assertMatchesRegularExpression('/Topic created:/', $output); + $this->assertMatchesRegularExpression(sprintf('/%s/', $topic), $output); + + $output = $this->runFunctionSnippet('delete_topic', [ + self::$projectId, + $topic, + ]); + $this->assertMatchesRegularExpression('/Topic deleted:/', $output); + $this->assertMatchesRegularExpression(sprintf('/%s/', $topic), $output); + } }