migrate rabbitmq to service bus (Azure-Samples#18)

Author: yiliuTo

rabbitmq-custom-formula-source/pom.xml

@@ -16,10 +16,25 @@
 	<properties>
 		<java.version>17</java.version>
 	</properties>
+	<dependencyManagement>
+		<dependencies>
+			<dependency>
+				<groupId>com.azure.spring</groupId>
+				<artifactId>spring-cloud-azure-dependencies</artifactId>
+				<version>5.22.0</version>
+				<type>pom</type>
+				<scope>import</scope>
+			</dependency>
+		</dependencies>
+	</dependencyManagement>
 	<dependencies>
 		<dependency>
-			<groupId>org.springframework.boot</groupId>
-			<artifactId>spring-boot-starter-amqp</artifactId>
+			<groupId>com.azure.spring</groupId>
+			<artifactId>spring-cloud-azure-starter</artifactId>
+		</dependency>
+		<dependency>
+			<groupId>com.azure.spring</groupId>
+			<artifactId>spring-messaging-azure-servicebus</artifactId>
 		</dependency>
 
 		<dependency>

rabbitmq-custom-formula-source/src/main/java/com/example/messagingrabbitmq/MessagingRabbitmqApplication.java

@@ -1,10 +1,15 @@
 package com.example.messagingrabbitmq;
 
+import com.azure.core.credential.TokenCredential;
+import com.azure.core.exception.ResourceNotFoundException;
+import com.azure.messaging.servicebus.administration.ServiceBusAdministrationClient;
+import com.azure.messaging.servicebus.administration.ServiceBusAdministrationClientBuilder;
+import com.azure.messaging.servicebus.administration.models.QueueProperties;
+import com.azure.spring.cloud.autoconfigure.implementation.servicebus.properties.AzureServiceBusProperties;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 import org.springframework.context.ConfigurableApplicationContext;
 import org.springframework.context.annotation.Bean;
-import org.springframework.amqp.core.Queue;
 
 @SpringBootApplication
 public class MessagingRabbitmqApplication {
@@ -18,8 +23,21 @@ public static void main(String[] args) throws InterruptedException {
     }
 
     @Bean
-    public Queue queue() {
-        return new Queue(queueName, true);
+    ServiceBusAdministrationClient serviceBusAdministrationClient(
+        AzureServiceBusProperties properties,
+        TokenCredential credential) {
+        return new ServiceBusAdministrationClientBuilder()
+            .credential(properties.getFullyQualifiedNamespace(), credential)
+            .buildClient();
+    }
+
+    @Bean
+    QueueProperties queue(ServiceBusAdministrationClient adminClient) {
+        try {
+            return adminClient.getQueue(queueName);
+        } catch (ResourceNotFoundException e) {
+            return adminClient.createQueue(queueName);
+        }
     }
 
 }

rabbitmq-custom-formula-source/src/main/java/com/example/messagingrabbitmq/Producer.java

@@ -1,25 +1,26 @@
 package com.example.messagingrabbitmq;
 
-import org.springframework.amqp.core.Message;
+import org.springframework.messaging.Message;
 import org.springframework.stereotype.Component;
-import org.springframework.amqp.rabbit.core.RabbitTemplate;
+import com.azure.spring.messaging.servicebus.core.ServiceBusTemplate;
+import org.springframework.messaging.support.MessageBuilder;
 import org.springframework.beans.factory.annotation.Autowired;
 
 @Component
 public class Producer {
 
     @Autowired
-    private final RabbitTemplate rabbitTemplate;
+    private final ServiceBusTemplate serviceBusTemplate;
 
-    public Producer(RabbitTemplate rabbitTemplate) {
-        this.rabbitTemplate = rabbitTemplate;
+    public Producer(ServiceBusTemplate serviceBusTemplate) {
+        this.serviceBusTemplate = serviceBusTemplate;
     }
 
     public void run()  {
         for (int i = 0; i < 10; i++) {
             System.out.println("Sending message..." + i);
             String responseString = "test " + i;
-            rabbitTemplate.convertAndSend(MessagingRabbitmqApplication.queueName, new Message(responseString.getBytes()));
+            serviceBusTemplate.send(MessagingRabbitmqApplication.queueName, MessageBuilder.withPayload(responseString).build());
         }
     }
 }

rabbitmq-custom-formula-source/src/main/resources/application.properties

@@ -1,4 +1,4 @@
-spring.rabbitmq.password=secret
-spring.rabbitmq.username=myuser
-spring.rabbitmq.host=localhost
-spring.rabbitmq.port=5671
+spring.cloud.azure.servicebus.entity-type=queue
+spring.cloud.azure.servicebus.namespace=
+spring.cloud.azure.credential.managed-identity-enabled= true
+spring.cloud.azure.credential.client-id=